langtrace-python-sdk 2.1.11__tar.gz → 2.1.12__tar.gz

{langtrace_python_sdk-2.1.11 → langtrace_python_sdk-2.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.11
+Version: 2.1.12
 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-2.1.11 → langtrace_python_sdk-2.1.12}/src/langtrace_python_sdk/extensions/langtrace_exporter.py

@@ -48,14 +48,19 @@ class LangTraceExporter(SpanExporter):
     """
 
     api_key: str
+    api_host: str
 
     def __init__(
         self,
+        api_host,
         api_key: str = None,
-        api_host: typing.Optional[str] = None,
     ) -> None:
         self.api_key = api_key or os.environ.get("LANGTRACE_API_KEY")
-        self.api_host: str = api_host or LANGTRACE_REMOTE_URL
+        self.api_host = (
+            f"{LANGTRACE_REMOTE_URL}/api/trace"
+            if api_host == LANGTRACE_REMOTE_URL
+            else api_host
+        )
 
     def export(self, spans: typing.Sequence[ReadableSpan]) -> SpanExportResult:
         """
@@ -90,7 +95,7 @@ class LangTraceExporter(SpanExporter):
         # Send data to remote URL
         try:
             response = requests.post(
-                url=f"{self.api_host}/api/trace",
+                url=f"{self.api_host}",
                 data=json.dumps(data),
                 headers={"Content-Type": "application/json", "x-api-key": self.api_key},
                 timeout=20,

{langtrace_python_sdk-2.1.11 → langtrace_python_sdk-2.1.12}/src/langtrace_python_sdk/langtrace.py

@@ -16,6 +16,9 @@ limitations under the License.
 
 from typing import Optional
 
+from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
+    LANGTRACE_REMOTE_URL,
+)
 from langtrace_python_sdk.types import DisableInstrumentations, InstrumentationType
 from opentelemetry import trace
 from opentelemetry.sdk.trace import TracerProvider
@@ -46,6 +49,7 @@ from langtrace_python_sdk.instrumentation import (
 from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
 from colorama import Fore
 from langtrace_python_sdk.utils import check_if_sdk_is_outdated
+import os
 
 
 def init(
@@ -53,15 +57,19 @@ def init(
     batch: bool = True,
     write_spans_to_console: bool = False,
     custom_remote_exporter=None,
-    api_host: Optional[str] = None,
+    api_host: Optional[str] = LANGTRACE_REMOTE_URL,
     disable_instrumentations: Optional[DisableInstrumentations] = None,
 ):
+
+    host = (
+        os.environ.get("LANGTRACE_API_HOST", None) or api_host or LANGTRACE_REMOTE_URL
+    )
     check_if_sdk_is_outdated()
     print(Fore.GREEN + "Initializing Langtrace SDK.." + Fore.RESET)
     provider = TracerProvider(resource=Resource.create({"service.name": sys.argv[0]}))
 
     remote_write_exporter = (
-        LangTraceExporter(api_key=api_key, api_host=api_host)
+        LangTraceExporter(api_key=api_key, api_host=host)
         if custom_remote_exporter is None
         else custom_remote_exporter
     )
@@ -70,6 +78,7 @@ def init(
     simple_processor_remote = SimpleSpanProcessor(remote_write_exporter)
     simple_processor_console = SimpleSpanProcessor(console_exporter)
 
+    os.environ["LANGTRACE_API_HOST"] = host.replace("/api/trace", "")
     # Initialize tracer
     trace.set_tracer_provider(provider)
     all_instrumentations = {
@@ -101,8 +110,8 @@ def init(
         else:
             provider.add_span_processor(simple_processor_remote)
 
-    elif api_host is not None:
-        print(Fore.BLUE + f"Exporting spans to custom host: {api_host}.." + Fore.RESET)
+    elif host != LANGTRACE_REMOTE_URL:
+        print(Fore.BLUE + f"Exporting spans to custom host: {host}.." + Fore.RESET)
         if batch:
             provider.add_span_processor(batch_processor_remote)
         else:

{langtrace_python_sdk-2.1.11 → langtrace_python_sdk-2.1.12}/src/langtrace_python_sdk/utils/prompt_registry.py

@@ -1,7 +1,4 @@
 import os
-from langtrace_python_sdk.constants.exporter.langtrace_exporter import (
-    LANGTRACE_REMOTE_URL,
-)
 import requests
 from urllib.parse import urlencode
 from typing import Optional, TypedDict, Dict, List
@@ -59,7 +56,7 @@ def get_prompt_from_registry(
         print(query_params)
         # Make the GET request to the API
         response = requests.get(
-            f"{LANGTRACE_REMOTE_URL}/api/promptset?{query_string}",
+            f"{os.environ['LANGTRACE_API_HOST']}/api/promptset?{query_string}",
             headers=headers,
             timeout=None,
         )

{langtrace_python_sdk-2.1.11 → langtrace_python_sdk-2.1.12}/src/langtrace_python_sdk/utils/with_root_span.py

@@ -118,9 +118,7 @@ class SendUserFeedback:
     _langtrace_api_key: str
 
     def __init__(self):
-        self._langtrace_host = os.environ.get(
-            "LANGTRACE_API_HOST", LANGTRACE_REMOTE_URL
-        )
+        self._langtrace_host = os.environ["LANGTRACE_API_HOST"]
         self._langtrace_api_key = os.environ.get("LANGTRACE_API_KEY", None)
 
     def evaluate(self, data: EvaluationAPIData) -> None:
@@ -137,6 +135,7 @@ class SendUserFeedback:
             headers = {"x-api-key": self._langtrace_api_key}
             if evaluation is not None:
                 # Make a PUT request to update the evaluation
+                print(Fore.BLUE + "Updating Feedback.." + Fore.RESET)
                 response = requests.put(
                     f"{self._langtrace_host}/api/evaluation",
                     json=data,
@@ -147,6 +146,7 @@ class SendUserFeedback:
                 response.raise_for_status()
 
             else:
+                print(Fore.BLUE + "Sending User Feedback.." + Fore.RESET)
                 # Make a POST request to create a new evaluation
                 response = requests.post(
                     f"{self._langtrace_host}/api/evaluation",

langtrace_python_sdk-2.1.12/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-2.1.11/src/langtrace_python_sdk/version.py

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