소스 검색

Fix: retrieval debug mode issue. (#6150)

### What problem does this PR solve?

#6139

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.18.0
Kevin Hu 7 달 전
부모
커밋
bfa8d342b3
No account linked to committer's email address
4개의 변경된 파일4개의 추가작업 그리고 4개의 파일을 삭제
  1. 1
    1
      agent/component/base.py
  2. 1
    2
      agent/component/retrieval.py
  3. 1
    0
      deepdoc/parser/pdf_parser.py
  4. 1
    1
      rag/app/naive.py

+ 1
- 1
agent/component/base.py 파일 보기

@@ -521,7 +521,7 @@ class ComponentBase(ABC):
if u.lower().find("answer") >= 0:
for r, c in self._canvas.history[::-1]:
if r == "user":
upstream_outs.append(pd.DataFrame([{"content": f"USER:{c}", "component_id": u}]))
upstream_outs.append(pd.DataFrame([{"content": c, "component_id": u}]))
break
break
if self.component_name.lower().find("answer") >= 0 and self.get_component_name(u) in ["relevant"]:

+ 1
- 2
agent/component/retrieval.py 파일 보기

@@ -57,8 +57,7 @@ class Retrieval(ComponentBase, ABC):
query = self.get_input()
query = str(query["content"][0]) if "content" in query else ""
lines = query.split('\n')
user_queries = [line.split("USER:", 1)[1] for line in lines if line.startswith("USER:")]
query = user_queries[-1] if user_queries else ""
query = lines[-1] if lines else ""
kbs = KnowledgebaseService.get_by_ids(self._param.kb_ids)
if not kbs:
return Retrieval.be_output("")

+ 1
- 0
deepdoc/parser/pdf_parser.py 파일 보기

@@ -41,6 +41,7 @@ LOCK_KEY_pdfplumber = "global_shared_lock_pdfplumber"
if LOCK_KEY_pdfplumber not in sys.modules:
sys.modules[LOCK_KEY_pdfplumber] = threading.Lock()


class RAGFlowPdfParser:
def __init__(self, parallel_devices: int | None = None):
"""

+ 1
- 1
rag/app/naive.py 파일 보기

@@ -197,7 +197,7 @@ class Markdown(MarkdownParser):


def chunk(filename, binary=None, from_page=0, to_page=100000,
lang="Chinese", parallel_devices=None, callback=None, **kwargs):
lang="Chinese", callback=None, parallel_devices=None, **kwargs):
"""
Supported file formats are docx, pdf, excel, txt.
This method apply the naive ways to chunk files.

Loading…
취소
저장