Temporarily removes full_text from vector searches

nulib · Feb 6, 2024 · e4b8efc · e4b8efc
1 parent e913cb3
commit e4b8efc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/chat/src/event_config.py b/chat/src/event_config.py
@@ -122,7 +122,7 @@ def _get_attributes(self):
         attributes = [
             item
             for item in self._get_request_attributes()
-            if item not in [self._get_text_key(), "source"]
+            if item not in [self._get_text_key(), "source", "full_text"]
         ]
         return attributes
 

diff --git a/chat/src/handlers/chat.py b/chat/src/handlers/chat.py
@@ -2,7 +2,6 @@
 from event_config import EventConfig
 from helpers.response import prepare_response
 
-
 def handler(event, _context):
     try:
         config = EventConfig(event)
@@ -21,5 +20,10 @@ def handler(event, _context):
             final_response = prepare_response(config)
             config.socket.send(final_response)
         return {"statusCode": 200}
+
     except Exception as err:
-        raise err
+        if err.__class__.__name__ == "PayloadTooLargeException":
+            config.socket.send({"type": "error", "message": "Payload too large"})
+            return {"statusCode": 413, "body": "Payload too large"}
+        else:
+            raise err