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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.9
+Version: 2.1.11
 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.9 → langtrace_python_sdk-2.1.11}/src/examples/pinecone_example/basic.py

@@ -7,7 +7,7 @@ from openai import OpenAI
 from pinecone import Pinecone, ServerlessSpec
 
 from langtrace_python_sdk import langtrace, with_langtrace_root_span
-from langtrace_python_sdk.utils.with_root_span import send_user_feedback
+from langtrace_python_sdk.utils.with_root_span import SendUserFeedback
 
 _ = load_dotenv(find_dotenv())
 
@@ -48,7 +48,7 @@ def basic(span_id=None, trace_id=None):
     resp = index.query(
         vector=embedding, top_k=1, include_values=False, namespace="test-namespace"
     )
-    send_user_feedback(
+    SendUserFeedback().evaluate(
         {"spanId": span_id, "traceId": trace_id, "userScore": 1, "userId": "123"}
     )
     return [res, resp]

{langtrace_python_sdk-2.1.9 → langtrace_python_sdk-2.1.11}/src/langtrace_python_sdk/__init__.py

@@ -17,11 +17,11 @@ limitations under the License.
 from langtrace_python_sdk import langtrace
 from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 from langtrace_python_sdk.utils.prompt_registry import get_prompt_from_registry
-from langtrace_python_sdk.utils.with_root_span import send_user_feedback
+from langtrace_python_sdk.utils.with_root_span import SendUserFeedback
 
 __all__ = [
     "langtrace",
     "with_langtrace_root_span",
     "get_prompt_from_registry",
-    "send_user_feedback",
+    "SendUserFeedback",
 ]

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

@@ -70,30 +70,8 @@ def init(
     simple_processor_remote = SimpleSpanProcessor(remote_write_exporter)
     simple_processor_console = SimpleSpanProcessor(console_exporter)
 
-    if write_spans_to_console:
-        print(Fore.BLUE + "Writing spans to console" + Fore.RESET)
-        provider.add_span_processor(simple_processor_console)
-
-    elif custom_remote_exporter is not None:
-        print(Fore.BLUE + "Exporting spans to custom remote exporter.." + Fore.RESET)
-        if batch:
-            provider.add_span_processor(batch_processor_remote)
-        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)
-        if batch:
-            provider.add_span_processor(batch_processor_remote)
-        else:
-            provider.add_span_processor(simple_processor_remote)
-    else:
-        print(Fore.BLUE + "Exporting spans to Langtrace cloud.." + Fore.RESET)
-        provider.add_span_processor(batch_processor_remote)
-
     # Initialize tracer
     trace.set_tracer_provider(provider)
-
     all_instrumentations = {
         "openai": OpenAIInstrumentation(),
         "groq": GroqInstrumentation(),
@@ -112,13 +90,33 @@ def init(
     }
 
     init_instrumentations(disable_instrumentations, all_instrumentations)
+    if write_spans_to_console:
+        print(Fore.BLUE + "Writing spans to console" + Fore.RESET)
+        provider.add_span_processor(simple_processor_console)
+
+    elif custom_remote_exporter is not None:
+        print(Fore.BLUE + "Exporting spans to custom remote exporter.." + Fore.RESET)
+        if batch:
+            provider.add_span_processor(batch_processor_remote)
+        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)
+        if batch:
+            provider.add_span_processor(batch_processor_remote)
+        else:
+            provider.add_span_processor(simple_processor_remote)
+    else:
+        print(Fore.BLUE + "Exporting spans to Langtrace cloud.." + Fore.RESET)
+        provider.add_span_processor(batch_processor_remote)
 
 
 def init_instrumentations(
     disable_instrumentations: DisableInstrumentations, all_instrumentations: dict
 ):
     if disable_instrumentations is None:
-        for _, v in all_instrumentations.items():
+        for idx, (name, v) in enumerate(all_instrumentations.items()):
             v.instrument()
     else:
 

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

@@ -34,6 +34,7 @@ from langtrace_python_sdk.utils.types import (
     LangTraceApiError,
     LangTraceEvaluation,
 )
+from colorama import Fore
 
 
 def with_langtrace_root_span(
@@ -112,63 +113,78 @@ def with_additional_attributes(attributes={}):
     return decorator
 
 
-def send_user_feedback(data: EvaluationAPIData) -> None:
-    try:
-        evaluation = _get_evaluation(data["spanId"])
-        headers = {"x-api-key": os.environ["LANGTRACE_API_KEY"]}
-        if evaluation is not None:
-            # Make a PUT request to update the evaluation
-            response = requests.put(
-                f"{LANGTRACE_REMOTE_URL}/api/evaluation",
-                json=data,
-                params={"spanId": data["spanId"]},
-                headers=headers,
-                timeout=None,
-            )
-            response.raise_for_status()
+class SendUserFeedback:
+    _langtrace_host: str
+    _langtrace_api_key: str
 
-        else:
-            # Make a POST request to create a new evaluation
-            response = requests.post(
-                f"{LANGTRACE_REMOTE_URL}/api/evaluation",
-                json=data,
-                params={"spanId": data["spanId"]},
-                headers=headers,
+    def __init__(self):
+        self._langtrace_host = os.environ.get(
+            "LANGTRACE_API_HOST", LANGTRACE_REMOTE_URL
+        )
+        self._langtrace_api_key = os.environ.get("LANGTRACE_API_KEY", None)
+
+    def evaluate(self, data: EvaluationAPIData) -> None:
+        try:
+            if self._langtrace_api_key is None:
+                print(Fore.RED)
+                print(
+                    f"Missing Langtrace API key, proceed to {self._langtrace_host} to create one"
+                )
+                print("Set the API key as an environment variable LANGTRACE_API_KEY")
+                print(Fore.RESET)
+                return
+            evaluation = self.get_evaluation(data["spanId"])
+            headers = {"x-api-key": self._langtrace_api_key}
+            if evaluation is not None:
+                # Make a PUT request to update the evaluation
+                response = requests.put(
+                    f"{self._langtrace_host}/api/evaluation",
+                    json=data,
+                    params={"spanId": data["spanId"]},
+                    headers=headers,
+                    timeout=None,
+                )
+                response.raise_for_status()
+
+            else:
+                # Make a POST request to create a new evaluation
+                response = requests.post(
+                    f"{self._langtrace_host}/api/evaluation",
+                    json=data,
+                    params={"spanId": data["spanId"]},
+                    headers=headers,
+                    timeout=None,
+                )
+                response.raise_for_status()
+
+        except requests.RequestException as err:
+            if err.response is not None:
+                message = (
+                    err.response.json().get("message", "")
+                    if err.response.json().get("message", "")
+                    else err.response.text if err.response.text else str(err)
+                )
+                raise LangTraceApiError(message, err.response.status_code)
+            raise LangTraceApiError(str(err), 500)
+
+    def get_evaluation(self, span_id: str) -> Optional[LangTraceEvaluation]:
+        try:
+            response = requests.get(
+                f"{self._langtrace_host}/api/evaluation",
+                params={"spanId": span_id},
+                headers={"x-api-key": self._langtrace_api_key},
                 timeout=None,
             )
+            evaluations = response.json().get("evaluations", [])
             response.raise_for_status()
-
-    except requests.RequestException as err:
-        # Handle specific HTTP errors or general request exceptions
-        error_msg = str(err)
-        if err.response:
-            try:
-                # Try to extract server-provided error message
-                error_msg = err.response.json().get("error", error_msg)
-            except ValueError:
-                # Fallback if response is not JSON
-                error_msg = err.response.text
-        raise Exception(f"API error: {error_msg}") from err
-
-
-def _get_evaluation(span_id: str) -> Optional[LangTraceEvaluation]:
-    try:
-        response = requests.get(
-            f"{LANGTRACE_REMOTE_URL}/api/evaluation",
-            params={"spanId": span_id},
-            headers={"x-api-key": os.environ["LANGTRACE_API_KEY"]},
-            timeout=None,
-        )
-        evaluations = response.json().get("evaluations", [])
-        response.raise_for_status()
-        return None if not evaluations else evaluations[0]
-
-    except requests.RequestException as err:
-        if err.response is not None:
-            message = (
-                err.response.json().get("message", "")
-                if err.response.json().get("message", "")
-                else err.response.text if err.response.text else str(err)
-            )
-            raise LangTraceApiError(message, err.response.status_code)
-        raise LangTraceApiError(str(err), 500)
+            return None if not evaluations else evaluations[0]
+
+        except requests.RequestException as err:
+            if err.response is not None:
+                message = (
+                    err.response.json().get("message", "")
+                    if err.response.json().get("message", "")
+                    else err.response.text if err.response.text else str(err)
+                )
+                raise LangTraceApiError(message, err.response.status_code)
+            raise LangTraceApiError(str(err), 500)

langtrace_python_sdk-2.1.11/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-2.1.9/src/langtrace_python_sdk/version.py

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