langtrace-python-sdk 1.2.5__tar.gz → 1.2.7__tar.gz

{langtrace_python_sdk-1.2.5 → langtrace_python_sdk-1.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 1.2.5
+Version: 1.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>
@@ -10,6 +10,20 @@ Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or l
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.9
+Requires-Dist: opentelemetry-api
+Requires-Dist: opentelemetry-instrumentation
+Requires-Dist: opentelemetry-sdk
+Requires-Dist: pinecone-client
+Requires-Dist: tiktoken
+Requires-Dist: trace-attributes
+Provides-Extra: dev
+Requires-Dist: anthropic; extra == 'dev'
+Requires-Dist: chromadb; extra == 'dev'
+Requires-Dist: langchain; extra == 'dev'
+Requires-Dist: langchain-openai; extra == 'dev'
+Requires-Dist: llama-index; extra == 'dev'
+Requires-Dist: openai; extra == 'dev'
+Requires-Dist: python-dotenv; extra == 'dev'
 Description-Content-Type: text/markdown
 
 # What is Langtrace?

{langtrace_python_sdk-1.2.5 → langtrace_python_sdk-1.2.7}/pyproject.toml

@@ -17,10 +17,30 @@ classifiers=[
     "License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)",
     "Operating System :: OS Independent",
 ]
-dependencies = []
+dependencies = [
+  'trace-attributes',
+  'opentelemetry-api',
+  'opentelemetry-sdk',
+  'opentelemetry-instrumentation',
+  'pinecone-client',
+  'tiktoken'
+]
 
 requires-python = ">=3.9"
 
+[project.optional-dependencies]
+dev = [
+    "openai",
+    "anthropic",
+    "python-dotenv",
+    "langchain",
+    "langchain-openai",
+    "llama-index",
+    "langchain-openai",
+    "chromadb"
+]
+
+
 
 [project.urls]
 Homepage = "https://github.com/Scale3-Labs/langtrace-python-sdk"
@@ -35,7 +55,7 @@ include = [
 ]
 
 [tool.hatch.build.targets.wheel]
-packages = ["src/langtrace_python_sdk"]
+packages = ["src/langtrace_python_sdk", "src/examples"]
 
 # [tool.semantic_release]
 # version_variable = "pyproject.toml:version" # version location

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

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

{langtrace_python_sdk-1.2.5 → langtrace_python_sdk-1.2.7}/src/langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -2,9 +2,11 @@ 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.sdk.trace.export import ReadableSpan, SpanExporter, SpanExportResult
 from opentelemetry.trace.span import format_trace_id
 
 
@@ -22,7 +24,6 @@ class LangTraceExporter(SpanExporter):
     **Attributes:**
 
     * `api_key` (str): An API key to authenticate with the LangTrace collector (required).
-    * `url` (str): The URL of the LangTrace collector endpoint (required if `write_to_remote_url` is True).
     * `write_to_remote_url` (bool): A flag indicating whether to send spans to the remote URL (defaults to False).
 
     **Methods:**
@@ -46,22 +47,15 @@ class LangTraceExporter(SpanExporter):
     """
 
     api_key: str
-    url: str
     write_to_remote_url: bool
 
-    def __init__(
-        self, api_key: str = None, url: str = None, write_to_remote_url: bool = False
-    ) -> None:
-        self.api_key = api_key if api_key else os.environ.get("API_KEY")
-        self.url = url if url else os.environ.get("URL")
+    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")
         self.write_to_remote_url = write_to_remote_url
 
         if self.write_to_remote_url and not self.api_key:
             raise ValueError("No API key provided")
 
-        if self.write_to_remote_url and not self.url:
-            raise ValueError("No URL provided")
-
     def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
         """
         Exports a batch of telemetry data.
@@ -92,11 +86,11 @@ class LangTraceExporter(SpanExporter):
         # Send data to remote URL
         try:
             requests.post(
-                url=self.url,
+                url=LANGTRACE_REMOTE_URL,
                 data=json.dumps(data),
                 headers={"Content-Type": "application/json", "x-api-key": self.api_key},
             )
-            print(f"Traces sent To {self.url}")
+            print(f"Traces sent To {LANGTRACE_REMOTE_URL}")
             return SpanExportResult.SUCCESS
         except Exception as e:
             print("Error sending data to remote URL", e)

{langtrace_python_sdk-1.2.5 → langtrace_python_sdk-1.2.7}/src/langtrace_python_sdk/instrumentation/anthropic/patch.py

@@ -8,7 +8,8 @@ from opentelemetry.trace import SpanKind
 from opentelemetry.trace.status import Status, StatusCode
 
 from langtrace_python_sdk.constants.instrumentation.anthropic import APIS
-from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
+from langtrace_python_sdk.constants.instrumentation.common import \
+    SERVICE_PROVIDERS
 
 
 def messages_create(original_method, version, tracer):
@@ -21,6 +22,16 @@ def messages_create(original_method, version, tracer):
             else ""
         )
         service_provider = SERVICE_PROVIDERS["ANTHROPIC"]
+
+        # extract system from kwargs and attach as a role to the prompts
+        # we do this to keep it consistent with the openai
+        prompts = json.dumps(kwargs.get("messages", []))
+        system = kwargs.get("system")
+        if system:
+            prompts = json.dumps(
+                [{"role": "system", "content": system}] + kwargs.get("messages", [])
+            )
+
         span_attributes = {
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
@@ -29,7 +40,7 @@ def messages_create(original_method, version, tracer):
             "url.full": base_url,
             "llm.api": APIS["MESSAGES_CREATE"]["ENDPOINT"],
             "llm.model": kwargs.get("model"),
-            "llm.prompts": json.dumps(kwargs.get("messages", [])),
+            "llm.prompts": prompts,
             "llm.stream": kwargs.get("stream"),
         }
 

{langtrace_python_sdk-1.2.5 → langtrace_python_sdk-1.2.7}/src/langtrace_python_sdk/langtrace.py

@@ -37,17 +37,19 @@ def init(
     api_key: str = None,
     batch: bool = True,
     write_to_langtrace_cloud: bool = True,
-    remote_url: str = None,
     debug_log_to_console: bool = False,
-    custom_remote_exporter = None
+    custom_remote_exporter=None,
 ):
 
     provider = TracerProvider()
 
-    remote_write_exporter = LangTraceExporter(api_key, remote_url, write_to_langtrace_cloud) if custom_remote_exporter is None else custom_remote_exporter
+    remote_write_exporter = (
+        LangTraceExporter(api_key, write_to_langtrace_cloud)
+        if custom_remote_exporter is None
+        else custom_remote_exporter
+    )
     console_exporter = ConsoleSpanExporter()
     batch_processor_remote = BatchSpanProcessor(remote_write_exporter)
-    simple_processor_remote = SimpleSpanProcessor(remote_write_exporter)
     batch_processor_console = BatchSpanProcessor(console_exporter)
     simple_processor_console = SimpleSpanProcessor(console_exporter)
 

langtrace_python_sdk-1.2.7/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-1.2.5/src/langtrace_python_sdk/version.py

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