enh: full context web search

This commit is contained in:
Timothy Jaeryang Baek
2025-02-17 18:14:26 -08:00
parent 16ce8ab16c
commit ca0b7217d2
6 changed files with 93 additions and 29 deletions

View File

@@ -304,7 +304,14 @@ def get_sources_from_files(
relevant_contexts = []
for file in files:
if file.get("context") == "full":
if file.get("docs"):
print("file.get('docs')", file.get("docs"))
context = {
"documents": [[doc.get("content") for doc in file.get("docs")]],
"metadatas": [[doc.get("metadata") for doc in file.get("docs")]],
}
elif file.get("context") == "full":
context = {
"documents": [[file.get("file").get("data", {}).get("content")]],
"metadatas": [[{"file_id": file.get("id"), "name": file.get("name")}]],