merge

Author: karthikscale3

src/langtrace_python_sdk/instrumentation/crewai/patch.py

@@ -51,7 +51,6 @@ def traced_method(wrapped, instance, args, kwargs):
                     )
                 if result:
                     span.set_status(Status(StatusCode.OK))
-                span.end()
                 return result
 
             except Exception as err:
@@ -102,7 +101,6 @@ def traced_method(wrapped, instance, args, kwargs):
                                 span.set_attribute(
                                     f"crewai.crew.{attr}", str(getattr(result, attr))
                                 )
-                span.end()
                 return result
 
             except Exception as err:
@@ -139,21 +137,27 @@ def run(self):
             for key, value in self.crew.items():
                 key = f"crewai.crew.{key}"
                 if value is not None:
-                    set_span_attribute(self.span, key, value)
+                    set_span_attribute(
+                        self.span, key, str(value) if isinstance(value, list) else value
+                    )
 
         elif instance_name == "Agent":
             agent = self.set_agent_attributes()
             for key, value in agent.items():
                 key = f"crewai.agent.{key}"
                 if value is not None:
-                    set_span_attribute(self.span, key, value)
+                    set_span_attribute(
+                        self.span, key, str(value) if isinstance(value, list) else value
+                    )
 
         elif instance_name == "Task":
             task = self.set_task_attributes()
             for key, value in task.items():
                 key = f"crewai.task.{key}"
                 if value is not None:
-                    set_span_attribute(self.span, key, value)
+                    set_span_attribute(
+                        self.span, key, str(value) if isinstance(value, list) else value
+                    )
 
     def set_crew_attributes(self):
         for key, value in self.instance.__dict__.items():

src/langtrace_python_sdk/langtrace.py

@@ -17,7 +17,7 @@
 import os
 import sys
 from typing import Optional
-
+import importlib.util
 from colorama import Fore
 from opentelemetry import trace
 from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
@@ -120,25 +120,25 @@ def init(
     all_instrumentations = {
         "openai": OpenAIInstrumentation(),
         "groq": GroqInstrumentation(),
-        "pinecone": PineconeInstrumentation(),
-        "llamaindex": LlamaindexInstrumentation(),
-        "chroma": ChromaInstrumentation(),
+        "pinecone-client": PineconeInstrumentation(),
+        "llama-index": LlamaindexInstrumentation(),
+        "chromadb": ChromaInstrumentation(),
         "embedchain": EmbedchainInstrumentation(),
-        "qdrant": QdrantInstrumentation(),
+        "qdrant-client": QdrantInstrumentation(),
         "langchain": LangchainInstrumentation(),
-        "langchain_core": LangchainCoreInstrumentation(),
-        "langchain_community": LangchainCommunityInstrumentation(),
+        "langchain-core": LangchainCoreInstrumentation(),
+        "langchain-community": LangchainCommunityInstrumentation(),
         "langgraph": LanggraphInstrumentation(),
         "anthropic": AnthropicInstrumentation(),
         "cohere": CohereInstrumentation(),
-        "weaviate": WeaviateInstrumentation(),
+        "weaviate-client": WeaviateInstrumentation(),
         "sqlalchemy": SQLAlchemyInstrumentor(),
         "ollama": OllamaInstrumentor(),
-        "dspy": DspyInstrumentation(),
+        "dspy-ai": DspyInstrumentation(),
         "crewai": CrewAIInstrumentation(),
-        "vertexai": VertexAIInstrumentation(),
-        "gemini": GeminiInstrumentation(),
-        "mistral": MistralInstrumentation(),
+        "google-cloud-aiplatform": VertexAIInstrumentation(),
+        "google-generativeai": GeminiInstrumentation(),
+        "mistralai": MistralInstrumentation(),
     }
 
     init_instrumentations(disable_instrumentations, all_instrumentations)
@@ -171,8 +171,9 @@ def init_instrumentations(
 ):
     if disable_instrumentations is None:
         for idx, (name, v) in enumerate(all_instrumentations.items()):
+            if is_package_installed(name):
+                v.instrument()
 
-            v.instrument()
     else:
 
         validate_instrumentations(disable_instrumentations)
@@ -229,3 +230,10 @@ def validate_instrumentations(disable_instrumentations):
             raise ValueError(
                 "Cannot specify both only and all_except in disable_instrumentations"
             )
+
+
+def is_package_installed(package_name):
+    import pkg_resources
+
+    installed_packages = {p.key for p in pkg_resources.working_set}
+    return package_name in installed_packages

src/langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "2.3.9"
+__version__ = "2.3.12"