lanna_lalala;- commited on
Commit
de0ba00
·
2 Parent(s): fa3f286ab389eb

Merge branch 'main' of https://github.com/Alalalallalalalalalalalal/Frontend

Browse files
Files changed (1) hide show
  1. utils/api.py +2 -2
utils/api.py CHANGED
@@ -321,7 +321,7 @@ def record_profit_puzzler_play(user_id: int, scenario_id: str, title: str,
321
  return _json_or_raise(_req("POST", "/games/profit_puzzler/record", json=payload))
322
 
323
  # ---------- RAG passthrough ----------
324
- def retrieve(query: str, lesson_id: int, level_slug: str = "beginner", k: int = 5) -> str:
325
  try:
326
  d = _json_or_raise(_req("GET", "/retrieve",
327
  params={"query": query, "lesson_id": lesson_id, "level_slug": level_slug, "k": k}))
@@ -339,7 +339,7 @@ def chat_ai(query: str, lesson_id: int, level_slug: str, history=None) -> str:
339
  try:
340
  # Force the full path without relying on _prefixes
341
  url = f"{BACKEND}/chat" # make sure BACKEND has no trailing slash
342
- r = _session.post(url, json=payload, timeout=DEFAULT_TIMEOUT)
343
  r.raise_for_status()
344
  d = r.json()
345
  return d.get("answer", "")
 
321
  return _json_or_raise(_req("POST", "/games/profit_puzzler/record", json=payload))
322
 
323
  # ---------- RAG passthrough ----------
324
+ def retrieve(query: str, lesson_id: int, level_slug: str = "beginner", k: int = 3) -> str:
325
  try:
326
  d = _json_or_raise(_req("GET", "/retrieve",
327
  params={"query": query, "lesson_id": lesson_id, "level_slug": level_slug, "k": k}))
 
339
  try:
340
  # Force the full path without relying on _prefixes
341
  url = f"{BACKEND}/chat" # make sure BACKEND has no trailing slash
342
+ r = _session.post(url, json=payload, timeout=120)
343
  r.raise_for_status()
344
  d = r.json()
345
  return d.get("answer", "")