Merge pull request #21 from marcburnie/fix/null-history-case

fix for null history case
aws-samples · Nov 30, 2023 · 3a2ed33 · 3a2ed33
2 parents 8286966 + 9622fa7
commit 3a2ed33
Showing 1 changed file with 11 additions and 13 deletions.
diff --git a/lambdas/qna_bot_qbusiness_lambdahook/src/lambdahook.py b/lambdas/qna_bot_qbusiness_lambdahook/src/lambdahook.py
@@ -19,21 +19,19 @@
 
 def get_amazonq_response(prompt, context, amazonq_userid):
     print(f"get_amazonq_response: prompt={prompt}, app_id={AMAZONQ_APP_ID}, context={context}")
+    input = {
+        "applicationId": AMAZONQ_APP_ID,
+        "userMessage": prompt,
+        "userId": amazonq_userid
+    }
     if context:
-        input = {
-            "applicationId": AMAZONQ_APP_ID,
-            "conversationId": context["conversationId"],
-            "parentMessageId": context["parentMessageId"],
-            "userMessage": prompt,
-            "userId": amazonq_userid
-        }
+        if context["conversationId"]:
+            input["conversationId"] = context["conversationId"]
+        if context["parentMessageId"]:
+            input["parentMessageId"] = context["parentMessageId"]
     else:
-        input = {
-            "applicationId": AMAZONQ_APP_ID,
-            "userMessage": prompt,
-            "userId": amazonq_userid,
-            "clientToken": str(uuid.uuid4())
-        }
+        input["clientToken"] = str(uuid.uuid4())
+
     print("Amazon Q Input: ", json.dumps(input))
     try:
         resp = qbusiness_client.chat_sync(**input)