Release (#61)

* Updated test_chat_completion and added test_anthropic. Updated tests/utils * Updated tests/utils * updated langchain test cases * updated test_langchain * updated test_image_generation * Bugfix * Bump version * Cleanup logging (#56) * Fix streaming bug (#57) * Cleanup logging * Fix streaming bug * bump version * add sdk name (#60) --------- Co-authored-by: Ali Waleed <[email protected]> Co-authored-by: MayuriS24 <[email protected]> Co-authored-by: Rohit Kadhe <[email protected]>
Scale3-Labs · Apr 4, 2024 · 6ad0a89 · 6ad0a89
1 parent 0b3a74c
commit 6ad0a89
Show file tree

Hide file tree

Showing 16 changed files with 20 additions and 0 deletions.
diff --git a/src/langtrace_python_sdk/instrumentation/anthropic/patch.py b/src/langtrace_python_sdk/instrumentation/anthropic/patch.py
@@ -35,6 +35,7 @@ def traced_method(wrapped, instance, args, kwargs):
         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,

diff --git a/src/langtrace_python_sdk/instrumentation/chroma/patch.py b/src/langtrace_python_sdk/instrumentation/chroma/patch.py
@@ -23,6 +23,7 @@ def traced_method(wrapped, instance, args, kwargs):
         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,

diff --git a/src/langtrace_python_sdk/instrumentation/langchain/patch.py b/src/langtrace_python_sdk/instrumentation/langchain/patch.py
@@ -25,6 +25,7 @@ def traced_method(wrapped, instance, args, kwargs):
         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,

diff --git a/src/langtrace_python_sdk/instrumentation/langchain_community/patch.py b/src/langtrace_python_sdk/instrumentation/langchain_community/patch.py
@@ -17,6 +17,7 @@ def traced_method(wrapped, instance, args, kwargs):
         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,

diff --git a/src/langtrace_python_sdk/instrumentation/langchain_core/patch.py b/src/langtrace_python_sdk/instrumentation/langchain_core/patch.py
@@ -31,6 +31,7 @@ def traced_method(wrapped, instance, args, kwargs):
         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,

diff --git a/src/langtrace_python_sdk/instrumentation/llamaindex/patch.py b/src/langtrace_python_sdk/instrumentation/llamaindex/patch.py
@@ -20,6 +20,7 @@ def traced_method(wrapped, instance, args, kwargs):
         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,

diff --git a/src/langtrace_python_sdk/instrumentation/openai/patch.py b/src/langtrace_python_sdk/instrumentation/openai/patch.py
@@ -30,6 +30,7 @@ def traced_method(wrapped, instance, args, kwargs):
         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 traced_method(wrapped, instance, args, kwargs):
         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,
@@ -313,6 +315,7 @@ def traced_method(wrapped, instance, args, kwargs):
         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,

diff --git a/src/langtrace_python_sdk/instrumentation/pinecone/patch.py b/src/langtrace_python_sdk/instrumentation/pinecone/patch.py
@@ -21,6 +21,7 @@ def traced_method(wrapped, instance, args, kwargs):
         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,

diff --git a/src/tests/anthropic/test_anthropic.py b/src/tests/anthropic/test_anthropic.py
@@ -46,6 +46,7 @@ def test_anthropic(self):
         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,

diff --git a/src/tests/chroma/test_chroma.py b/src/tests/chroma/test_chroma.py
@@ -39,6 +39,7 @@ def test_collection_patch_success(self):
 
         # 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',

diff --git a/src/tests/langchain/test_langchain.py b/src/tests/langchain/test_langchain.py
@@ -38,6 +38,7 @@ def test_generic_patch(self):
 
         service_provider = "Langchain"
         expected_attributes = {
+            'langtrace.sdk.name': 'langtrace-python-sdk',
             "langtrace.service.name": service_provider,
             "langtrace.service.type": "framework",
             "langtrace.service.version": version,

diff --git a/src/tests/langchain/test_langchain_community.py b/src/tests/langchain/test_langchain_community.py
@@ -38,6 +38,7 @@ def test_generic_patch(self):
 
         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,

diff --git a/src/tests/langchain/test_langchain_core.py b/src/tests/langchain/test_langchain_core.py
@@ -38,6 +38,7 @@ def test_generic_patch(self):
 
         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 @@ def test_runnable_patch(self):
 
         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,

diff --git a/src/tests/openai/test_chat_completion.py b/src/tests/openai/test_chat_completion.py
@@ -63,6 +63,7 @@ def test_chat_completions_create_non_streaming(self):
         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,

diff --git a/src/tests/openai/test_image_generation.py b/src/tests/openai/test_image_generation.py
@@ -51,6 +51,7 @@ def test_image_generation(self):
         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,

diff --git a/src/tests/pinecone/test_pinecone.py b/src/tests/pinecone/test_pinecone.py
@@ -42,6 +42,7 @@ def test_pinecone(self):
         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,