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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.7
+Version: 2.1.9
 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/src/langtrace_python_sdk/constants/__init__.py

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

{langtrace_python_sdk-2.1.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/src/langtrace_python_sdk/langtrace.py

@@ -45,6 +45,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
 
 
 def init(
@@ -55,6 +56,7 @@ def init(
     api_host: Optional[str] = None,
     disable_instrumentations: Optional[DisableInstrumentations] = None,
 ):
+    check_if_sdk_is_outdated()
     print(Fore.GREEN + "Initializing Langtrace SDK.." + Fore.RESET)
     provider = TracerProvider(resource=Resource.create({"service.name": sys.argv[0]}))
 

{langtrace_python_sdk-2.1.7 → langtrace_python_sdk-2.1.9}/src/langtrace_python_sdk/utils/__init__.py

@@ -1,5 +1,13 @@
+from .sdk_version_checker import SDKVersionChecker
+
+
 def set_span_attribute(span, name, value):
     if value is not None:
         if value != "":
             span.set_attribute(name, value)
     return
+
+
+def check_if_sdk_is_outdated():
+    SDKVersionChecker().check()
+    return

langtrace_python_sdk-2.1.9/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.9/src/langtrace_python_sdk/version.py

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

{langtrace_python_sdk-2.1.7 → langtrace_python_sdk-2.1.9}/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.7 → langtrace_python_sdk-2.1.9}/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

{langtrace_python_sdk-2.1.7 → langtrace_python_sdk-2.1.9}/src/tests/cohere/test_cohere_embed.py

@@ -1,7 +1,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 importlib_metadata import version as v
+
+from langtrace_python_sdk.constants import LANGTRACE_SDK_NAME
 
 
 @pytest.mark.vcr
@@ -25,11 +27,9 @@ def test_cohere_embed(cohere_client, exporter):
     assert attributes.get("langtrace.sdk.name") == "langtrace-python-sdk"
     assert attributes.get("langtrace.service.name") == SERVICE_PROVIDERS["COHERE"]
     assert attributes.get("langtrace.service.type") == "llm"
-    assert attributes.get("langtrace.service.version") == importlib.metadata.version(
-        "cohere"
-    )
+    assert attributes.get("langtrace.service.version") == v("cohere")
 
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.version") == v(LANGTRACE_SDK_NAME)
     assert attributes.get("url.full") == APIS["EMBED"]["URL"]
     assert attributes.get("llm.api") == APIS["EMBED"]["ENDPOINT"]
     assert attributes.get("llm.model") == llm_model_value

{langtrace_python_sdk-2.1.7 → langtrace_python_sdk-2.1.9}/src/tests/cohere/test_cohere_rerank.py

@@ -1,10 +1,9 @@
 from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
 import pytest
 import json
-import importlib
 from langtrace_python_sdk.constants.instrumentation.cohere import APIS
-from tests.utils import assert_response_format, assert_token_count
-import pprint
+from tests.utils import assert_token_count
+from importlib_metadata import version as v
 
 
 @pytest.mark.vcr
@@ -34,11 +33,9 @@ def test_cohere_rerank(cohere_client, exporter):
     assert attributes.get("langtrace.sdk.name") == "langtrace-python-sdk"
     assert attributes.get("langtrace.service.name") == SERVICE_PROVIDERS["COHERE"]
     assert attributes.get("langtrace.service.type") == "llm"
-    assert attributes.get("langtrace.service.version") == importlib.metadata.version(
-        "cohere"
-    )
+    assert attributes.get("langtrace.service.version") == v("cohere")
 
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.version") == v("langtrace-python-sdk")
     assert attributes.get("url.full") == APIS["RERANK"]["URL"]
     assert attributes.get("llm.api") == APIS["RERANK"]["ENDPOINT"]
     assert attributes.get("llm.model") == llm_model_value

{langtrace_python_sdk-2.1.7 → langtrace_python_sdk-2.1.9}/src/tests/openai/test_chat_completion.py

@@ -1,8 +1,8 @@
 import pytest
-import importlib
 import json
 from langtrace_python_sdk.constants.instrumentation.openai import APIS
 from tests.utils import assert_response_format, assert_token_count
+from importlib_metadata import version as v
 
 
 @pytest.mark.vcr()
@@ -25,10 +25,8 @@ def test_chat_completion(exporter, openai_client):
     assert attributes.get("langtrace.sdk.name") == "langtrace-python-sdk"
     assert attributes.get("langtrace.service.name") == "OpenAI"
     assert attributes.get("langtrace.service.type") == "llm"
-    assert attributes.get("langtrace.service.version") == importlib.metadata.version(
-        "openai"
-    )
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.service.version") == v("openai")
+    assert attributes.get("langtrace.version") == v("langtrace-python-sdk")
     assert attributes.get("url.full") == "https://api.openai.com/v1/"
     assert attributes.get("llm.api") == APIS["CHAT_COMPLETION"]["ENDPOINT"]
     assert attributes.get("llm.model") == "gpt-4-0613"
@@ -78,10 +76,8 @@ def test_chat_completion_streaming(exporter, openai_client):
     assert attributes.get("langtrace.sdk.name") == "langtrace-python-sdk"
     assert attributes.get("langtrace.service.name") == "OpenAI"
     assert attributes.get("langtrace.service.type") == "llm"
-    assert attributes.get("langtrace.service.version") == importlib.metadata.version(
-        "openai"
-    )
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.service.version") == v("openai")
+    assert attributes.get("langtrace.version") == v("langtrace-python-sdk")
     assert attributes.get("url.full") == "https://api.openai.com/v1/"
     assert attributes.get("llm.api") == APIS["CHAT_COMPLETION"]["ENDPOINT"]
     assert attributes.get("llm.model") == "gpt-4-0613"
@@ -135,10 +131,8 @@ async def test_async_chat_completion_streaming(exporter, async_openai_client):
     assert attributes.get("langtrace.sdk.name") == "langtrace-python-sdk"
     assert attributes.get("langtrace.service.name") == "OpenAI"
     assert attributes.get("langtrace.service.type") == "llm"
-    assert attributes.get("langtrace.service.version") == importlib.metadata.version(
-        "openai"
-    )
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.service.version") == v("openai")
+    assert attributes.get("langtrace.version") == v("langtrace-python-sdk")
     assert attributes.get("url.full") == "https://api.openai.com/v1/"
     assert attributes.get("llm.api") == APIS["CHAT_COMPLETION"]["ENDPOINT"]
     assert attributes.get("llm.model") == "gpt-4-0613"

{langtrace_python_sdk-2.1.7 → langtrace_python_sdk-2.1.9}/src/tests/openai/test_image_generation.py

@@ -1,7 +1,7 @@
 import pytest
 import json
-import importlib
 from langtrace_python_sdk.constants.instrumentation.openai import APIS
+from importlib_metadata import version as v
 
 
 @pytest.mark.vcr()
@@ -23,10 +23,8 @@ def test_image_generation(openai_client, exporter):
     assert attributes.get("langtrace.sdk.name") == "langtrace-python-sdk"
     assert attributes.get("langtrace.service.name") == "OpenAI"
     assert attributes.get("langtrace.service.type") == "llm"
-    assert attributes.get("langtrace.service.version") == importlib.metadata.version(
-        "openai"
-    )
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.service.version") == v("openai")
+    assert attributes.get("langtrace.version") == v("langtrace-python-sdk")
     assert attributes.get("url.full") == "https://api.openai.com/v1/"
     assert attributes.get("llm.api") == APIS["IMAGES_GENERATION"]["ENDPOINT"]
     assert attributes.get("llm.model") == llm_model_value
@@ -66,10 +64,8 @@ async def test_async_image_generation(async_openai_client, exporter):
     assert attributes.get("langtrace.sdk.name") == "langtrace-python-sdk"
     assert attributes.get("langtrace.service.name") == "OpenAI"
     assert attributes.get("langtrace.service.type") == "llm"
-    assert attributes.get("langtrace.service.version") == importlib.metadata.version(
-        "openai"
-    )
-    assert attributes.get("langtrace.version") == "1.0.0"
+    assert attributes.get("langtrace.service.version") == v("openai")
+    assert attributes.get("langtrace.version") == v("langtrace-python-sdk")
     assert attributes.get("url.full") == "https://api.openai.com/v1/"
     assert attributes.get("llm.api") == APIS["IMAGES_GENERATION"]["ENDPOINT"]
     assert attributes.get("llm.model") == llm_model_value