Spaces:
Running
Running
Update document_generator_v2.py
Browse files- document_generator_v2.py +2 -2
document_generator_v2.py
CHANGED
@@ -478,6 +478,7 @@ async def load_documents(documents: List[UploadFile]) -> List[str]:
|
|
478 |
|
479 |
try:
|
480 |
# Use LlamaParse to extract content
|
|
|
481 |
parsed_documents = await parser.aload_data(temp_files)
|
482 |
documents_list = [doc.text for doc in parsed_documents]
|
483 |
return documents_list
|
@@ -623,7 +624,6 @@ async def generate_document_outline_endpoint(
|
|
623 |
documents_list = await load_documents(documents)
|
624 |
# Store the documents_list in the cache
|
625 |
docs_cache_key = f"documents_list_{conversation_id}"
|
626 |
-
print("saving document as cache key:",f"documents_list_{conversation_id}")
|
627 |
await FastAPICache.get_backend().set(docs_cache_key, documents_list, expire=3600) # Cache for 1 hour
|
628 |
|
629 |
# Generate document outline
|
@@ -631,7 +631,7 @@ async def generate_document_outline_endpoint(
|
|
631 |
query,
|
632 |
template,
|
633 |
image_context=image_context,
|
634 |
-
|
635 |
)
|
636 |
|
637 |
if json_document is None:
|
|
|
478 |
|
479 |
try:
|
480 |
# Use LlamaParse to extract content
|
481 |
+
print(f"processing files {str(temp_files)}")
|
482 |
parsed_documents = await parser.aload_data(temp_files)
|
483 |
documents_list = [doc.text for doc in parsed_documents]
|
484 |
return documents_list
|
|
|
624 |
documents_list = await load_documents(documents)
|
625 |
# Store the documents_list in the cache
|
626 |
docs_cache_key = f"documents_list_{conversation_id}"
|
|
|
627 |
await FastAPICache.get_backend().set(docs_cache_key, documents_list, expire=3600) # Cache for 1 hour
|
628 |
|
629 |
# Generate document outline
|
|
|
631 |
query,
|
632 |
template,
|
633 |
image_context=image_context,
|
634 |
+
documents_context=documents_list
|
635 |
)
|
636 |
|
637 |
if json_document is None:
|