langtrace-python-sdk 2.2.5__tar.gz → 2.2.7__tar.gz

{langtrace_python_sdk-2.2.5 → langtrace_python_sdk-2.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.2.5
+Version: 2.2.7
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -20,7 +20,7 @@ Requires-Dist: opentelemetry-instrumentation>=0.46b0
 Requires-Dist: opentelemetry-sdk>=1.25.0
 Requires-Dist: sqlalchemy
 Requires-Dist: tiktoken>=0.1.1
-Requires-Dist: trace-attributes<7.0.0,>=6.0.3
+Requires-Dist: trace-attributes==7.0.0
 Provides-Extra: dev
 Requires-Dist: anthropic; extra == 'dev'
 Requires-Dist: chromadb; extra == 'dev'

{langtrace_python_sdk-2.2.5 → langtrace_python_sdk-2.2.7}/pyproject.toml

@@ -18,7 +18,7 @@ classifiers=[
     "Operating System :: OS Independent",
 ]
 dependencies = [
-  'trace-attributes>=6.0.3,<7.0.0',
+  'trace-attributes==7.0.0',
   'opentelemetry-api>=1.25.0',
   'opentelemetry-sdk>=1.25.0',
   'opentelemetry-instrumentation>=0.46b0',

langtrace_python_sdk-2.2.7/src/examples/otlp_example/otlp_with_langtrace.py

@@ -0,0 +1,59 @@
+# Instructions
+# 1. Run the OpenTelemetry Collector with the OTLP receiver enabled
+# Create otel-config.yaml with the following content:
+# receivers:
+#   otlp:
+#     protocols:
+#       grpc:
+#         endpoint: "0.0.0.0:4317"
+#       http:
+#         endpoint: "0.0.0.0:4318"
+
+# exporters:
+#   logging:
+#     loglevel: debug
+
+# service:
+#   pipelines:
+#     traces:
+#       receivers: [otlp]
+#       exporters: [logging]
+# docker pull otel/opentelemetry-collector:latest
+# docker run --rm -p 4317:4317 -p 4318:4318 -v $(pwd)/otel-config.yaml:/otel-config.yaml otel/opentelemetry-collector --config otel-config.yaml
+# 2. Run the following code
+
+from langtrace_python_sdk import langtrace
+from openai import OpenAI
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+
+
+# Configure the OTLP exporter to use the correct endpoint and API key
+otlp_endpoint = "http://localhost:4318/v1/traces"
+otlp_exporter = OTLPSpanExporter(
+    endpoint=otlp_endpoint,
+    headers=(("Content-Type", "application/json"),))
+langtrace.init(custom_remote_exporter=otlp_exporter, batch=False)
+
+
+def chat_with_openai():
+    client = OpenAI()
+    messages = [
+        {
+            "role": "user",
+            "content": "Hello, I'm a human.",
+        },
+    ]
+    chat_completion = client.chat.completions.create(
+        messages=messages,
+        stream=False,
+        model="gpt-3.5-turbo",
+    )
+    print(chat_completion.choices[0].message.content)
+
+
+def main():
+    chat_with_openai()
+
+
+if __name__ == "__main__":
+    main()

{langtrace_python_sdk-2.2.5 → langtrace_python_sdk-2.2.7}/src/langtrace_python_sdk/instrumentation/cohere/patch.py

@@ -44,7 +44,7 @@ def rerank(original_method, version, tracer):
 
         span_attributes = {
             **get_langtrace_attributes(version, service_provider),
-            **get_llm_request_attributes(kwargs),
+            **get_llm_request_attributes(kwargs, operation_name="rerank"),
             **get_llm_url(instance),
             SpanAttributes.LLM_REQUEST_MODEL: kwargs.get("model") or "command-r-plus",
             SpanAttributes.LLM_URL: APIS["RERANK"]["URL"],
@@ -121,7 +121,7 @@ def embed(original_method, version, tracer):
 
         span_attributes = {
             **get_langtrace_attributes(version, service_provider),
-            **get_llm_request_attributes(kwargs),
+            **get_llm_request_attributes(kwargs, operation_name="embed"),
             **get_llm_url(instance),
             SpanAttributes.LLM_URL: APIS["EMBED"]["URL"],
             SpanAttributes.LLM_PATH: APIS["EMBED"]["ENDPOINT"],

{langtrace_python_sdk-2.2.5 → langtrace_python_sdk-2.2.7}/src/langtrace_python_sdk/instrumentation/crewai/instrumentation.py

@@ -33,21 +33,24 @@ class CrewAIInstrumentation(BaseInstrumentor):
         tracer_provider = kwargs.get("tracer_provider")
         tracer = get_tracer(__name__, "", tracer_provider)
         version = v("crewai")
-        _W(
-            "crewai.crew",
-            "Crew.kickoff",
-            patch_crew("Crew.kickoff", version, tracer),
-        )
-        _W(
-            "crewai.agent",
-            "Agent.execute_task",
-            patch_crew("Agent.execute_task", version, tracer),
-        )
-        _W(
-            "crewai.task",
-            "Task.execute",
-            patch_crew("Task.execute", version, tracer),
-        )
+        try:
+            _W(
+                "crewai.crew",
+                "Crew.kickoff",
+                patch_crew("Crew.kickoff", version, tracer),
+            )
+            _W(
+                "crewai.agent",
+                "Agent.execute_task",
+                patch_crew("Agent.execute_task", version, tracer),
+            )
+            _W(
+                "crewai.task",
+                "Task.execute",
+                patch_crew("Task.execute", version, tracer),
+            )
+        except Exception as e:
+            pass
 
     def _uninstrument(self, **kwargs):
         pass

{langtrace_python_sdk-2.2.5 → langtrace_python_sdk-2.2.7}/src/langtrace_python_sdk/instrumentation/openai/patch.py

@@ -55,7 +55,7 @@ def images_generate(original_method, version, tracer):
         service_provider = SERVICE_PROVIDERS["OPENAI"]
         span_attributes = {
             **get_langtrace_attributes(version, service_provider, vendor_type="llm"),
-            **get_llm_request_attributes(kwargs),
+            **get_llm_request_attributes(kwargs, operation_name="images_generate"),
             **get_llm_url(instance),
             SpanAttributes.LLM_PATH: APIS["IMAGES_GENERATION"]["ENDPOINT"],
             **get_extra_attributes(),
@@ -118,7 +118,7 @@ def async_images_generate(original_method, version, tracer):
 
         span_attributes = {
             **get_langtrace_attributes(version, service_provider, vendor_type="llm"),
-            **get_llm_request_attributes(kwargs),
+            **get_llm_request_attributes(kwargs, operation_name="images_generate"),
             **get_llm_url(instance),
             SpanAttributes.LLM_PATH: APIS["IMAGES_GENERATION"]["ENDPOINT"],
             **get_extra_attributes(),
@@ -181,7 +181,7 @@ def images_edit(original_method, version, tracer):
 
         span_attributes = {
             **get_langtrace_attributes(version, service_provider, vendor_type="llm"),
-            **get_llm_request_attributes(kwargs),
+            **get_llm_request_attributes(kwargs, operation_name="images_edit"),
             **get_llm_url(instance),
             SpanAttributes.LLM_PATH: APIS["IMAGES_EDIT"]["ENDPOINT"],
             SpanAttributes.LLM_RESPONSE_FORMAT: kwargs.get("response_format"),
@@ -432,7 +432,7 @@ def embeddings_create(original_method, version, tracer):
 
         span_attributes = {
             **get_langtrace_attributes(version, service_provider, vendor_type="llm"),
-            **get_llm_request_attributes(kwargs),
+            **get_llm_request_attributes(kwargs, operation_name="embed"),
             **get_llm_url(instance),
             SpanAttributes.LLM_PATH: APIS["EMBEDDINGS_CREATE"]["ENDPOINT"],
             SpanAttributes.LLM_REQUEST_DIMENSIONS: kwargs.get("dimensions"),
@@ -490,7 +490,7 @@ def async_embeddings_create(original_method, version, tracer):
 
         span_attributes = {
             **get_langtrace_attributes(version, service_provider, vendor_type="llm"),
-            **get_llm_request_attributes(kwargs),
+            **get_llm_request_attributes(kwargs, operation_name="embed"),
             SpanAttributes.LLM_PATH: APIS["EMBEDDINGS_CREATE"]["ENDPOINT"],
             SpanAttributes.LLM_REQUEST_DIMENSIONS: kwargs.get("dimensions"),
             **get_extra_attributes(),

{langtrace_python_sdk-2.2.5 → langtrace_python_sdk-2.2.7}/src/langtrace_python_sdk/utils/llm.py

@@ -88,10 +88,11 @@ def get_langtrace_attributes(version, service_provider, vendor_type="llm"):
         SpanAttributes.LANGTRACE_SERVICE_VERSION: version,
         SpanAttributes.LANGTRACE_SERVICE_NAME: service_provider,
         SpanAttributes.LANGTRACE_SERVICE_TYPE: vendor_type,
+        SpanAttributes.LLM_SYSTEM: service_provider,
     }
 
 
-def get_llm_request_attributes(kwargs, prompts=None, model=None):
+def get_llm_request_attributes(kwargs, prompts=None, model=None, operation_name="chat"):
 
     user = kwargs.get("user", None)
     if prompts is None:
@@ -110,6 +111,7 @@ def get_llm_request_attributes(kwargs, prompts=None, model=None):
     top_p = kwargs.get("p", None) or kwargs.get("top_p", None)
     tools = kwargs.get("tools", None)
     return {
+        SpanAttributes.LLM_OPERATION_NAME: operation_name,
         SpanAttributes.LLM_REQUEST_MODEL: model or kwargs.get("model"),
         SpanAttributes.LLM_IS_STREAMING: kwargs.get("stream"),
         SpanAttributes.LLM_REQUEST_TEMPERATURE: kwargs.get("temperature"),

langtrace_python_sdk-2.2.7/src/langtrace_python_sdk/version.py

@@ -0,0 +1 @@
+__version__ = "2.2.7"

langtrace_python_sdk-2.2.5/src/examples/ollama_example/basic_example_2.py

@@ -1,34 +0,0 @@
-from langtrace_python_sdk import langtrace
-from openai import OpenAI
-from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
-
-service_name = "langtrace-python-ollama"
-otlp_endpoint = "http://localhost:4318/v1/traces"
-otlp_exporter = OTLPSpanExporter(
-    endpoint=otlp_endpoint,
-    headers=(("Content-Type", "application/json"),))
-langtrace.init(custom_remote_exporter=otlp_exporter, batch=False)
-
-
-def chat_with_ollama():
-    # Use the OpenAI endpoint, not the Ollama API.
-    base_url = "http://localhost:11434/v1"
-    client = OpenAI(base_url=base_url, api_key="unused")
-    messages = [
-        {
-            "role": "user",
-            "content": "Hello, I'm a human.",
-        },
-    ]
-    chat_completion = client.chat.completions.create(
-        model="llama3", messages=messages
-    )
-    print(chat_completion.choices[0].message.content)
-
-
-def main():
-    chat_with_ollama()
-
-
-if __name__ == "__main__":
-    main()

langtrace_python_sdk-2.2.5/src/langtrace_python_sdk/version.py

@@ -1 +0,0 @@
-__version__ = "2.2.5"