jack 1 周之前
父节点
当前提交
e2001cbf7b
共有 1 个文件被更改,包括 6 次插入1 次删除
  1. 6 1
      app/services/message/message.py

+ 6 - 1
app/services/message/message.py

@@ -72,7 +72,7 @@ class MessageService:
         *, session: AsyncSession, body: MessageCreate, thread_id: str
     ) -> Message:
         # get thread
-        thread = await ThreadService.get_thread(thread_id=thread_id, session=session)
+        thread = await ThreadService.get_thread(session=session, thread_id=thread_id)
         print(
             "create_messagecreate_messagecreate_messagecreate_messagecreate_messagecreate_message"
         )
@@ -106,10 +106,15 @@ class MessageService:
             thread.tool_resources.get("file_search").get("vector_stores")[0][
                 "file_ids"
             ] = thread_file_ids
+            setattr(thread, "tool_resources", thread.tool_resources)
+            # thread.tool_resources = thread.tool_resources
             print(thread)
             session.add(thread)
             await session.commit()
             await session.refresh(thread)
+            # session.add(thread)
+            # await session.commit()
+            # await session.refresh(thread)
 
         content = MessageService.format_message_content(body)
         db_message = Message.model_validate(