add bedrock examples and update version

Author: obinnaokafor

pyproject.toml

@@ -18,7 +18,7 @@ classifiers = [
   "Operating System :: OS Independent",
 ]
 dependencies = [
-  'trace-attributes==7.0.4',
+  'trace-attributes==7.1.0',
   'opentelemetry-api>=1.25.0',
   'opentelemetry-sdk>=1.25.0',
   'opentelemetry-instrumentation>=0.47b0',
@@ -57,7 +57,8 @@ dev = [
   "google-generativeai",
   "google-cloud-aiplatform",
   "mistralai",
-  "embedchain"
+  "boto3",
+  "embedchain",
 ]
 
 test = ["pytest", "pytest-vcr", "pytest-asyncio"]

src/examples/awsbedrock_examples/__init__.py

@@ -0,0 +1,10 @@
+from examples.awsbedrock_examples.converse import use_converse
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
+
+langtrace.init()
+
+
+class AWSBedrockRunner:
+    @with_langtrace_root_span("AWS_Bedrock")
+    def run(self):
+        use_converse()

src/examples/awsbedrock_examples/converse.py

@@ -0,0 +1,34 @@
+import os
+import boto3
+from langtrace_python_sdk import langtrace
+
+langtrace.init(api_key=os.environ["LANGTRACE_API_KEY"])
+
+def use_converse():
+    model_id = "anthropic.claude-3-haiku-20240307-v1:0"
+    client = boto3.client(
+        "bedrock-runtime",
+        region_name="us-east-1",
+        aws_access_key_id=os.environ["AWS_ACCESS_KEY_ID"],
+        aws_secret_access_key=os.environ["AWS_SECRET_ACCESS_KEY"],
+    )
+    conversation = [
+        {
+            "role": "user",
+            "content": [{"text": "Write a story about a magic backpack."}],
+        }
+    ]
+
+    try:
+        response = client.converse(
+            modelId=model_id,
+            messages=conversation,
+            inferenceConfig={"maxTokens":4096,"temperature":0},
+            additionalModelRequestFields={"top_k":250}
+        )
+        response_text = response["output"]["message"]["content"][0]["text"]
+        print(response_text)
+
+    except (Exception) as e:
+        print(f"ERROR: Can't invoke '{model_id}'. Reason: {e}")
+        exit(1)

src/langtrace_python_sdk/version.py

@@ -1 +1 @@
-__version__ = "3.1.3"
+__version__ = "3.2.0"