langtrace-python-sdk 1.2.20__tar.gz → 1.2.23__tar.gz

{langtrace_python_sdk-1.2.20 → langtrace_python_sdk-1.2.23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 1.2.20
+Version: 1.2.23
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>

langtrace_python_sdk-1.2.23/src/langtrace_python_sdk/constants/exporter/langtrace_exporter.py

@@ -0,0 +1 @@
+LANGTRACE_REMOTE_URL = "https://langtrace.ai"

{langtrace_python_sdk-1.2.20 → langtrace_python_sdk-1.2.23}/src/langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -2,13 +2,12 @@ import json
 import os
 import typing
 
-from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
-    LANGTRACE_REMOTE_URL,
-)
 import requests
 from opentelemetry.sdk.trace.export import ReadableSpan, SpanExporter, SpanExportResult
 from opentelemetry.trace.span import format_trace_id
 
+from langtrace_python_sdk.constants.exporter.langtrace_exporter import LANGTRACE_REMOTE_URL
+
 
 class LangTraceExporter(SpanExporter):
     """
@@ -49,9 +48,10 @@ class LangTraceExporter(SpanExporter):
     api_key: str
     write_to_remote_url: bool
 
-    def __init__(self, api_key: str = None, write_to_remote_url: bool = False) -> None:
-        self.api_key = api_key if api_key else os.environ.get("LANGTRACE_API_KEY")
+    def __init__(self, api_key: str = None, write_to_remote_url: bool = False, api_host: typing.Optional[str] = None) -> None:
+        self.api_key = api_key or os.environ.get("LANGTRACE_API_KEY")
         self.write_to_remote_url = write_to_remote_url
+        self.api_host: str = api_host or LANGTRACE_REMOTE_URL
 
         if self.write_to_remote_url and not self.api_key:
             raise ValueError("No API key provided")
@@ -66,7 +66,6 @@ class LangTraceExporter(SpanExporter):
         Returns:
             The result of the export SUCCESS or FAILURE
         """
-
         data = [
             {
                 "traceId": format_trace_id(span.get_span_context().trace_id),
@@ -86,7 +85,7 @@ class LangTraceExporter(SpanExporter):
         # Send data to remote URL
         try:
             requests.post(
-                url=LANGTRACE_REMOTE_URL,
+                url=f"{self.api_host}/api/trace",
                 data=json.dumps(data),
                 headers={"Content-Type": "application/json", "x-api-key": self.api_key},
             )

{langtrace_python_sdk-1.2.20 → langtrace_python_sdk-1.2.23}/src/langtrace_python_sdk/instrumentation/openai/patch.py

@@ -281,30 +281,19 @@ def chat_completions_create(original_method, version, tracer):
                     }
                 ),
             )
-            if function_call is False:
-                span.set_attribute(
-                    "llm.responses",
-                    json.dumps(
-                        {
-                            "message": {
-                                "role": "assistant",
-                                "content": "".join(result_content),
-                            }
-                        }
-                    ),
-                )
-            else:
-                span.set_attribute(
-                    "llm.responses",
-                    json.dumps(
+            span.set_attribute(
+                "llm.responses",
+                json.dumps(
+                    [
                         {
                             "message": {
                                 "role": "assistant",
                                 "content": "".join(result_content),
                             }
                         }
-                    ),
-                )
+                    ]
+                ),
+            )
             span.set_status(StatusCode.OK)
             span.end()
 

{langtrace_python_sdk-1.2.20 → langtrace_python_sdk-1.2.23}/src/langtrace_python_sdk/langtrace.py

@@ -1,4 +1,4 @@
-
+from typing import Optional
 from opentelemetry import trace
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import (
@@ -40,12 +40,12 @@ def init(
     write_to_langtrace_cloud: bool = True,
     debug_log_to_console: bool = False,
     custom_remote_exporter=None,
+    api_host: Optional[str] = None,
 ):
-
     provider = TracerProvider()
 
     remote_write_exporter = (
-        LangTraceExporter(api_key, write_to_langtrace_cloud)
+        LangTraceExporter(api_key, write_to_langtrace_cloud, api_host=api_host)
         if custom_remote_exporter is None
         else custom_remote_exporter
     )

langtrace_python_sdk-1.2.23/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-1.2.20/src/langtrace_python_sdk/constants/exporter/langtrace_exporter.py

@@ -1 +0,0 @@
-LANGTRACE_REMOTE_URL = "https://langtrace.ai/api/trace"

langtrace_python_sdk-1.2.20/src/langtrace_python_sdk/version.py

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