Skip to content

Commit 8ae6146

Browse files
karthikscale3obinnaokaforobinnascale3
authored
Release (#515)
* remove logs * remove requirements * Bump version * Squash * minor * switch to http exporter * add instrumentation for neo4j * update version * update trace attributes version * update imports * add neo4j result span attributes * add neo4j result span attributes * bump bersion * fix db attributes error and add examples * handle result transformer * fix genai and pinecone instrumentations (#512) * fix genai and pinecone instrumentations * bump version * fix pinecone Index attribute --------- Co-authored-by: Obinna Okafor <obinna.okafor01@gmail.com> * Improve Agno instrumentation (#513) --------- Co-authored-by: Obinna Okafor <obinna.okafor01@gmail.com> Co-authored-by: obinnascale3 <109410793+obinnascale3@users.noreply.github.com>
1 parent f53ad18 commit 8ae6146

6 files changed

Lines changed: 208 additions & 75 deletions

File tree

src/langtrace_python_sdk/instrumentation/agno/instrumentation.py

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from opentelemetry.trace import get_tracer
2222
from wrapt import wrap_function_wrapper as _W
2323

24-
from .patch import patch_agent, patch_memory
24+
from .patch import patch_agent, patch_memory, patch_team
2525

2626

2727
class AgnoInstrumentation(BaseInstrumentor):
@@ -76,6 +76,27 @@ def _instrument(self, **kwargs):
7676
patch_memory("AgentMemory.aupdate_summary", version, tracer),
7777
)
7878

79+
_W(
80+
"agno.team.team",
81+
"Team.run",
82+
patch_team("Team.run", version, tracer),
83+
)
84+
_W(
85+
"agno.team.team",
86+
"Team.arun",
87+
patch_team("Team.arun", version, tracer),
88+
)
89+
_W(
90+
"agno.team.team",
91+
"Team._run",
92+
patch_team("Team._run", version, tracer),
93+
)
94+
_W(
95+
"agno.team.team",
96+
"Team._arun",
97+
patch_team("Team._arun", version, tracer),
98+
)
99+
79100
except Exception:
80101
pass
81102

0 commit comments

Comments
 (0)