Merge branch 'development' into release-3.8.2

Author: obinnaokafor

src/examples/cleanlab_example/simple.py

@@ -0,0 +1,18 @@
+import os
+
+from cleanlab_tlm import TLM
+from dotenv import find_dotenv, load_dotenv
+
+from langtrace_python_sdk import langtrace
+
+_ = load_dotenv(find_dotenv())
+
+langtrace.init()
+
+api_key = os.getenv("CLEANLAB_API_KEY")
+tlm = TLM(api_key=api_key, options={"log": ["explanation"], "model": "gpt-4o-mini"}) # GPT, Claude, etc
+out = tlm.prompt("What's the third month of the year alphabetically?")
+trustworthiness_score = tlm.get_trustworthiness_score("What's the first month of the year?", response="January")
+
+print(out)
+print(trustworthiness_score)

src/langtrace_python_sdk/constants/instrumentation/common.py

@@ -13,6 +13,8 @@
     "ARCH": "Arch",
     "AZURE": "Azure",
     "CHROMA": "Chroma",
+    "CLEANLAB": "CleanLab",
+    "COHERE": "Cohere",
     "CREWAI": "CrewAI",
     "DEEPSEEK": "DeepSeek",
     "DSPY": "DSPy",
@@ -25,7 +27,6 @@
     "LLAMAINDEX": "LlamaIndex",
     "OPENAI": "OpenAI",
     "PINECONE": "Pinecone",
-    "COHERE": "Cohere",
     "PPLX": "Perplexity",
     "QDRANT": "Qdrant",
     "WEAVIATE": "Weaviate",
@@ -42,6 +43,7 @@
     "MILVUS": "Milvus",
     "GRAPHLIT": "Graphlit",
     "PHIDATA": "Phidata",
+    "AGNO": "Agno",
 }
 
 LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY = "langtrace_additional_attributes"

src/langtrace_python_sdk/instrumentation/__init__.py

@@ -1,15 +1,18 @@
+from .agno import AgnoInstrumentation
 from .anthropic import AnthropicInstrumentation
 from .autogen import AutogenInstrumentation
 from .aws_bedrock import AWSBedrockInstrumentation
 from .cerebras import CerebrasInstrumentation
 from .chroma import ChromaInstrumentation
+from .cleanlab import CleanLabInstrumentation
 from .cohere import CohereInstrumentation
 from .crewai import CrewAIInstrumentation
 from .crewai_tools import CrewaiToolsInstrumentation
 from .dspy import DspyInstrumentation
 from .embedchain import EmbedchainInstrumentation
 from .gemini import GeminiInstrumentation
 from .google_genai import GoogleGenaiInstrumentation
+from .graphlit import GraphlitInstrumentation
 from .groq import GroqInstrumentation
 from .langchain import LangchainInstrumentation
 from .langchain_community import LangchainCommunityInstrumentation
@@ -21,16 +24,12 @@
 from .mistral import MistralInstrumentation
 from .ollama import OllamaInstrumentor
 from .openai import OpenAIInstrumentation
+from .phidata import PhiDataInstrumentation
 from .pinecone import PineconeInstrumentation
 from .pymongo import PyMongoInstrumentation
 from .qdrant import QdrantInstrumentation
 from .vertexai import VertexAIInstrumentation
 from .weaviate import WeaviateInstrumentation
-from .cerebras import CerebrasInstrumentation
-from .milvus import MilvusInstrumentation
-from .google_genai import GoogleGenaiInstrumentation
-from .graphlit import GraphlitInstrumentation
-from .phidata import PhiDataInstrumentation
 
 __all__ = [
     "AnthropicInstrumentation",
@@ -63,4 +62,6 @@
     "CrewaiToolsInstrumentation",
     "GraphlitInstrumentation",
     "PhiDataInstrumentation",
+    "AgnoInstrumentation",
+    "CleanLabInstrumentation",
 ]

src/langtrace_python_sdk/instrumentation/agno/__init__.py

@@ -0,0 +1,5 @@
+from .instrumentation import AgnoInstrumentation
+
+__all__ = [
+    "AgnoInstrumentation",
+]

src/langtrace_python_sdk/instrumentation/agno/instrumentation.py

@@ -0,0 +1,80 @@
+"""
+Copyright (c) 2024 Scale3 Labs
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
+from opentelemetry.trace import get_tracer
+from wrapt import wrap_function_wrapper as _W
+from typing import Collection
+from importlib_metadata import version as v
+from .patch import patch_agent, patch_memory
+
+class AgnoInstrumentation(BaseInstrumentor):
+    def instrumentation_dependencies(self) -> Collection[str]:
+        return ["agno >= 1.1.4"]
+
+    def _instrument(self, **kwargs):
+        tracer_provider = kwargs.get("tracer_provider")
+        tracer = get_tracer(__name__, "", tracer_provider)
+        version = v("agno")
+
+        try:
+            _W(
+                "agno.agent.agent",
+                "Agent.run",
+                patch_agent("Agent.run", version, tracer),
+            )
+            _W(
+                "agno.agent.agent",
+                "Agent.arun",
+                patch_agent("Agent.arun", version, tracer),
+            )
+            _W(
+                "agno.agent.agent",
+                "Agent._run",
+                patch_agent("Agent._run", version, tracer),
+            )
+            _W(
+                "agno.agent.agent", 
+                "Agent._arun",
+                patch_agent("Agent._arun", version, tracer),
+            )
+
+            _W(
+                "agno.memory.agent",
+                "AgentMemory.update_memory",
+                patch_memory("AgentMemory.update_memory", version, tracer),
+            )
+            _W(
+                "agno.memory.agent",
+                "AgentMemory.aupdate_memory",
+                patch_memory("AgentMemory.aupdate_memory", version, tracer),
+            )
+            _W(
+                "agno.memory.agent",
+                "AgentMemory.update_summary",
+                patch_memory("AgentMemory.update_summary", version, tracer),
+            )
+            _W(
+                "agno.memory.agent",
+                "AgentMemory.aupdate_summary",
+                patch_memory("AgentMemory.aupdate_summary", version, tracer),
+            )
+
+        except Exception:
+            pass
+
+    def _uninstrument(self, **kwargs):
+        pass