bupa1018 commited on
Commit
e2aed95
·
1 Parent(s): d68d26f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -14
app.py CHANGED
@@ -313,7 +313,7 @@ def format_kadi_api_doc_context(docs):
313
 
314
  for doc in docs:
315
  source_info = doc.metadata.get("source", "Unknown Type")
316
- print("YYYYYYYEEEEEEEEEEEEEEE:}\n\n", doc.pagecontent)
317
  formatted_doc = f"# source: {source_info}\n{doc.page_content}\n\n\n"
318
  doc_context.append(formatted_doc)
319
 
@@ -326,21 +326,9 @@ def rag_workflow(query):
326
  # retrieved_doc_chunks = retrieve_from_vectorstore (docstore, query, k=5)
327
  # retrieved_code_chunks = retrieve_from_vectorstore(codestore, query, k=5)
328
 
329
- # docs = retrieve_docs_from_vectorstore(docstore, query, k=5)
330
 
331
 
332
- kadi_apy_docs = retrieve_from_vectorstore(vector_store, query, k = 5)
333
- #kadi_apy_docs = retrieve_within_kadiApy_docs (vector_store, query, k = 5)
334
-
335
-
336
-
337
-
338
- # Print each document's content and metadata
339
- for doc in kadi_apy_docs:
340
- print("Content:", doc.page_content)
341
- print("Metadata:", doc.metadata)
342
- print("-" * 20)
343
-
344
  kadi_apy_library_docs = retrieve_within_kadiApy_library (vector_store, query, k = 10)
345
 
346
  doc_context = format_kadi_api_doc_context(kadi_apy_docs)
 
313
 
314
  for doc in docs:
315
  source_info = doc.metadata.get("source", "Unknown Type")
316
+ print("YYYYYYYEEEEEEEEEEEEEEE:}\n\n", doc.page_content)
317
  formatted_doc = f"# source: {source_info}\n{doc.page_content}\n\n\n"
318
  doc_context.append(formatted_doc)
319
 
 
326
  # retrieved_doc_chunks = retrieve_from_vectorstore (docstore, query, k=5)
327
  # retrieved_code_chunks = retrieve_from_vectorstore(codestore, query, k=5)
328
 
 
329
 
330
 
331
+ kadi_apy_docs = retrieve_within_kadiApy_docs (vector_store, query, k = 5)
 
 
 
 
 
 
 
 
 
 
 
332
  kadi_apy_library_docs = retrieve_within_kadiApy_library (vector_store, query, k = 10)
333
 
334
  doc_context = format_kadi_api_doc_context(kadi_apy_docs)