Fix JSON parsing issue in history.py

FloRul · Mar 21, 2024 · 7847dcf · 7847dcf
1 parent 51585ae
commit 7847dcf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lambdas/inference/src/history.py b/lambdas/inference/src/history.py
@@ -21,7 +21,9 @@ def get(self, limit: int = 10):
                 InvocationType="RequestResponse",
                 Payload=json.dumps(payload),
             )
-            body = json.loads(response["Payload"].read().decode("utf-8-sig"))["body"]
+            body = json.loads(
+                json.loads(response["Payload"].read().decode("utf-8-sig"))["body"]
+            )
             for x in body:
                 result.append({"role": "user", "content": x["HumanMessage"]})
                 result.append({"role": "assistant", "content": x["AssistantMessage"]})