Spaces:
Sleeping
Sleeping
Update server.py
Browse files
server.py
CHANGED
|
@@ -59,11 +59,11 @@ for source, rel_path in SOURCES.items():
|
|
| 59 |
mime_type="application/json",
|
| 60 |
tags={"llm", "retrieval"}
|
| 61 |
)
|
| 62 |
-
def retrieve_docs(fuente: str = 'oms',
|
| 63 |
"""
|
| 64 |
Parameters:
|
| 65 |
-
fuente: Source name (default: oms).
|
| 66 |
query: Search query (required).
|
|
|
|
| 67 |
top_k: Number of results to return (default: 3).
|
| 68 |
"""
|
| 69 |
if fuente not in indices:
|
|
|
|
| 59 |
mime_type="application/json",
|
| 60 |
tags={"llm", "retrieval"}
|
| 61 |
)
|
| 62 |
+
def retrieve_docs(query: str, fuente: str = 'oms', top_k: int = 3) -> dict:
|
| 63 |
"""
|
| 64 |
Parameters:
|
|
|
|
| 65 |
query: Search query (required).
|
| 66 |
+
fuente: Source name (default: oms).
|
| 67 |
top_k: Number of results to return (default: 3).
|
| 68 |
"""
|
| 69 |
if fuente not in indices:
|