File size: 351 Bytes
ea99abb |
1 2 3 4 5 6 7 8 |
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", "")
|