Krika commited on
Commit
ab389eb
·
1 Parent(s): edaee4b

Update api.py

Browse files
Files changed (1) hide show
  1. 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 = 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}))
 
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}))