langtrace-python-sdk 3.3.25__py3-none-any.whl → 3.3.27__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/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/version.py

@@ -1 +1 @@
-__version__ = "3.3.25"
+__version__ = "3.3.27"

{langtrace_python_sdk-3.3.25.dist-info → langtrace_python_sdk-3.3.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langtrace-python-sdk
-Version: 3.3.25
+Version: 3.3.27
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -27,6 +27,7 @@ Requires-Dist: ujson>=5.10.0
 Provides-Extra: dev
 Requires-Dist: anthropic; extra == 'dev'
 Requires-Dist: boto3; extra == 'dev'
+Requires-Dist: botocore; extra == 'dev'
 Requires-Dist: chromadb; extra == 'dev'
 Requires-Dist: cohere; extra == 'dev'
 Requires-Dist: embedchain; extra == 'dev'

{langtrace_python_sdk-3.3.25.dist-info → langtrace_python_sdk-3.3.27.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=az2Dhu6yZKkkThV8SgI4rYzbUuCu_ZG2DXjZd5fMiF0,23
+langtrace_python_sdk/version.py,sha256=LAz3X1vgk9Gsvl-zwf37Ce34JFFJ7mYEirRyrbMqxhA,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
@@ -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
@@ -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.25.dist-info/METADATA,sha256=s4G6wsJb-vgM1moTavHUaw6q_aWzKvpxrLgapDjxOsU,15676
-langtrace_python_sdk-3.3.25.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langtrace_python_sdk-3.3.25.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
-langtrace_python_sdk-3.3.25.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-langtrace_python_sdk-3.3.25.dist-info/RECORD,,
+langtrace_python_sdk-3.3.27.dist-info/METADATA,sha256=uib3DFDvv41ZHpvKdp1g1M6-DenvXhuvilf0Y4MH_e0,15716
+langtrace_python_sdk-3.3.27.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langtrace_python_sdk-3.3.27.dist-info/entry_points.txt,sha256=1_b9-qvf2fE7uQNZcbUei9vLpFZBbbh9LrtGw95ssAo,70
+langtrace_python_sdk-3.3.27.dist-info/licenses/LICENSE,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+langtrace_python_sdk-3.3.27.dist-info/RECORD,,