Update api.py
Browse files- utils/api.py +1 -1
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}))
|
|
|
|
| 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}))
|