Fix indentation and add user message to message history

FloRul · Mar 20, 2024 · c2f90f6 · c2f90f6
1 parent d7fdfab
commit c2f90f6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lambdas/inference/src/index.py b/lambdas/inference/src/index.py
@@ -79,8 +79,9 @@ def invoke_model(
     if source == "call":
         maxtokens //= 2
 
+
     messages = message_history.append({"role": "user", "content": user_message})
-
+    
     body = json.dumps(
         {
             "anthropic_version": "bedrock-2023-05-31",
@@ -111,7 +112,7 @@ def lambda_handler(event, context):
     embedding_collection_name = event["queryStringParameters"]["collectionName"]
 
     sessionId = str(uuid.uuid1())
-    
+
     if "sessionId" in event["queryStringParameters"]:
         sessionId = event["queryStringParameters"]["sessionId"]