giers10/Heimgeist
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Refactor RAG and web search logic for improved compatibility checks in runtime

Browse Source
This commit is contained in:
Victor Giers
2026-06-15 22:55:28 +02:00
parent 60b32513cf
commit 5da8b3ba6d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
backend/agent/runtime.py
View File

@@ -240,7 +240,9 @@ class WorkflowRuntime:
run_values=run_values, consume_llm_call=consume_llm_call,
)
prompt = str(workflow_inputs.get("prompt") or "")
if run_values.get("library_slug") and run_values.get("manual_library_enabled") and prompt:
rag_enabled = bool(run_values.get("compatibility_rag_enabled", run_values.get("manual_library_enabled")))
web_enabled = bool(run_values.get("compatibility_web_enabled", run_values.get("web_search_enabled")))
if run_values.get("library_slug") and rag_enabled and prompt:
try:
counters["tool_calls"] += 1
await writer.emit("tool_started", {"tool": "heimgeist.knowledge_search"}, "compatibility")
@@ -252,7 +254,7 @@ class WorkflowRuntime:
await writer.emit("tool_result", {"tool": "heimgeist.knowledge_search", "result": result}, "compatibility")
except Exception as exc:
await writer.emit("tool_result", {"tool": "heimgeist.knowledge_search", "error": str(exc)}, "compatibility")
if run_values.get("web_search_enabled") and prompt:
if web_enabled and prompt:
try:
query_args = {"prompt": prompt, "model": run_values["chat_model"], "messages": run_values.get("messages") or []}
counters["tool_calls"] += 1