langtrace-python-sdk 2.1.8__tar.gz → 2.1.10__tar.gz

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.8
+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.8 → langtrace_python_sdk-2.1.10}/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.8 → langtrace_python_sdk-2.1.10}/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.10/src/langtrace_python_sdk/constants/__init__.py

@@ -0,0 +1 @@
+LANGTRACE_SDK_NAME = "langtrace-python-sdk"

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/anthropic/patch.py

@@ -26,6 +26,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
     SERVICE_PROVIDERS,
 )
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def messages_create(original_method, version, tracer):
@@ -54,7 +57,7 @@ def messages_create(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": base_url,
             "llm.api": APIS["MESSAGES_CREATE"]["ENDPOINT"],
             "llm.model": kwargs.get("model"),

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/chroma/patch.py

@@ -27,6 +27,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     SERVICE_PROVIDERS,
 )
 import json
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def collection_patch(method, version, tracer):
@@ -44,7 +47,7 @@ def collection_patch(method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "vectordb",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "db.system": "chromadb",
             "db.operation": api["OPERATION"],
             "db.query": json.dumps(kwargs.get("query")),

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/cohere/patch.py

@@ -26,6 +26,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
     SERVICE_PROVIDERS,
 )
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def rerank(original_method, version, tracer):
@@ -40,7 +43,7 @@ def rerank(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": APIS["RERANK"]["URL"],
             "llm.api": APIS["RERANK"]["ENDPOINT"],
             "llm.model": kwargs.get("model"),
@@ -132,7 +135,7 @@ def embed(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": APIS["EMBED"]["URL"],
             "llm.api": APIS["EMBED"]["ENDPOINT"],
             "llm.model": kwargs.get("model"),
@@ -242,7 +245,7 @@ def chat_create(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": APIS["CHAT_CREATE"]["URL"],
             "llm.api": APIS["CHAT_CREATE"]["ENDPOINT"],
             "llm.model": (
@@ -437,7 +440,7 @@ def chat_stream(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": APIS["CHAT_STREAM"]["URL"],
             "llm.api": APIS["CHAT_STREAM"]["ENDPOINT"],
             "llm.model": (

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/groq/patch.py

@@ -27,6 +27,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
 )
 from langtrace_python_sdk.constants.instrumentation.groq import APIS
 from langtrace_python_sdk.utils.llm import calculate_prompt_tokens, estimate_tokens
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def chat_completions_create(original_method, version, tracer):
@@ -80,7 +83,7 @@ def chat_completions_create(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": base_url,
             "llm.api": APIS["CHAT_COMPLETION"]["ENDPOINT"],
             "llm.prompts": json.dumps(llm_prompts),
@@ -340,7 +343,7 @@ def async_chat_completions_create(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": base_url,
             "llm.api": APIS["CHAT_COMPLETION"]["ENDPOINT"],
             "llm.prompts": json.dumps(llm_prompts),

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/langchain/patch.py

@@ -25,6 +25,7 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
     SERVICE_PROVIDERS,
 )
+from importlib_metadata import version as v
 
 
 def generic_patch(
@@ -43,7 +44,7 @@ def generic_patch(
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "langchain.task.name": task,
             **(extra_attributes if extra_attributes is not None else {}),
         }

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/langchain_community/patch.py

@@ -25,6 +25,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
     SERVICE_PROVIDERS,
 )
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def generic_patch(
@@ -39,7 +42,7 @@ def generic_patch(
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "langchain.task.name": task,
             **(extra_attributes if extra_attributes is not None else {}),
         }

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/langchain_core/patch.py

@@ -25,6 +25,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
     SERVICE_PROVIDERS,
 )
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def generic_patch(
@@ -49,7 +52,7 @@ def generic_patch(
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "langchain.task.name": task,
             **(extra_attributes if extra_attributes is not None else {}),
         }
@@ -115,7 +118,7 @@ def runnable_patch(
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "langchain.task.name": task,
         }
 

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/langgraph/patch.py

@@ -25,6 +25,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
     SERVICE_PROVIDERS,
 )
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def patch_graph_methods(method_name, tracer, version):
@@ -37,7 +40,7 @@ def patch_graph_methods(method_name, tracer, version):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             **(extra_attributes if extra_attributes is not None else {}),
         }
 

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/llamaindex/patch.py

@@ -23,6 +23,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY,
     SERVICE_PROVIDERS,
 )
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def generic_patch(method, task, tracer, version):
@@ -38,7 +41,7 @@ def generic_patch(method, task, tracer, version):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "llamaindex.task.name": task,
             **(extra_attributes if extra_attributes is not None else {}),
         }
@@ -80,7 +83,7 @@ def async_generic_patch(method, task, tracer, version):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "llamaindex.task.name": task,
             **(extra_attributes if extra_attributes is not None else {}),
         }

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/openai/patch.py

@@ -27,6 +27,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
 )
 from langtrace_python_sdk.constants.instrumentation.openai import APIS
 from langtrace_python_sdk.utils.llm import calculate_prompt_tokens, estimate_tokens
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def images_generate(original_method, version, tracer):
@@ -48,7 +51,7 @@ def images_generate(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": base_url,
             "llm.api": APIS["IMAGES_GENERATION"]["ENDPOINT"],
             "llm.model": kwargs.get("model"),
@@ -125,7 +128,7 @@ def async_images_generate(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": base_url,
             "llm.api": APIS["IMAGES_GENERATION"]["ENDPOINT"],
             "llm.model": kwargs.get("model"),
@@ -235,7 +238,7 @@ def chat_completions_create(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": base_url,
             "llm.api": APIS["CHAT_COMPLETION"]["ENDPOINT"],
             "llm.prompts": json.dumps(llm_prompts),
@@ -495,7 +498,7 @@ def async_chat_completions_create(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": base_url,
             "llm.api": APIS["CHAT_COMPLETION"]["ENDPOINT"],
             "llm.prompts": json.dumps(llm_prompts),
@@ -725,7 +728,7 @@ def embeddings_create(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": base_url,
             "llm.api": APIS["EMBEDDINGS_CREATE"]["ENDPOINT"],
             "llm.model": kwargs.get("model"),
@@ -792,7 +795,7 @@ def async_embeddings_create(original_method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "url.full": base_url,
             "llm.api": APIS["EMBEDDINGS_CREATE"]["ENDPOINT"],
             "llm.model": kwargs.get("model"),

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/pinecone/patch.py

@@ -28,6 +28,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
 from langtrace_python_sdk.constants.instrumentation.pinecone import APIS
 from langtrace_python_sdk.utils import set_span_attribute
 from langtrace_python_sdk.utils.silently_fail import silently_fail
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def generic_patch(operation_name, version, tracer):
@@ -44,7 +47,7 @@ def generic_patch(operation_name, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "vectordb",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "db.system": "pinecone",
             "db.operation": api["OPERATION"],
             "db.query": json.dumps(kwargs.get("query")),

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/qdrant/patch.py

@@ -27,6 +27,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
     SERVICE_PROVIDERS,
 )
 from langtrace_python_sdk.constants.instrumentation.qdrant import APIS
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 def collection_patch(method, version, tracer):
@@ -44,7 +47,7 @@ def collection_patch(method, version, tracer):
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "vectordb",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "db.system": "qdrant",
             "db.operation": api["OPERATION"],
             "db.query": json.dumps(kwargs.get("query")),

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/langtrace_python_sdk/instrumentation/weaviate/patch.py

@@ -27,6 +27,9 @@ from langtrace_python_sdk.constants.instrumentation.common import (
 )
 from langtrace_python_sdk.constants.instrumentation.weaviate import APIS
 from langtrace_python_sdk.utils.misc import extract_input_params, to_iso_format
+from importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 # Predefined metadata response attributes
 METADATA_ATTRIBUTES = [
@@ -118,7 +121,7 @@ def create_traced_method(method_name, version, tracer, get_collection_name=None)
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "vectordb",
             "langtrace.service.version": version,
-            "langtrace.version": "1.0.0",
+            "langtrace.version": v(LANGTRACE_SDK_NAME),
             "db.system": "weaviate",
             "db.operation": api["OPERATION"],
             "db.collection.name": collection_name,

langtrace_python_sdk-2.1.10/src/langtrace_python_sdk/utils/sdk_version_checker.py

@@ -0,0 +1,53 @@
+from importlib.metadata import version
+import time
+import requests
+from colorama import Fore
+
+
+class SDKVersionChecker:
+    _cache: None
+    _cache_duration: int
+    _current_version: str
+    _latest_version: str
+
+    def __init__(self):
+        self._cache = {"timestamp": 0, "latest_version": None}
+        self._cache_duration = 3600  # Cache for 1 hour
+        self._current_version = version("langtrace_python_sdk")
+
+    def fetch_latest(self):
+        try:
+            current_time = time.time()
+            if (
+                current_time - self._cache["timestamp"] < self._cache_duration
+                and self._cache["latest_version"]
+            ):
+                return self._cache["latest_version"]
+
+            response = requests.get(
+                "https://api.github.com/repos/Scale3-Labs/langtrace-python-sdk/releases/latest",
+                timeout=20,
+            )
+            response.raise_for_status()
+            latest_version = response.json()["tag_name"]
+            self._cache.update(
+                {"timestamp": current_time, "latest_version": latest_version}
+            )
+            self._latest_version = latest_version
+            return latest_version
+        except Exception as err:
+            return None
+
+    def is_outdated(self):
+        latest_version = self.fetch_latest()
+        if latest_version:
+            return self._current_version < latest_version
+        return False
+
+    def check(self):
+        if self.is_outdated():
+            print(
+                Fore.YELLOW
+                + f"Warning: Your Langtrace SDK version {self._current_version} is outdated. Please upgrade to {self._latest_version}."
+                + Fore.RESET
+            )

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/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.10/src/langtrace_python_sdk/version.py

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

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/tests/anthropic/test_anthropic.py

@@ -2,7 +2,9 @@ import pytest
 import json
 import importlib
 from langtrace_python_sdk.constants.instrumentation.anthropic import APIS
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 from tests.utils import assert_response_format, assert_token_count
+from importlib_metadata import version as v
 
 
 @pytest.mark.vcr()
@@ -30,7 +32,7 @@ def test_anthropic(anthropic_client, exporter):
     assert attributes.get("langtrace.service.version") == importlib.metadata.version(
         "anthropic"
     )
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.version") == v(LANGTRACE_SDK_NAME)
     assert attributes.get("url.full") == "https://api.anthropic.com"
     assert attributes.get("llm.api") == APIS["MESSAGES_CREATE"]["ENDPOINT"]
     assert attributes.get("llm.model") == llm_model_value
@@ -72,7 +74,7 @@ def test_anthropic_streaming(anthropic_client, exporter):
     assert attributes.get("langtrace.service.version") == importlib.metadata.version(
         "anthropic"
     )
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.version") == v(LANGTRACE_SDK_NAME)
     assert attributes.get("url.full") == "https://api.anthropic.com"
     assert attributes.get("llm.api") == APIS["MESSAGES_CREATE"]["ENDPOINT"]
     assert attributes.get("llm.model") == llm_model_value

{langtrace_python_sdk-2.1.8 → langtrace_python_sdk-2.1.10}/src/tests/cohere/test_cohere_chat.py

@@ -3,7 +3,9 @@ from langtrace_python_sdk.constants.instrumentation.cohere import APIS
 from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
 import pytest
 import importlib
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 from tests.utils import assert_response_format, assert_token_count
+from importlib_metadata import version as v
 
 
 @pytest.mark.vcr
@@ -39,7 +41,7 @@ def test_cohere_chat(cohere_client, exporter):
         "cohere"
     )
 
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.version") == v(LANGTRACE_SDK_NAME)
     assert attributes.get("url.full") == APIS["CHAT_CREATE"]["URL"]
     assert attributes.get("llm.api") == APIS["CHAT_CREATE"]["ENDPOINT"]
     assert attributes.get("llm.model") == llm_model_value
@@ -97,7 +99,7 @@ def test_cohere_chat_streaming(cohere_client, exporter):
         "cohere"
     )
 
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.version") == v(LANGTRACE_SDK_NAME)
     assert attributes.get("url.full") == APIS["CHAT_STREAM"]["URL"]
     assert attributes.get("llm.api") == APIS["CHAT_STREAM"]["ENDPOINT"]
     assert attributes.get("llm.model") == llm_model_value