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

Update embed model handling in local_rag.py

Browse Source
This commit is contained in:
Victor Giers
2026-03-19 22:34:26 +01:00
parent aff44c76b8
commit 49f6b14c24
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
backend/local_rag.py
View File

@@ -835,6 +835,7 @@ def _run_prepare_pipeline(slug: str, on_progress=None, **opts):
overlap_chars=opts.get("overlap_chars", 200), overlap_chars=opts.get("overlap_chars", 200),
concurrency=opts.get("concurrency", 6), concurrency=opts.get("concurrency", 6),
) )
_set_pipeline_embed_model(slug, results["embed"].get("embed_model") if isinstance(results.get("embed"), dict) else None)
_mark_pipeline_stage(slug, "embed", prepare_signature) _mark_pipeline_stage(slug, "embed", prepare_signature)
if on_progress: if on_progress:
@@ -872,6 +873,10 @@ async def _run_job(job_id: str, fn_name: str, **kwargs):
call = functools.partial(runner, on_progress=on_progress, **kwargs) call = functools.partial(runner, on_progress=on_progress, **kwargs)
result = await loop.run_in_executor(JOB_EXECUTOR, call) result = await loop.run_in_executor(JOB_EXECUTOR, call)
if fn_name in {"embed", "prepare"} and isinstance(result, dict):
embed_model = result.get("embed_model")
if embed_model:
_set_pipeline_embed_model(job["slug"], embed_model)
if fn_name in {"build", "enrich", "embed"} and stage_signature: if fn_name in {"build", "enrich", "embed"} and stage_signature:
_mark_pipeline_stage(job["slug"], fn_name, stage_signature) _mark_pipeline_stage(job["slug"], fn_name, stage_signature)
job["status"] = "succeeded" job["status"] = "succeeded"