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 ...@@ -54,10 +54,10 @@ def _ensure_context_task_fields(state: base.ChatState, query_text: str) -> tuple
return selected_file_id, selected_task_id 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) refs = task_catalog.get_selected_task_subsection_refs(state.sheet)
if not refs: if not refs:
return return 0
def _retrieve() -> dict: def _retrieve() -> dict:
sources = retrieval_store.retrieve_for_subsections( sources = retrieval_store.retrieve_for_subsections(
...@@ -70,7 +70,7 @@ def _retrieve_context_for_task(state: base.ChatState, query_text: str) -> None: ...@@ -70,7 +70,7 @@ def _retrieve_context_for_task(state: base.ChatState, query_text: str) -> None:
"source_count": len(sources), "source_count": len(sources),
} }
base.log_timed_call( result = base.log_timed_call(
state.tool_log, state.tool_log,
"retrieve_context_with_task_subsections", "retrieve_context_with_task_subsections",
{ {
...@@ -79,17 +79,20 @@ def _retrieve_context_for_task(state: base.ChatState, query_text: str) -> None: ...@@ -79,17 +79,20 @@ def _retrieve_context_for_task(state: base.ChatState, query_text: str) -> None:
}, },
_retrieve, _retrieve,
) )
return int(result.get("source_count", 0))
# beim ersten Aufruf # beim ersten Aufruf
def _on_bootstrap(state: base.ChatState, query_text: str) -> None: def _on_bootstrap(state: base.ChatState, query_text: str) -> None:
_ensure_context_task_fields(state, query_text) source_count = _retrieve_context_for_task(state, query_text)
task_text = context_store.context_store_new.get_task(state.sheet).strip() print(f"_retrieve_context_for_task source_count={source_count}")
retrieval_query = query_text if source_count > 0:
if task_text: _ensure_context_task_fields(state, query_text)
retrieval_query = f"Aufgabe:\n{task_text}\n\n{query_text}" task_text = context_store.context_store_new.get_task(state.sheet).strip()
base.bootstrap_retrieval(state.sheet, retrieval_query, state.tool_log) retrieval_query = query_text
_retrieve_context_for_task(state, 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)
# Sonstiges bei jedem Aufruf # 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