next

Author: eupharis

science/completion.py

@@ -302,8 +302,6 @@ def get_completion_answer(user_id: Optional[int], question: str) -> str:
         store_message(
             user_id,
             message,
-            function_ids=message.get("function_ids", []),
-            webhook_ids=message.get("webhook_ids", []),
         )
     store_message(user_id, {"role": "assistant", "content": answer})
 

science/utils.py

@@ -65,14 +65,14 @@ def store_message(
     db = get_client()
     create_input = {"userId": user_id, "role": data["role"], "content": data["content"]}
 
-    if data["function_ids"]:
+    if data.get("function_ids"):
         create_input["functions"] = {
             "create": [
                 {"functionPublicId": fid}
                 for fid in data["function_ids"]
             ]
         }
-    if data["webhook_ids"]:
+    if data.get("webhook_ids"):
         create_input["webhooks"] = {
             "create": [
                 {"webhookPublicId": wid}