temporary patch

Signed-off-by: Adrian Cole <[email protected]>

Author: codefromthecrypt

example-apps/chatbot-rag-app/README.md

@@ -127,6 +127,13 @@ dotenv run -- python api/app.py
 If you set `OTEL_SDK_DISABLED=false` in your `.env` file, the app will send
 logs, metrics and traces to an OpenTelemetry compatible endpoint.
 
+This happens automatically, when using docker. If running with python directly,
+prefix `python` with `opentelemetry-instrument` to enable OpenTelemetry.
+
+```bash
+dotenv run -- opentelemetry-instrument python api/app.py
+```
+
 [env.example](env.example) defaults to use Elastic APM server, started by
 [docker-compose-elastic.yml](../../docker). If you start your Elastic stack
 this way, you can access Kibana like this, authenticating with the username

example-apps/chatbot-rag-app/api/llm_integrations.py

@@ -1,6 +1,5 @@
 import os
 
-import boto3
 from langchain_aws import ChatBedrock
 from langchain_cohere import ChatCohere
 from langchain_google_vertexai import ChatVertexAI
@@ -41,8 +40,25 @@ def init_bedrock(temperature):
     from langtrace_python_sdk.instrumentation import AWSBedrockInstrumentation
 
     AWSBedrockInstrumentation().instrument()
+
+    # TODO: Remove after https://github.com/Scale3-Labs/langtrace-python-sdk/issues/458
+    from langtrace_python_sdk.instrumentation.aws_bedrock.patch import patch_aws_bedrock
+    from opentelemetry.trace import get_tracer
+    import importlib.metadata
+    from wrapt import wrap_function_wrapper as _W
+
+    tracer = get_tracer(
+        __name__,
+    )
+    version = importlib.metadata.version("boto3")
+
+    _W(
+        module="boto3.session",
+        name="Session.client",
+        wrapper=patch_aws_bedrock(tracer, version),
+    )
+
     return ChatBedrock(
-        client=boto3.client("bedrock-runtime"),
         model_id=os.getenv("CHAT_MODEL"),
         streaming=True,
         model_kwargs={"temperature": temperature},