langtrace-python-sdk 1.2.12__py3-none-any.whl → 1.2.16__py3-none-any.whl

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/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/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/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/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/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):
 
@@ -25,17 +28,17 @@ class OpenAIInstrumentation(BaseInstrumentor):
         wrap_function_wrapper(
             "openai.resources.chat.completions",
             "Completions.create",
-            chat_completions_create(openai.chat.completions.create, version, tracer),
+            chat_completions_create("openai.chat.completions.create", version, tracer),
         )
         wrap_function_wrapper(
             "openai.resources.images",
             "Images.generate",
-            images_generate(openai.images.generate, version, tracer),
+            images_generate("openai.images.generate", version, tracer),
         )
         wrap_function_wrapper(
             "openai.resources.embeddings",
             "Embeddings.create",
-            embeddings_create(openai.embeddings.create, version, tracer),
+            embeddings_create("openai.embeddings.create", version, tracer),
         )
 
     def _uninstrument(self, **kwargs):

langtrace_python_sdk/instrumentation/openai/patch.py

@@ -52,7 +52,7 @@ def images_generate(original_method, version, tracer):
                     span.set_attribute(field, value)
             try:
                 # Attempt to call the original method
-                result = original_method(*args, **kwargs)
+                result = wrapped(*args, **kwargs)
                 if kwargs.get("stream") is False or kwargs.get("stream") is None:
                     data = (
                         result.data[0]
@@ -132,8 +132,8 @@ def chat_completions_create(original_method, version, tracer):
                 span.set_attribute(field, value)
         try:
             # Attempt to call the original method
-            result = original_method(*args, **kwargs)
-            if kwargs.get("stream") is False:
+            result = wrapped(*args, **kwargs)
+            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 = [
@@ -338,14 +338,12 @@ 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)
             try:
                 # Attempt to call the original method
-                result = original_method(*args, **kwargs)
+                result = wrapped(*args, **kwargs)
                 span.set_status(StatusCode.OK)
                 return result
             except Exception as e:

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

@@ -1 +1 @@
-__version__ = "1.2.12"
+__version__ = "1.2.16"

{langtrace_python_sdk-1.2.12.dist-info → langtrace_python_sdk-1.2.16.dist-info}/METADATA

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

@@ -18,7 +18,7 @@ examples/pinecone_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 examples/pinecone_example/basic.py,sha256=hdV6-5Fmol9zeyFzDtdabD62vkqUJ4lCHG2YcVNpIpI,933
 langtrace_python_sdk/__init__.py,sha256=SlHg447-nQBbw8exRNJP_OyHUZ39Sldb7aaQ35hIRm8,262
 langtrace_python_sdk/langtrace.py,sha256=sWC7WqbOdOUB1sZ_hzLZVp0_Q8fBTRJRMUvOPyO6f20,3170
-langtrace_python_sdk/version.py,sha256=dbW85A2PinQCZabwD2DNDTfOE9315GDtQQKAsJP8IXk,23
+langtrace_python_sdk/version.py,sha256=7RIJ-Nh9kHJf5O5NvahUeP8DNXq6oIzbYcIt_yKv0lQ,23
 langtrace_python_sdk/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=OfgvphRBPENwsAizqzka9lFu9aDw-A7ErYJ0atl3bpI,56
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -28,16 +28,16 @@ langtrace_python_sdk/constants/instrumentation/common.py,sha256=KzNCWEl_44pkZ5Dc
 langtrace_python_sdk/constants/instrumentation/openai.py,sha256=9VF6ic9Ed3bpSvdp6iNmrpx2Ppo6DPav3hoUcqSQSv0,1048
 langtrace_python_sdk/constants/instrumentation/pinecone.py,sha256=Xaqqw-xBO0JJLGk75hiCUQGztNm0HiVaLQvjtYK7VJM,472
 langtrace_python_sdk/extensions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=7lIsChmQSOWqFxiegNbHEA_rOTpbkm-RBlBYvKIHxg4,4097
+langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=dEMSwkbqVMMToD6mH1m3x4i0Dk0g4I8-oNhOmoFZDBA,3980
 langtrace_python_sdk/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/anthropic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/anthropic/instrumentation.py,sha256=1shNkDE7Yb-JFVlEZHMcDbJ6zW8SkSFdxf_yFo9wTNA,1101
+langtrace_python_sdk/instrumentation/anthropic/instrumentation.py,sha256=DTvz6fNMMxYc2z6oqOveBX6mIXHoOFeOoosyXB2HJ6Y,1159
 langtrace_python_sdk/instrumentation/anthropic/patch.py,sha256=_-dNjxJbJTgJfHfwC_zGoxAXupL86jfBLlZefLuCBKk,7115
 langtrace_python_sdk/instrumentation/chroma/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/chroma/instrumentation.py,sha256=fSR1ZjoViwjbPYkUWJnRpFwWtwERyf4UqnMiDl8eGmY,1223
+langtrace_python_sdk/instrumentation/chroma/instrumentation.py,sha256=9k8KueizRJuicJnuYmSvjLwdeCNszZ1KpwZPMsQ_7ig,1280
 langtrace_python_sdk/instrumentation/chroma/patch.py,sha256=IORHg5GMrzFiuzMWzAbrZzmxQTRu-zzksgh5RDPGoG8,2245
 langtrace_python_sdk/instrumentation/langchain/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/langchain/instrumentation.py,sha256=im9RSJ3twIWVtzSE0sT-0VY82b_CxSzs-Tk7Z4nas3Q,2880
+langtrace_python_sdk/instrumentation/langchain/instrumentation.py,sha256=UBUx31IErih1zDGkwez5ZFT2qGvCHxhdGHvJsEyhvEo,2938
 langtrace_python_sdk/instrumentation/langchain/patch.py,sha256=z_brH_5BSze1MWjWBycyOPMrJAAw65OUM5vcvvatoZM,3236
 langtrace_python_sdk/instrumentation/langchain_community/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/langchain_community/instrumentation.py,sha256=CS6oAxnhQ98gjD5Xj7LrWDE9i-FVjWI3kOwZ4phspnU,4677
@@ -46,18 +46,18 @@ langtrace_python_sdk/instrumentation/langchain_core/__init__.py,sha256=47DEQpj8H
 langtrace_python_sdk/instrumentation/langchain_core/instrumentation.py,sha256=1Y4q_F2nN79y-dap_if35vH9oowIuDi6mo0ELYIltNI,5417
 langtrace_python_sdk/instrumentation/langchain_core/patch.py,sha256=zr5nvqeOX6IlD1gDbRLRz7Frep40EvrOSz8BCy1XgJc,7843
 langtrace_python_sdk/instrumentation/llamaindex/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py,sha256=udDxk87H731x0mRFNr0jdtstfnSwiirxVPo6Z4UWAtY,2430
+langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py,sha256=wVSvxSJb1owoEBCSqPQRzsbdEkPAQzTqXKKSxGk0sSY,2488
 langtrace_python_sdk/instrumentation/llamaindex/patch.py,sha256=wTWmyTD09J44ULGoMVmgmq72VN1H0xmdrISTdzqo1Yw,1985
 langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=6RmbONTwgr8_do92l3ll7-ia4maXMxQVCwPBRRa-3NA,1358
-langtrace_python_sdk/instrumentation/openai/patch.py,sha256=ffj5W-R1WtcLkahur2e6s5eYJIGCiV7PkNIA8ZXwP1c,14982
+langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=stlFopPggB8mgnn4QdVh41rmA8mlzH5x-C00XHfApCE,1408
+langtrace_python_sdk/instrumentation/openai/patch.py,sha256=YBEv4OEXIvKQXmRxGmNsFEOqPtQU2mzZ7jNS3bdWk4A,14917
 langtrace_python_sdk/instrumentation/pinecone/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=PBkCMiaDr-WXkGfHt9iXtYqyl7a-xIO52pRmUFuMURo,1726
+langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=o0EUd5jvHaDKOUTj4NjnL5UfDHDHxyXkWGlTW4oeRDk,1784
 langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=lvWL_xM7mxTQ1D-eY5Gh949c45BAKrN2N5IzKRkUhMY,2142
 langtrace_python_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/utils/llm.py,sha256=4z2e-md_ELXCEuOIRVWracR6qH2pmsOxCqpkuF9_3Nw,1589
 langtrace_python_sdk/utils/with_root_span.py,sha256=N7ONrcF0myZbHBy5gpQffDbX-Kf63Crsz9szG0i3m08,1889
-langtrace_python_sdk-1.2.12.dist-info/METADATA,sha256=gD8Lo6pcTxuqf8X9Di30NSpe7AYvSKGiBmrs2c0otIo,7695
-langtrace_python_sdk-1.2.12.dist-info/WHEEL,sha256=uNdcs2TADwSd5pVaP0Z_kcjcvvTUklh2S7bxZMF8Uj0,87
-langtrace_python_sdk-1.2.12.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-1.2.12.dist-info/RECORD,,
+langtrace_python_sdk-1.2.16.dist-info/METADATA,sha256=S-mgSIqMpKlKju44vtCpxjfESTY0ohot-O3kvh8QFgY,7695
+langtrace_python_sdk-1.2.16.dist-info/WHEEL,sha256=uNdcs2TADwSd5pVaP0Z_kcjcvvTUklh2S7bxZMF8Uj0,87
+langtrace_python_sdk-1.2.16.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-1.2.16.dist-info/RECORD,,