feat: 引入deepseek结构化输出

2026-03-19 11:27:43 +08:00
parent f3629efec3
commit 4c75c6f4f4
10 changed files with 74 additions and 26 deletions
--- a/api/app/agents/memoir/extraction_agent.py
+++ b/api/app/agents/memoir/extraction_agent.py
@@ -52,7 +52,11 @@ class ExtractionAgent:
                    for k, v in (stage_slots or {}).items()
                },
            )
-            response = llm.invoke(prompt)
+            json_llm = llm.bind(
+                model_kwargs={"response_format": {"type": "json_object"}},
+                max_tokens=1024,
+            )
+            response = json_llm.invoke(prompt)
            parsed = json.loads(extract_json_payload(response.content))
            detected_stage = parsed.get("detected_stage", detected_stage)
            raw_slots = parsed.get("slots", {}) or {}