modify ag_retrieve
This commit is contained in:
parent
b618cb12d2
commit
ecf332add5
@ -314,7 +314,12 @@ async def handle_request(request: Dict[str, Any]) -> Dict[str, Any]:
|
||||
top_k = arguments.get("top_k", 100)
|
||||
|
||||
if not query:
|
||||
return create_error_response(request_id, -32602, "Missing required parameter: query")
|
||||
return create_success_response(request_id, {
|
||||
"content": [{
|
||||
"type": "text",
|
||||
"text": "Error: missing required parameter 'query'. Please call this tool again with a non-empty 'query' argument describing what you want to retrieve."
|
||||
}]
|
||||
})
|
||||
|
||||
result = rag_retrieve(query, top_k, trace_id)
|
||||
|
||||
@ -328,7 +333,12 @@ async def handle_request(request: Dict[str, Any]) -> Dict[str, Any]:
|
||||
query = arguments.get("query", "")
|
||||
|
||||
if not query:
|
||||
return create_error_response(request_id, -32602, "Missing required parameter: query")
|
||||
return create_success_response(request_id, {
|
||||
"content": [{
|
||||
"type": "text",
|
||||
"text": "Error: missing required parameter 'query'. Please call this tool again with a non-empty 'query' argument describing what you want to retrieve."
|
||||
}]
|
||||
})
|
||||
|
||||
result = table_rag_retrieve(query, trace_id)
|
||||
|
||||
|
||||
@ -314,7 +314,12 @@ async def handle_request(request: Dict[str, Any]) -> Dict[str, Any]:
|
||||
top_k = arguments.get("top_k", 100)
|
||||
|
||||
if not query:
|
||||
return create_error_response(request_id, -32602, "Missing required parameter: query")
|
||||
return create_success_response(request_id, {
|
||||
"content": [{
|
||||
"type": "text",
|
||||
"text": "Error: missing required parameter 'query'. Please call this tool again with a non-empty 'query' argument describing what you want to retrieve."
|
||||
}]
|
||||
})
|
||||
|
||||
result = rag_retrieve(query, top_k, trace_id)
|
||||
|
||||
@ -328,7 +333,12 @@ async def handle_request(request: Dict[str, Any]) -> Dict[str, Any]:
|
||||
query = arguments.get("query", "")
|
||||
|
||||
if not query:
|
||||
return create_error_response(request_id, -32602, "Missing required parameter: query")
|
||||
return create_success_response(request_id, {
|
||||
"content": [{
|
||||
"type": "text",
|
||||
"text": "Error: missing required parameter 'query'. Please call this tool again with a non-empty 'query' argument describing what you want to retrieve."
|
||||
}]
|
||||
})
|
||||
|
||||
result = table_rag_retrieve(query, trace_id)
|
||||
|
||||
|
||||
@ -167,7 +167,12 @@ async def handle_request(request: Dict[str, Any]) -> Dict[str, Any]:
|
||||
top_k = arguments.get("top_k", 100)
|
||||
|
||||
if not query:
|
||||
return create_error_response(request_id, -32602, "Missing required parameter: query")
|
||||
return create_success_response(request_id, {
|
||||
"content": [{
|
||||
"type": "text",
|
||||
"text": "Error: missing required parameter 'query'. Please call this tool again with a non-empty 'query' argument describing what you want to retrieve."
|
||||
}]
|
||||
})
|
||||
|
||||
result = rag_retrieve(query, top_k)
|
||||
|
||||
|
||||
@ -193,7 +193,12 @@ async def handle_request(request: Dict[str, Any]) -> Dict[str, Any]:
|
||||
top_k = arguments.get("top_k", 100)
|
||||
|
||||
if not query:
|
||||
return create_error_response(request_id, -32602, "Missing required parameter: query")
|
||||
return create_success_response(request_id, {
|
||||
"content": [{
|
||||
"type": "text",
|
||||
"text": "Error: missing required parameter 'query'. Please call this tool again with a non-empty 'query' argument describing what you want to retrieve."
|
||||
}]
|
||||
})
|
||||
|
||||
result = rag_retrieve(query, top_k)
|
||||
|
||||
|
||||
@ -193,7 +193,12 @@ async def handle_request(request: Dict[str, Any]) -> Dict[str, Any]:
|
||||
top_k = arguments.get("top_k", 100)
|
||||
|
||||
if not query:
|
||||
return create_error_response(request_id, -32602, "Missing required parameter: query")
|
||||
return create_success_response(request_id, {
|
||||
"content": [{
|
||||
"type": "text",
|
||||
"text": "Error: missing required parameter 'query'. Please call this tool again with a non-empty 'query' argument describing what you want to retrieve."
|
||||
}]
|
||||
})
|
||||
|
||||
result = rag_retrieve(query, top_k)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user