AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-12 07:39:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Display thinking for deepseek r1 (#4904)

Browse Source 
### What problem does this PR solve?
#4903
### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
Kevin Hu 2025-02-12 15:43:13 +08:00 committed by GitHub
parent 9bcccadebd
commit b08bb56f6c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deepdoc/vision/operators.py
View File

@ -108,7 +108,7 @@ class NormalizeImage(object):
def __init__(self, scale=None, mean=None, std=None, order='chw', **kwargs): def __init__(self, scale=None, mean=None, std=None, order='chw', **kwargs):
if isinstance(scale, str): if isinstance(scale, str):
scale = np.float32(scale) if scale != 'None' else None scale = eval(scale)
self.scale = np.float32(scale if scale is not None else 1.0 / 255.0) self.scale = np.float32(scale if scale is not None else 1.0 / 255.0)
mean = mean if mean is not None else [0.485, 0.456, 0.406] mean = mean if mean is not None else [0.485, 0.456, 0.406]
std = std if std is not None else [0.229, 0.224, 0.225] std = std if std is not None else [0.229, 0.224, 0.225]

6
rag/llm/chat_model.py
View File

@ -73,6 +73,12 @@ class Base(ABC):
continue continue
if not resp.choices[0].delta.content: if not resp.choices[0].delta.content:
resp.choices[0].delta.content = "" resp.choices[0].delta.content = ""
if hasattr(resp.choices[0].delta, "reasoning_content") and resp.choices[0].delta.reasoning_content:
if ans.find("<think>") < 0:
ans += "<think>"
ans = ans.replace("</think>", "")
ans += resp.choices[0].delta.reasoning_content + "</think>"
else:
ans += resp.choices[0].delta.content ans += resp.choices[0].delta.content
tol = self.total_token_count(resp) tol = self.total_token_count(resp)