zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-25 09:53:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: Add graph keyword search and fix dataset synthesizing issue. (#17342)

Browse Source 
### Summary

 Add graph keyword search and fix dataset synthesizing issue.
This commit is contained in:
Kevin Hu
2026-07-24 18:00:43 +08:00
committed by GitHub
parent 55c863ae2f
commit 742837ce56
12 changed files with 681 additions and 171 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
agent/component/llm.py
View File

@@ -88,10 +88,15 @@ class LLM(ComponentBase):
def __init__(self, canvas, component_id, param: ComponentParamBase):
super().__init__(canvas, component_id, param)
model_types = resolve_model_type(self._canvas.get_tenant_id(), self._param.llm_id)
model_type = "chat" if "chat" in model_types else model_types[0]
chat_model_config = resolve_model_config(self._canvas.get_tenant_id(), model_type, self._param.llm_id)
self.chat_mdl = LLMBundle(self._canvas.get_tenant_id(), chat_model_config, max_retries=self._param.max_retries, retry_interval=self._param.delay_after_error)
try:
model_types = resolve_model_type(self._canvas.get_tenant_id(), self._param.llm_id)
model_type = "chat" if "chat" in model_types else model_types[0]
chat_model_config = resolve_model_config(self._canvas.get_tenant_id(), model_type, self._param.llm_id)
self.chat_mdl = LLMBundle(self._canvas.get_tenant_id(), chat_model_config, max_retries=self._param.max_retries, retry_interval=self._param.delay_after_error)
except Exception as e:
logging.warning(f"Fail to load LLM configuration for component. {e}")
self.chat_mdl = None
self.imgs = []
def get_input_form(self) -> dict[str, dict]: