fix: preserve reasoning_content in multi-turn conversations

When using thinking-enabled models (like Kimi K2.5, DeepSeek), the API
expects reasoning_content on all assistant messages. The original
ChatDeepSeek stores reasoning_content in additional_kwargs but doesn't
include it when making subsequent API calls, causing "reasoning_content
is missing" errors.

This adds PatchedChatDeepSeek which overrides _get_request_payload to
restore reasoning_content from additional_kwargs into the payload.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
hetaoBackend
2026-01-28 14:03:43 +08:00
parent d84a34b7cd
commit fa9fba3f8e
4 changed files with 93 additions and 7 deletions

View File

@@ -6,7 +6,7 @@ from typing import NotRequired, override
from langchain.agents import AgentState
from langchain.agents.middleware import AgentMiddleware
from langchain_core.messages import SystemMessage
from langchain_core.messages import HumanMessage
from langgraph.runtime import Runtime
from src.agents.middlewares.thread_data_middleware import THREAD_DATA_BASE_DIR
@@ -124,17 +124,19 @@ class UploadsMiddleware(AgentMiddleware[UploadsMiddlewareState]):
# List uploaded files
files = self._list_uploaded_files(thread_id)
if not files:
return None
# Create system message with file list
files_message = self._create_files_message(files)
system_message = SystemMessage(content=files_message)
files_human_message = HumanMessage(content=files_message)
# Inject the message into the message history
# This will be added after the system prompt but before user messages
# This will be added before user messages
messages = list(state.get("messages", []))
# Insert after the first system message (the main prompt)
insert_index = 1 if messages and hasattr(messages[0], "type") and messages[0].type == "system" else 0
messages.insert(insert_index, system_message)
insert_index = 0
messages.insert(insert_index, files_human_message)
return {
"uploaded_files": files,