langtrace-python-sdk 3.3.24__tar.gz → 3.3.26__tar.gz

{langtrace_python_sdk-3.3.24 → langtrace_python_sdk-3.3.26}/PKG-INFO

@@ -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 → langtrace_python_sdk-3.3.26}/src/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()

{langtrace_python_sdk-3.3.24 → langtrace_python_sdk-3.3.26}/src/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-3.3.24 → langtrace_python_sdk-3.3.26}/src/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-3.3.24 → langtrace_python_sdk-3.3.26}/src/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-3.3.24 → langtrace_python_sdk-3.3.26}/src/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-3.3.26/src/langtrace_python_sdk/version.py

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

langtrace_python_sdk-3.3.24/src/langtrace_python_sdk/version.py

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