pvanand commited on
Commit
2aef098
·
verified ·
1 Parent(s): 9a796f1

Update document_generator_v2.py

Browse files
Files changed (1) hide show
  1. 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
- # documents_context=documents_list
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: