fix migration

This commit is contained in:
2026-06-16 09:59:39 +03:00
parent 2f892bef19
commit c69186c13b
5 changed files with 50 additions and 23 deletions
+28 -21
View File
@@ -5,10 +5,11 @@ from typing import Any
from sqlalchemy.orm import Session
from app.config import get_settings
from app.db.base import SessionLocal
from app.llm.client import LLMClient
from app.memory.service import MemoryService
from app.projects.structuring import strip_markdown_json
from app.settings.service import SettingsService
logger = logging.getLogger(__name__)
@@ -62,28 +63,34 @@ async def _call_extractor(
*[f"- {f.get('content')}" for f in facts[:30]],
]
settings = get_settings()
extract_model = settings.memory_extract_model.strip() or None
db = SessionLocal()
try:
extract_model = str(SettingsService(db).get_effective("memory_extract_model")).strip() or None
finally:
db.close()
llm = LLMClient()
result = await llm.complete(
[
{"role": "system", "content": EXTRACTION_PROMPT},
{
"role": "user",
"content": (
"\n".join(known)
+ "\n\n---\nДиалог:\nПользователь: "
+ user_text
+ "\nАссистент: "
+ assistant_text[:1500]
),
},
],
temperature=0.2,
model=extract_model,
for_extraction=True,
)
try:
result = await llm.complete(
[
{"role": "system", "content": EXTRACTION_PROMPT},
{
"role": "user",
"content": (
"\n".join(known)
+ "\n\n---\nДиалог:\nПользователь: "
+ user_text
+ "\nАссистент: "
+ assistant_text[:1500]
),
},
],
temperature=0.2,
model=extract_model,
for_extraction=True,
)
finally:
await llm.aclose()
raw = strip_markdown_json(result.get("content") or "")
if not raw:
return {"facts": [], "profile": {}}