From dda4efb97f64b63fb17384eca27ea0a828b11aae Mon Sep 17 00:00:00 2001
From: Florian Rumiel <rumiel.florian@gmail.com>
Date: Tue, 20 Feb 2024 14:09:35 -0500
Subject: [PATCH] Update retrieval documents to use JSON serialization

---
 lambdas/inference/src/index.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lambdas/inference/src/index.py b/lambdas/inference/src/index.py
index 66c68e1..a6c9a8a 100644
--- a/lambdas/inference/src/index.py
+++ b/lambdas/inference/src/index.py
@@ -114,7 +114,7 @@ def lambda_handler(event, context):
                 {
                     "completion": response,
                     "retrieval": {
-                        "documents": [doc.to_dict() for doc in docs],
+                        "documents": [json.dumps(doc) for doc in docs],
                     },
                 }
             ),