Commit 8763c3f1 authored by Kantz's avatar Kantz
Browse files

Quellenangabe bei Task nur wenn Thema gesetzt

parent fc169f78
......@@ -54,10 +54,10 @@ def _ensure_context_task_fields(state: base.ChatState, query_text: str) -> tuple
return selected_file_id, selected_task_id
def _retrieve_context_for_task(state: base.ChatState, query_text: str) -> None:
def _retrieve_context_for_task(state: base.ChatState, query_text: str) -> int:
refs = task_catalog.get_selected_task_subsection_refs(state.sheet)
if not refs:
return
return 0
def _retrieve() -> dict:
sources = retrieval_store.retrieve_for_subsections(
......@@ -70,7 +70,7 @@ def _retrieve_context_for_task(state: base.ChatState, query_text: str) -> None:
"source_count": len(sources),
}
base.log_timed_call(
result = base.log_timed_call(
state.tool_log,
"retrieve_context_with_task_subsections",
{
......@@ -79,17 +79,20 @@ def _retrieve_context_for_task(state: base.ChatState, query_text: str) -> None:
},
_retrieve,
)
return int(result.get("source_count", 0))
# beim ersten Aufruf
def _on_bootstrap(state: base.ChatState, query_text: str) -> None:
source_count = _retrieve_context_for_task(state, query_text)
print(f"_retrieve_context_for_task source_count={source_count}")
if source_count > 0:
_ensure_context_task_fields(state, query_text)
task_text = context_store.context_store_new.get_task(state.sheet).strip()
retrieval_query = query_text
if task_text:
retrieval_query = f"Aufgabe:\n{task_text}\n\n{query_text}"
base.bootstrap_retrieval(state.sheet, retrieval_query, state.tool_log)
_retrieve_context_for_task(state, query_text)
# Sonstiges bei jedem Aufruf
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment