llm/bedrock.py aktualisiert

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 self.client.converse(**request)
-
-        return response