langtrace-python-sdk 1.2.14__tar.gz → 1.2.17__tar.gz

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/PKG-INFO

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

@@ -90,10 +90,8 @@ class LangTraceExporter(SpanExporter):
                 data=json.dumps(data),
                 headers={"Content-Type": "application/json", "x-api-key": self.api_key},
             )
-            print(f"Traces sent To {LANGTRACE_REMOTE_URL}")
             return SpanExportResult.SUCCESS
         except Exception as e:
-            print("Error sending data to remote URL", e)
             return SpanExportResult.FAILURE
 
     def shutdown(self) -> None:

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/anthropic/instrumentation.py

@@ -11,6 +11,10 @@ from wrapt import wrap_function_wrapper
 
 from langtrace_python_sdk.instrumentation.anthropic.patch import messages_create
 
+import logging
+
+logging.basicConfig(level=logging.FATAL)
+
 
 class AnthropicInstrumentation(BaseInstrumentor):
     """

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/anthropic/patch.py

@@ -35,6 +35,7 @@ def messages_create(original_method, version, tracer):
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/chroma/instrumentation.py

@@ -11,6 +11,9 @@ from wrapt import wrap_function_wrapper
 
 from langtrace_python_sdk.constants.instrumentation.chroma import APIS
 from langtrace_python_sdk.instrumentation.chroma.patch import collection_patch
+import logging
+
+logging.basicConfig(level=logging.FATAL)
 
 
 class ChromaInstrumentation(BaseInstrumentor):

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/chroma/patch.py

@@ -23,6 +23,7 @@ def collection_patch(method, version, tracer):
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "vectordb",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/langchain/instrumentation.py

@@ -12,6 +12,10 @@ from wrapt import wrap_function_wrapper
 
 from langtrace_python_sdk.instrumentation.langchain.patch import generic_patch
 
+import logging
+
+logging.basicConfig(level=logging.FATAL)
+
 
 def patch_module_classes(
     module_name, tracer, version, task, trace_output=True, trace_input=True

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/langchain/patch.py

@@ -25,6 +25,7 @@ def generic_patch(
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/langchain_community/patch.py

@@ -17,6 +17,7 @@ def generic_patch(
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/langchain_core/patch.py

@@ -31,6 +31,7 @@ def generic_patch(
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
@@ -96,6 +97,7 @@ def runnable_patch(
     def traced_method(wrapped, instance, args, kwargs):
         service_provider = SERVICE_PROVIDERS["LANGCHAIN_CORE"]
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py

@@ -12,6 +12,10 @@ from wrapt import wrap_function_wrapper
 
 from langtrace_python_sdk.instrumentation.llamaindex.patch import generic_patch
 
+import logging
+
+logging.basicConfig(level=logging.FATAL)
+
 
 class LlamaindexInstrumentation(BaseInstrumentor):
     """

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/llamaindex/patch.py

@@ -20,6 +20,7 @@ def generic_patch(method, task, tracer, version):
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/openai/instrumentation.py

@@ -1,7 +1,6 @@
 import importlib.metadata
 from typing import Collection
 
-import openai
 from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
 from opentelemetry.trace import get_tracer
 from wrapt import wrap_function_wrapper
@@ -12,6 +11,10 @@ from langtrace_python_sdk.instrumentation.openai.patch import (
     images_generate,
 )
 
+import logging
+
+logging.basicConfig(level=logging.FATAL)
+
 
 class OpenAIInstrumentation(BaseInstrumentor):
 

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/openai/patch.py

@@ -30,6 +30,7 @@ def images_generate(original_method, version, tracer):
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
@@ -99,6 +100,7 @@ def chat_completions_create(original_method, version, tracer):
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
@@ -133,7 +135,7 @@ def chat_completions_create(original_method, version, tracer):
         try:
             # Attempt to call the original method
             result = wrapped(*args, **kwargs)
-            if kwargs.get("stream") is False:
+            if kwargs.get("stream") is False or kwargs.get("stream") is None:
                 span.set_attribute("llm.model", result.model)
                 if hasattr(result, "choices") and result.choices is not None:
                     responses = [
@@ -313,6 +315,7 @@ def embeddings_create(original_method, version, tracer):
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,
@@ -338,8 +341,6 @@ def embeddings_create(original_method, version, tracer):
             APIS["EMBEDDINGS_CREATE"]["METHOD"], kind=SpanKind.CLIENT
         ) as span:
 
-            print("Inside embeddings_create", trace.get_current_span())
-
             for field, value in attributes.model_dump(by_alias=True).items():
                 if value is not None:
                     span.set_attribute(field, value)

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/pinecone/instrumentation.py

@@ -14,6 +14,10 @@ from wrapt import wrap_function_wrapper
 from langtrace_python_sdk.constants.instrumentation.pinecone import APIS
 from langtrace_python_sdk.instrumentation.pinecone.patch import generic_patch
 
+import logging
+
+logging.basicConfig(level=logging.FATAL)
+
 
 class PineconeInstrumentation(BaseInstrumentor):
     """

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/langtrace_python_sdk/instrumentation/pinecone/patch.py

@@ -21,6 +21,7 @@ def generic_patch(original_method, method, version, tracer):
         extra_attributes = baggage.get_baggage(LANGTRACE_ADDITIONAL_SPAN_ATTRIBUTES_KEY)
 
         span_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "vectordb",
             "langtrace.service.version": version,

langtrace_python_sdk-1.2.17/src/langtrace_python_sdk/version.py

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

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/tests/anthropic/test_anthropic.py

@@ -46,6 +46,7 @@ class TestAnthropic(unittest.TestCase):
         self.assertTrue(self.span.set_status.has_calls([call(Status(StatusCode.OK))]))
         
         expected_attributes = {
+            "langtrace.sdk.name": "langtrace-python-sdk",
             "langtrace.service.name": "Anthropic",
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/tests/chroma/test_chroma.py

@@ -39,6 +39,7 @@ class TestChromaPatch(unittest.TestCase):
 
         # Verify span attributes are set as expected
         expected_attributes = {
+            'langtrace.sdk.name': 'langtrace-python-sdk',
             'langtrace.service.name': 'Chroma',
             'langtrace.service.type': 'vectordb',
             'langtrace.service.version': '1.2.3',

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/tests/langchain/test_langchain.py

@@ -38,6 +38,7 @@ class TestGenericPatch(unittest.TestCase):
         
         service_provider = "Langchain"
         expected_attributes = {
+            'langtrace.sdk.name': 'langtrace-python-sdk',
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/tests/langchain/test_langchain_community.py

@@ -38,6 +38,7 @@ class TestGenericPatch(unittest.TestCase):
         
         service_provider = "Langchain Community"
         expected_attributes = {
+            'langtrace.sdk.name': 'langtrace-python-sdk',
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/tests/langchain/test_langchain_core.py

@@ -38,6 +38,7 @@ class TestGenericPatch(unittest.TestCase):
         
         service_provider = "Langchain Core"
         expected_attributes = {
+            'langtrace.sdk.name': 'langtrace-python-sdk',
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,
@@ -82,6 +83,7 @@ class TestGenericPatch(unittest.TestCase):
         
         service_provider = "Langchain Core"
         expected_attributes = {
+            'langtrace.sdk.name': 'langtrace-python-sdk',
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/tests/openai/test_chat_completion.py

@@ -63,6 +63,7 @@ class TestChatCompletion(unittest.TestCase):
         self.assertTrue(self.tracer.start_as_current_span.called_once_with("openai.chat.completions.create", kind=SpanKind.CLIENT))
 
         expected_attributes = {
+            'langtrace.sdk.name': 'langtrace-python-sdk',
             "langtrace.service.name": "OpenAI",
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/tests/openai/test_image_generation.py

@@ -51,6 +51,7 @@ class TestImageGeneration(unittest.TestCase):
         self.assertTrue(self.tracer.start_as_current_span.called_once_with("openai.images.generate", kind=SpanKind.CLIENT))
 
         expected_attributes = {
+            'langtrace.sdk.name': 'langtrace-python-sdk',
             "langtrace.service.name": "OpenAI",
             "langtrace.service.type": "llm",
             "langtrace.service.version": version,

{langtrace_python_sdk-1.2.14 → langtrace_python_sdk-1.2.17}/src/tests/pinecone/test_pinecone.py

@@ -42,6 +42,7 @@ class TestPinecone(unittest.TestCase):
         api = APIS[method]
         service_provider = SERVICE_PROVIDERS["PINECONE"]
         expected_attributes = {
+            'langtrace.sdk.name': 'langtrace-python-sdk',
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "vectordb",
             "langtrace.service.version": version,

langtrace_python_sdk-1.2.14/src/langtrace_python_sdk/version.py

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