fix/various fixes

api/app/agents/memoir/extraction_agent.py

@@ -9,11 +9,11 @@ import json
 from dataclasses import dataclass
 from typing import Any, Dict
 
+from app.agents.memoir.prompts import get_state_extraction_prompt
+from app.core.langchain_llm import bind_json_object_mode
 from app.core.logging import get_logger
 from app.features.memoir.memoir_images.json_payload import extract_json_payload
 
-from app.agents.memoir.prompts import get_state_extraction_prompt
-
 logger = get_logger(__name__)
 
 
@@ -56,10 +56,7 @@ class ExtractionAgent:
                     for k, v in (stage_slots or {}).items()
                 },
             )
-            json_llm = llm.bind(
-                model_kwargs={"response_format": {"type": "json_object"}},
-                max_tokens=1024,
-            )
+            json_llm = bind_json_object_mode(llm, max_tokens=1024)
             response = json_llm.invoke(prompt)
             parsed = json.loads(extract_json_payload(response.content))
             detected_stage = parsed.get("detected_stage", detected_stage)