diff --git a/llm/bedrock_client.py b/llm/bedrock.py similarity index 63% rename from llm/bedrock_client.py rename to llm/bedrock.py index 74b8adf..f3f6a72 100644 --- a/llm/bedrock_client.py +++ b/llm/bedrock.py @@ -1,5 +1,4 @@ import boto3 -import json from config import AWS_REGION, BEDROCK_MODEL @@ -7,26 +6,20 @@ from config import AWS_REGION, BEDROCK_MODEL class BedrockClient: def __init__(self): self.client = boto3.client( - service_name="bedrock-runtime", + "bedrock-runtime", region_name=AWS_REGION ) async def chat(self, messages, tools=None): - """ - Converse API für Amazon Nova / Bedrock Models - """ request = { "modelId": BEDROCK_MODEL, "messages": messages, } - # Tools nur hinzufügen wenn vorhanden if tools: request["toolConfig"] = { "tools": tools } - response = self.client.converse(**request) - - return response \ No newline at end of file + return self.client.converse(**request) \ No newline at end of file