langtrace-python-sdk 2.1.9__py3-none-any.whl → 2.1.10__py3-none-any.whl

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/__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/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/version.py

@@ -1 +1 @@
-__version__ = "2.1.9"
+__version__ = "2.1.10"

{langtrace_python_sdk-2.1.9.dist-info → langtrace_python_sdk-2.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.9
+Version: 2.1.10
 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.dist-info → langtrace_python_sdk-2.1.10.dist-info}/RECORD

@@ -32,13 +32,13 @@ examples/openai_example/tool_calling_nonstreaming.py,sha256=Yc848IooZRXNynHL6z0k
 examples/openai_example/tool_calling_streaming.py,sha256=mV1RbyAoVhumGRPpqPWQ6PMhnJyeifrlELd2-K1qJ_w,7015
 examples/perplexity_example/basic.py,sha256=bp7n27gaugJkaFVyt8pjaEfi66lYcqP6eFFjPewUShY,668
 examples/pinecone_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/pinecone_example/basic.py,sha256=b-5-cSf3_N6Gk4zGVecGF47mBcFY9ZrC5ztrIZVgTQo,1430
+examples/pinecone_example/basic.py,sha256=gkb1uQwKFLnWmRuM2KGvAeTpT1AYlHmz2kRey5_-Dqs,1437
 examples/qdrant_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 examples/qdrant_example/basic.py,sha256=DCMjHSuBZKkhEjCkwy5d5La9WMyW0lCWqtcZWiFCEm4,1425
 examples/weaviate_example/query_text.py,sha256=iE9OiHsibjsprbCGzabE03eZsGN06e6ym2iS1A9P3ig,64650
-langtrace_python_sdk/__init__.py,sha256=R4dv73QARUw4WRfCdSKkKzZTLDumF9jcSdwtVrzWhb4,962
+langtrace_python_sdk/__init__.py,sha256=FuvyRuStRe_N2wo2SB2_ZQ0w7LGNIjV0lLi6S1IgGwY,958
 langtrace_python_sdk/langtrace.py,sha256=_qJsbLo-4GqhXd8apg3W7A_l3ZnFYEX1wOA-_rg-TZA,6655
-langtrace_python_sdk/version.py,sha256=Gf0zUMmdRLkPmvwnoP7oCpWn7Ep93jlNIlBD6nifugI,22
+langtrace_python_sdk/version.py,sha256=zBzBCTHx_lpM20-2wHeccRHskIUls_oSQ1tzAOiGGrU,23
 langtrace_python_sdk/constants/__init__.py,sha256=P8QvYwt5czUNDZsKS64vxm9Dc41ptGbuF1TFtAF6nv4,44
 langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=5MNjnAOg-4am78J3gVMH6FSwq5N8TOj72ugkhsw4vi0,46
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -101,7 +101,7 @@ langtrace_python_sdk/utils/prompt_registry.py,sha256=7FFB4Pj0414qgf02h5zL5vXBZgN
 langtrace_python_sdk/utils/sdk_version_checker.py,sha256=FzjIWZjn53cX0LEVPdipQd1fO9lG8iGVUEVUs9Hyk6M,1713
 langtrace_python_sdk/utils/silently_fail.py,sha256=F_9EteXCO9Cyq-8MA1OT2Zy_dx8n06nt31I7t7ui24E,478
 langtrace_python_sdk/utils/types.py,sha256=l-N6o7cnWUyrD6dBvW7W3Pf5CkPo5QaoT__k1XLbrQg,383
-langtrace_python_sdk/utils/with_root_span.py,sha256=rSPrTnKndJo6cFUWL4907IkJFbtLwqMS87paADUxD6A,5957
+langtrace_python_sdk/utils/with_root_span.py,sha256=wORIk9N4BOWEP9zoDMXpD3qX4r6P6sDMUn9bzl8BbvM,6754
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/conftest.py,sha256=0Jo6iCZTXbdvyJVhG9UpYGkLabL75378oauCzmt-Sa8,603
 tests/utils.py,sha256=hP8sTH-M8WO6zlLkSFHPf6483ZcKEcnxul6JIIb1pLM,1396
@@ -138,7 +138,7 @@ tests/pinecone/cassettes/test_query.yaml,sha256=b5v9G3ssUy00oG63PlFUR3JErF2Js-5A
 tests/pinecone/cassettes/test_upsert.yaml,sha256=neWmQ1v3d03V8WoLl8FoFeeCYImb8pxlJBWnFd_lITU,38607
 tests/qdrant/conftest.py,sha256=9n0uHxxIjWk9fbYc4bx-uP8lSAgLBVx-cV9UjnsyCHM,381
 tests/qdrant/test_qdrant.py,sha256=pzjAjVY2kmsmGfrI2Gs2xrolfuaNHz7l1fqGQCjp5_o,3353
-langtrace_python_sdk-2.1.9.dist-info/METADATA,sha256=z55Vy4oeKQROK5MlKoeYpJfzHxqMvhlfEXjr2PSXAZc,11859
-langtrace_python_sdk-2.1.9.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
-langtrace_python_sdk-2.1.9.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-2.1.9.dist-info/RECORD,,
+langtrace_python_sdk-2.1.10.dist-info/METADATA,sha256=2k5jRe5pbkZ1IJ6iTSx67R_vg6_97O4_W9eSr5_f8Zs,11860
+langtrace_python_sdk-2.1.10.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+langtrace_python_sdk-2.1.10.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-2.1.10.dist-info/RECORD,,