import json, openai, requests
def search_documents(query: str) -> list:
resp = requests.post(
f"https://api.mixpeek.com/v1/retrievers/{RETRIEVER_ID}/execute",
headers={"Authorization": f"Bearer {API_KEY}", "X-Namespace": NS_ID, "Content-Type": "application/json"},
json={"inputs": {"query_text": query}, "limit": 5},
)
return resp.json()["results"]
retriever_tool = {
"type": "function",
"function": {
"name": "search_documents",
"description": "Search the knowledge base using multimodal retrieval",
"parameters": {
"type": "object",
"properties": {"query": {"type": "string", "description": "Natural language search query"}},
"required": ["query"],
},
},
}