Merge branch 'main' of https://github.com/Alalalallalalalalalalalal/Frontend
Browse files- 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 =
|
| 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=
|
| 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", "")
|