gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor.

Browse source
This commit is contained in:
Kevin Hu 2025-11-17 10:36:50 +08:00
parent dc299526d2
commit 2ebe459586
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
agent/canvas.py
View file

@ -25,7 +25,6 @@ from typing import Any, Union, Tuple
from agent.component import component_class from agent.component import component_class
from agent.component.base import ComponentBase from agent.component.base import ComponentBase
from api.db.services.file_service import FileService
from api.db.services.task_service import has_canceled from api.db.services.task_service import has_canceled
from common.misc_utils import get_uuid, hash_str2int from common.misc_utils import get_uuid, hash_str2int
from common.exceptions import TaskCanceledException from common.exceptions import TaskCanceledException
@ -545,6 +544,7 @@ class Canvas(Graph):
return self.components[cpnnm]["obj"].get_input_elements() return self.components[cpnnm]["obj"].get_input_elements()
def get_files(self, files: Union[None, list[dict]]) -> list[str]: def get_files(self, files: Union[None, list[dict]]) -> list[str]:
from api.db.services.file_service import FileService
if not files: if not files:
return [] return []
def image_to_base64(file): def image_to_base64(file):

2
rag/prompts/generator.py
View file

@ -52,7 +52,7 @@ def chunks_format(reference):
"similarity": chunk.get("similarity"), "similarity": chunk.get("similarity"),
"vector_similarity": chunk.get("vector_similarity"), "vector_similarity": chunk.get("vector_similarity"),
"term_similarity": chunk.get("term_similarity"), "term_similarity": chunk.get("term_similarity"),
"doc_type": chunk.get("doc_type_kwd"), "doc_type": get_value(chunk, "doc_type_kwd", "doc_type"),
} }
for chunk in reference.get("chunks", []) for chunk in reference.get("chunks", [])
] ]