Spaces:
Runtime error
Runtime error
updated monitoring
Browse files
RAG.py
CHANGED
|
@@ -151,7 +151,7 @@ def RAG(llm: Any, query: str,vectorstore:PineconeVectorStore, top: int = 10, k:
|
|
| 151 |
# query_prompt = query_template.invoke({"query":query})
|
| 152 |
# query_response = llm.invoke(query_prompt)
|
| 153 |
# new_query = parse_xml_and_query(query=query,xml_string=query_response.content)
|
| 154 |
-
logging.info(f"---\nQUERY: {query}
|
| 155 |
|
| 156 |
retrieved, _ = retrieve(query=query, vectorstore=vectorstore, k=k)
|
| 157 |
if not retrieved:
|
|
@@ -196,7 +196,8 @@ def RAG(llm: Any, query: str,vectorstore:PineconeVectorStore, top: int = 10, k:
|
|
| 196 |
|
| 197 |
# Parse and return response
|
| 198 |
parsed = parse_xml_and_check(response.content)
|
| 199 |
-
logging.info(f"
|
|
|
|
| 200 |
return parsed, reranked
|
| 201 |
|
| 202 |
except Exception as e:
|
|
|
|
| 151 |
# query_prompt = query_template.invoke({"query":query})
|
| 152 |
# query_response = llm.invoke(query_prompt)
|
| 153 |
# new_query = parse_xml_and_query(query=query,xml_string=query_response.content)
|
| 154 |
+
logging.info(f"\n---\nQUERY: {query}")
|
| 155 |
|
| 156 |
retrieved, _ = retrieve(query=query, vectorstore=vectorstore, k=k)
|
| 157 |
if not retrieved:
|
|
|
|
| 196 |
|
| 197 |
# Parse and return response
|
| 198 |
parsed = parse_xml_and_check(response.content)
|
| 199 |
+
logging.info(f"RESPONSE: {parsed}\nRETRIEVED: {reranked}")
|
| 200 |
+
logging.info(f"RAG Finished: {time.time()-start}\n---\n")
|
| 201 |
return parsed, reranked
|
| 202 |
|
| 203 |
except Exception as e:
|