Merge pull request #320 from Sema4AI/fix-missing-assistant-id

Fix thread state 500 error (when Retrieval tool is used)
langchain-ai · Apr 25, 2024 · e1ebc7b · e1ebc7b
2 parents 7bb64a8 + 8014057
commit e1ebc7b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/backend/app/storage.py b/backend/app/storage.py
@@ -107,6 +107,7 @@ async def get_thread_state(*, user_id: str, thread_id: str, assistant_id: str):
             "configurable": {
                 **assistant["config"]["configurable"],
                 "thread_id": thread_id,
+                "assistant_id": assistant_id,
             }
         }
     )
@@ -130,6 +131,7 @@ async def update_thread_state(
             "configurable": {
                 **assistant["config"]["configurable"],
                 **config["configurable"],
+                "assistant_id": assistant_id,
             }
         },
         values,
@@ -151,6 +153,7 @@ async def get_thread_history(*, user_id: str, thread_id: str, assistant_id: str)
                 "configurable": {
                     **assistant["config"]["configurable"],
                     "thread_id": thread_id,
+                    "assistant_id": assistant_id,
                 }
             }
         )