Skip to content

Commit 7655a47

Browse files
define model with bedrock
1 parent a2848bd commit 7655a47

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

strands-examples/strands-langfuse-demo/k8s_mcp_agent.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,8 @@
88
from set_telemetry import set_telemetry
99
import uuid
1010

11+
from model import model
12+
1113
def remove_html_tags(text_with_html):
1214
text_with_out_html = re.sub(r"<[^>]+>", "", text_with_html)
1315
return text_with_out_html
@@ -49,7 +51,7 @@ def __init__(self):
4951
# Create an agent with these tools
5052
self.agent = Agent(
5153
callback_handler=None,
52-
model="us.amazon.nova-micro-v1:0",
54+
model=model,
5355
tools=tools,
5456
trace_attributes={
5557
"session.id": str(uuid.uuid4()),
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
from strands.models import BedrockModel
2+
3+
model = BedrockModel(
4+
model_id="us.amazon.nova-micro-v1:0",
5+
)

0 commit comments

Comments
 (0)