Ling / tasks /retrieval.py
Nam Fam
update files
ea99abb
raw
history blame contribute delete
351 Bytes
from utils.remote_client import execute_remote_task
def text_retrieval(query: str, model: str, documents: list) -> str:
resp = execute_remote_task("retrieval", {"query": query, "model": model, "documents": documents})
if "error" in resp:
return "Oops! Something went wrong. Please try again later."
return resp.get("matches", "")