langtrace-python-sdk 3.3.24__py3-none-any.whl → 3.3.26__py3-none-any.whl

examples/awsbedrock_examples/__init__.py

@@ -1,5 +1,6 @@
 from examples.awsbedrock_examples.converse import (
     use_invoke_model_anthropic,
+    init_bedrock_langchain,
     use_invoke_model_titan,
     use_invoke_model_llama,
 )
@@ -15,4 +16,5 @@ class AWSBedrockRunner:
         # use_invoke_model_anthropic(stream=True)
         # use_invoke_model_cohere()
         # use_invoke_model_llama(stream=False)
-        use_invoke_model_titan(stream=False)
+        # use_invoke_model_titan(stream=False)
+        init_bedrock_langchain()

examples/awsbedrock_examples/converse.py

@@ -3,6 +3,7 @@ import json
 from langtrace_python_sdk import langtrace
 from dotenv import load_dotenv
 import botocore
+from langchain_aws import ChatBedrock
 
 load_dotenv()
 langtrace.init(write_spans_to_console=False)
@@ -186,3 +187,13 @@ def use_invoke_model_cohere():
     response = brt.invoke_model(body=body, modelId=model_id)
     response_body = json.loads(response.get("body").read())
     print(response_body)
+
+
+def init_bedrock_langchain(temperature=0.1):
+    chat = ChatBedrock(
+        model_id="anthropic.claude-v2",
+        streaming=True,
+        model_kwargs={"temperature": temperature},
+        region_name="us-east-1",
+    )
+    return chat.invoke("What is the capital of France?")

langtrace_python_sdk/constants/instrumentation/common.py

@@ -10,6 +10,7 @@ TIKTOKEN_MODEL_MAPPING = {
 
 SERVICE_PROVIDERS = {
     "ANTHROPIC": "Anthropic",
+    "ARCH": "Arch",
     "AZURE": "Azure",
     "CHROMA": "Chroma",
     "CREWAI": "CrewAI",

langtrace_python_sdk/instrumentation/aws_bedrock/instrumentation.py

@@ -42,6 +42,11 @@ class AWSBedrockInstrumentation(BaseInstrumentor):
             name="client",
             wrapper=patch_aws_bedrock(tracer, version),
         )
+        _W(
+            module="boto3.session",
+            name="Session.client",
+            wrapper=patch_aws_bedrock(tracer, version),
+        )
 
     def _uninstrument(self, **kwargs):
         pass

langtrace_python_sdk/instrumentation/openai/patch.py

@@ -254,6 +254,8 @@ def chat_completions_create(version: str, tracer: Tracer) -> Callable:
             service_provider = SERVICE_PROVIDERS["XAI"]
         elif "deepseek" in get_base_url(instance):
             service_provider = SERVICE_PROVIDERS["DEEPSEEK"]
+        elif ":12000" in get_base_url(instance) or ":10000" in get_base_url(instance):
+            service_provider = SERVICE_PROVIDERS["ARCH"]
         llm_prompts = []
         for item in kwargs.get("messages", []):
             tools = get_tool_calls(item)
@@ -348,6 +350,8 @@ def async_chat_completions_create(version: str, tracer: Tracer) -> Callable:
             service_provider = SERVICE_PROVIDERS["XAI"]
         elif "deepseek" in get_base_url(instance):
             service_provider = SERVICE_PROVIDERS["DEEPSEEK"]
+        elif ":12000" in get_base_url(instance) or ":10000" in get_base_url(instance):
+            service_provider = SERVICE_PROVIDERS["ARCH"]
         llm_prompts = []
         for item in kwargs.get("messages", []):
             tools = get_tool_calls(item)
@@ -447,6 +451,8 @@ def embeddings_create(version: str, tracer: Tracer) -> Callable:
             service_provider = SERVICE_PROVIDERS["XAI"]
         elif "deepseek" in base_url:
             service_provider = SERVICE_PROVIDERS["DEEPSEEK"]
+        elif ":12000" in base_url or ":10000" in base_url:
+            service_provider = SERVICE_PROVIDERS["ARCH"]
 
         span_attributes = {
             **get_langtrace_attributes(version, service_provider, vendor_type="llm"),
@@ -533,6 +539,8 @@ def async_embeddings_create(version: str, tracer: Tracer) -> Callable:
             service_provider = SERVICE_PROVIDERS["XAI"]
         elif "deepseek" in base_url:
             service_provider = SERVICE_PROVIDERS["DEEPSEEK"]
+        elif ":12000" in base_url or ":10000" in base_url:
+            service_provider = SERVICE_PROVIDERS["ARCH"]
 
         span_attributes = {
             **get_langtrace_attributes(version, service_provider, vendor_type="llm"),

langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "3.3.24"
+__version__ = "3.3.26"

{langtrace_python_sdk-3.3.24.dist-info → langtrace_python_sdk-3.3.26.dist-info}/METADATA

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

@@ -3,8 +3,8 @@ examples/anthropic_example/__init__.py,sha256=03us1YuvAJR6fqXX8NH2kROBfTmyz7KzFV
 examples/anthropic_example/completion.py,sha256=3_YEZrt0BLVNJT_RbLXg6JGP2bweuc_HPC2MWR73tOM,713
 examples/autogen_example/__init__.py,sha256=UJgpzL2yOmzir-DAiGFR1PB1Zz3YcQvYcq5bCN8nl0A,158
 examples/autogen_example/main.py,sha256=6OJ73VCdHgVrqnekF1S1nK8mXCUABLbUUkQtr7wOCdw,2312
-examples/awsbedrock_examples/__init__.py,sha256=sQ6dq-K2vunY9Y8r8uS-ZwrqKd8XnvV-BukWQ2Gm0fA,537
-examples/awsbedrock_examples/converse.py,sha256=41CyEla9vFT53ugLJYYHV6kchLEAfXZKMncoJHx1sds,5950
+examples/awsbedrock_examples/__init__.py,sha256=ssd7M-TxKUBsi86dsJTh39-NOpJ6cKLK3WMag7lr5go,600
+examples/awsbedrock_examples/converse.py,sha256=sZj4C8N-yQzUFM-ND1gMRVwqfrrvT37YdQJuXp3JR94,6270
 examples/azureopenai_example/__init__.py,sha256=PaZM90r6VN4eSOXxb6wGsyhf9-RJCNqBypzk1Xa2GJI,271
 examples/azureopenai_example/completion.py,sha256=K_GeU0TfJ9lLDfW5VI0Lmm8_I0JXf1x9Qi83ImJ350c,668
 examples/cerebras_example/__init__.py,sha256=ydfNi0DjFMGVcfo79XVG3VEbzIrHo5wYBgSJzl_asNA,295
@@ -111,7 +111,7 @@ examples/weaviate_example/__init__.py,sha256=8JMDBsRSEV10HfTd-YC7xb4txBjD3la56sn
 examples/weaviate_example/query_text.py,sha256=wPHQTc_58kPoKTZMygVjTj-2ZcdrIuaausJfMxNQnQc,127162
 langtrace_python_sdk/__init__.py,sha256=VZM6i71NR7pBQK6XvJWRelknuTYUhqwqE7PlicKa5Wg,1166
 langtrace_python_sdk/langtrace.py,sha256=jvfrnkAxc41dmIvwNuVcXwCWyV2vqhGV5qAr5KiMHBA,13239
-langtrace_python_sdk/version.py,sha256=BYdEglI7p7QqwHU1MEOvXpBRXVFpeqhDQ3obeVJACVs,23
+langtrace_python_sdk/version.py,sha256=DdbevBUmAlwA_uisoVeIGAf3j1sXPPksfsJs7_p3s2Q,23
 langtrace_python_sdk/constants/__init__.py,sha256=3CNYkWMdd1DrkGqzLUgNZXjdAlM6UFMlf_F-odAToyc,146
 langtrace_python_sdk/constants/exporter/langtrace_exporter.py,sha256=EVCrouYCpY98f0KSaKr4PzNxPULTZZO6dSA_crEOyJU,106
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -119,7 +119,7 @@ langtrace_python_sdk/constants/instrumentation/anthropic.py,sha256=YX3llt3zwDY6X
 langtrace_python_sdk/constants/instrumentation/aws_bedrock.py,sha256=QwKtO4NBarOZoGkt5cFCcpxAw3zvZxcMMWBbzPPGv-g,422
 langtrace_python_sdk/constants/instrumentation/chroma.py,sha256=hiPGYdHS0Yj4Kh3eaYBbuCAl_swqIygu80yFqkOgdak,955
 langtrace_python_sdk/constants/instrumentation/cohere.py,sha256=9yD133VdrYZ5BoJR4nJHlj67gHEImB9-KsD-NkzHW1I,1159
-langtrace_python_sdk/constants/instrumentation/common.py,sha256=DPDX8icb0Tj3OrgpbL9WeiIaMG7Si2IKiSL8YRwwor4,1203
+langtrace_python_sdk/constants/instrumentation/common.py,sha256=0_38AXPFLZnnI6T7wMiSDzTP8AAT-dmNi7Mui0q5WMg,1223
 langtrace_python_sdk/constants/instrumentation/embedchain.py,sha256=HodCJvaFjILoOG50OwFObxfVxt_8VUaIAIqvgoN3tzo,278
 langtrace_python_sdk/constants/instrumentation/gemini.py,sha256=UAmfgg9FM7uNeOCdPfWlir6OIH-8BoxFGPRpdBd9ZZs,358
 langtrace_python_sdk/constants/instrumentation/groq.py,sha256=VFXmIl4aqGY_fS0PAmjPj_Qm7Tibxbx7Ur_e7rQpqXc,134
@@ -146,7 +146,7 @@ langtrace_python_sdk/instrumentation/autogen/instrumentation.py,sha256=MVDUCBi6X
 langtrace_python_sdk/instrumentation/autogen/patch.py,sha256=7Sq3C8Q5tT27UkWXd1SZgnLC1pbQf_tpTrxoBIYsDw4,5273
 langtrace_python_sdk/instrumentation/aws_bedrock/__init__.py,sha256=IHqPgR1kdDvcoV1nUb-B21PaJ_qbQB0jc011Udi1ioU,96
 langtrace_python_sdk/instrumentation/aws_bedrock/bedrock_streaming_wrapper.py,sha256=_EMlxfBhjNsB-2TrP0ek-vx1CK5bpgjXPmQWNT9eBcE,1370
-langtrace_python_sdk/instrumentation/aws_bedrock/instrumentation.py,sha256=M7Dyw1tG18ptD0ctCT9yHzO4UyKCcS8JkHg9WVaz9Ck,1473
+langtrace_python_sdk/instrumentation/aws_bedrock/instrumentation.py,sha256=ohSlH-BPjIfQ6_wQbUyRDnVMWLY_XHeDv112J_MxC0k,1622
 langtrace_python_sdk/instrumentation/aws_bedrock/patch.py,sha256=KKNp9gKzoj9YuLrqA0VMDe0FUSBzKtrbXiNN1hdciJM,17941
 langtrace_python_sdk/instrumentation/aws_bedrock/stream_body_wrapper.py,sha256=ENdhRVHBhdkIlJIc_tkf8ASijUzZdVZM-oonNLdNM48,1584
 langtrace_python_sdk/instrumentation/cerebras/__init__.py,sha256=9rHNg7PWcZ7a9jExQZlqwWPkvLGcPT-DGWot0_6Bx9k,92
@@ -206,7 +206,7 @@ langtrace_python_sdk/instrumentation/ollama/instrumentation.py,sha256=jdsvkqUJAA
 langtrace_python_sdk/instrumentation/ollama/patch.py,sha256=w99r9wCCVDdJnZQEezYE2lW_iNFEtrldt9vq3ISAsag,5375
 langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=VPHRNCQEdkizIVP2d0Uw_a7t8XOTSTprEIB8oboJFbs,95
 langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=PZxI0qfoud1VsKGmJu49YDp0Z9z9TzCR8qxR3uznOMA,2810
-langtrace_python_sdk/instrumentation/openai/patch.py,sha256=u8qlCooN-JmGVWLe7jwAeBMHTIi35ktVvL9gWVBfDko,26809
+langtrace_python_sdk/instrumentation/openai/patch.py,sha256=EQRE_b_os6jqdqKOj6XZb58QT3tUyCAcFUiwM0CDDLU,27329
 langtrace_python_sdk/instrumentation/openai/types.py,sha256=aVkoa7tmAbYfyOhnyMrDaVjQuwhmRNLMthlNtKMtWX8,4311
 langtrace_python_sdk/instrumentation/pinecone/__init__.py,sha256=DzXyGh9_MGWveJvXULkFwdkf7PbG2s3bAWtT1Dmz7Ok,99
 langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=HDXkRITrVPwdQEoOYJOfMzZE_2-vDDvuqHTlD8W1lQw,1845
@@ -277,8 +277,8 @@ tests/pinecone/cassettes/test_query.yaml,sha256=b5v9G3ssUy00oG63PlFUR3JErF2Js-5A
 tests/pinecone/cassettes/test_upsert.yaml,sha256=neWmQ1v3d03V8WoLl8FoFeeCYImb8pxlJBWnFd_lITU,38607
 tests/qdrant/conftest.py,sha256=9n0uHxxIjWk9fbYc4bx-uP8lSAgLBVx-cV9UjnsyCHM,381
 tests/qdrant/test_qdrant.py,sha256=pzjAjVY2kmsmGfrI2Gs2xrolfuaNHz7l1fqGQCjp5_o,3353
-langtrace_python_sdk-3.3.24.dist-info/METADATA,sha256=V0Xhmeqs2wlRHimItoq_pExSCrYd9JPtWQn-F_brXW8,15676
-langtrace_python_sdk-3.3.24.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langtrace_python_sdk-3.3.24.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
-langtrace_python_sdk-3.3.24.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-3.3.24.dist-info/RECORD,,
+langtrace_python_sdk-3.3.26.dist-info/METADATA,sha256=G3PWh6W1uSCrD4V7aoGlwhuc7kvLM3wULDBT5r0TPnQ,15676
+langtrace_python_sdk-3.3.26.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langtrace_python_sdk-3.3.26.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
+langtrace_python_sdk-3.3.26.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-3.3.26.dist-info/RECORD,,