pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove <think> for KG extraction. (#5027)

Browse Source 
### What problem does this PR solve?

#4946

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2025-02-17 14:06:06 +08:00
committed by GitHub
parent ab17606e79
commit f46448d04c
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
graphrag/general/extractor.py
View File

@ -15,6 +15,7 @@
# #
import logging import logging
import os import os
import re
from collections import defaultdict, Counter from collections import defaultdict, Counter
from concurrent.futures import ThreadPoolExecutor from concurrent.futures import ThreadPoolExecutor
from copy import deepcopy from copy import deepcopy
@ -59,6 +60,7 @@ class Extractor:
if response: if response:
return response return response
response = self._llm.chat(system, hist, conf) response = self._llm.chat(system, hist, conf)
response = re.sub(r"<think>.*</think>", "", response)
if response.find("**ERROR**") >= 0: if response.find("**ERROR**") >= 0:
raise Exception(response) raise Exception(response)
set_llm_cache(self._llm.llm_name, system, response, history, gen_conf) set_llm_cache(self._llm.llm_name, system, response, history, gen_conf)