fix: improve AI coach chat context, 128k window, and output limits

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-06-11 00:35:01 +08:00
parent 6977bce64f
commit 0e2e360ccf
6 changed files with 124 additions and 25 deletions
@@ -22,6 +22,12 @@ def generate_text(
    user: str,
    temperature: float,
    images_b64: Optional[Sequence[str]] = None,
+    max_tokens: int | None = None,
 ) -> str:
    prompt = f"{system.strip()}\n\n---\n\n{user.strip()}"
-    return ai_generate(prompt, temperature=temperature, images_b64=images_b64)
+    return ai_generate(
+        prompt,
+        temperature=temperature,
+        images_b64=images_b64,
+        max_tokens=max_tokens,
+    )