add table_rag_retrieve
This commit is contained in:
parent
38c5d1e622
commit
f830b8bb08
@ -36,7 +36,7 @@ def rag_retrieve(query: str, top_k: int = 100) -> Dict[str, Any]:
|
|||||||
if len(sys.argv) > 1:
|
if len(sys.argv) > 1:
|
||||||
bot_id = sys.argv[1]
|
bot_id = sys.argv[1]
|
||||||
|
|
||||||
url = f"{BACKEND_HOST}/v1/rag_retrieve/{bot_id}"
|
url = f"{BACKEND_HOST}/v1/table_rag_retrieve/{bot_id}"
|
||||||
if not url:
|
if not url:
|
||||||
return {
|
return {
|
||||||
"content": [
|
"content": [
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user