feat: 删除聊天记录接口

app/controller/add_langgraph_chat_route.py

@@ -12,6 +12,8 @@ from langgraph.types import interrupt, Command
 from pydantic import BaseModel, Field
 from starlette.responses import StreamingResponse
 
+from app.model.ConversationModel import ConversationService
+from app.utils.db_utils import AsyncSessionDep
 from app.utils.llm_utils import create_llm
 from app.utils.postgres_checkpointer import PostgresCheckpointerManager, AsyncPostgresSaverDep
 
@@ -30,7 +32,6 @@ def tool_get_datetime():
   }
 
 
-
 @tool(
   name_or_callable="tool_multiply",
   description="一个乘法工具，用于计算两个数字相乘"
@@ -238,4 +239,11 @@ def add_langgraph_chat_route(app: FastAPI):
   # 查询聊天记录
   @app.get("/langgraph/chat_state/{thread_id}")
   async def langgraph_chat(thread_id: str):
-    return await ChatAgent.get_chat_state(thread_id)
+    return await ChatAgent.get_chat_state(thread_id)  # 查询聊天记录
+
+  # 删除聊天记录
+  @app.post("/langgraph/chat_remove/{thread_id}")
+  async def langgraph_chat(thread_id: str, checkpointer: AsyncPostgresSaverDep, session=AsyncSessionDep):
+    await ConversationService.item_delete(session=session, row_dict={"id": thread_id})
+    await checkpointer.adelete_thread(thread_id)
+    return {"result": "success"}