262a6f6055
Signed-off-by: minmin-intel <minmin.hou@intel.com> Signed-off-by: Chendi.Xue <chendi.xue@intel.com>
28 lines
653 B
Python
28 lines
653 B
Python
# Copyright (C) 2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import os
|
|
|
|
import requests
|
|
|
|
|
|
def search_knowledge_base(query: str) -> str:
|
|
"""Search the knowledge base for a specific query."""
|
|
url = os.environ.get("RETRIEVAL_TOOL_URL")
|
|
print(url)
|
|
proxies = {"http": ""}
|
|
payload = {
|
|
"text": query,
|
|
}
|
|
response = requests.post(url, json=payload, proxies=proxies)
|
|
print(response)
|
|
docs = response.json()["documents"]
|
|
context = ""
|
|
for i, doc in enumerate(docs):
|
|
if i == 0:
|
|
context = doc
|
|
else:
|
|
context += "\n" + doc
|
|
print(context)
|
|
return context
|