Daniel Foley commited on
Commit
bedac8d
·
1 Parent(s): d1935d1

Usage Monitoring

Browse files
Files changed (1) hide show
  1. RAG.py +1 -1
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"Old_Query: {query},New_Query: {new_query}")
155
 
156
  retrieved, _ = retrieve(query=query, vectorstore=vectorstore, k=k)
157
  if not retrieved:
 
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}\n---")
155
 
156
  retrieved, _ = retrieve(query=query, vectorstore=vectorstore, k=k)
157
  if not retrieved: