zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-05 15:20:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix deepcopy the chat model (#17560)

Browse Source
This commit is contained in:
Wang Qi
2026-07-30 14:29:24 +08:00
committed by GitHub
parent 493f605889
commit 4592d06a67
4 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rag/advanced_rag/agentic_rag.py
View File

@@ -82,7 +82,7 @@ class RAGTools:
thinking_mode: str = "medium",
):
self.tenant_ids = tenant_ids
self.chat_mdl = chat_mdl
self.chat_mdl = chat_mdl.clone()
self.embed_mdl = embed_mdl
self.thinking_mode = thinking_mode
self.field_map = {}

7
rag/advanced_rag/harness/agent.py
View File

@@ -1,6 +1,6 @@
"""Research Agent — inner tool-calling loop for high/ultra modes.
Native tool-calling: a chat model deep-copied from ``tools.chat_mdl`` is bound
Native tool-calling: a chat model cloned from ``tools.chat_mdl`` is bound
(via ``bind_tools``) to the phase-gated tool schemas plus ``think_tool`` /
``generate_report``, and a lightweight session routes each tool call to the
harness pipeline. Binding onto a *copy* keeps the shared ``tools.chat_mdl``
@@ -14,7 +14,6 @@ that the loop parses.
import json
import logging
import re
from copy import deepcopy
from rag.advanced_rag.harness.types import ClaimTarget, ExecutionStrategy, ToolResult
from rag.advanced_rag.harness.pipeline import Pipeline
@@ -158,8 +157,8 @@ async def research_agent_loop(
has_routed_scope=bool(getattr(pipeline, "_routed_docs", None)),
)
# Deep-copy so binding tools never leaks onto the shared chat model.
agent_mdl = deepcopy(tools.chat_mdl)
# Clone so binding tools never leaks onto the shared chat model.
agent_mdl = tools.chat_mdl.clone()
if getattr(agent_mdl, "is_tools", False):
return await _research_native(claim, agent_mdl, pipeline, phase, phase_config, gated_defs, mode)