Daniel Foley commited on
Commit
fea960e
·
1 Parent(s): 1ba93a0

Updated xml parse call for query alignment

Browse files
Files changed (1) hide show
  1. RAG.py +1 -1
RAG.py CHANGED
@@ -146,7 +146,7 @@ def RAG(llm: Any, query: str, index_name: str, embeddings: Any, top: int = 10, k
146
  )
147
  query_prompt = query_template.invoke({"query":query})
148
  query_response = llm.invoke(query_prompt)
149
- new_query = parse_xml_and_query(query_response.content)
150
 
151
  retrieved, _ = retrieve(index_name=index_name, query=new_query, embeddings=embeddings, k=k)
152
  if not retrieved:
 
146
  )
147
  query_prompt = query_template.invoke({"query":query})
148
  query_response = llm.invoke(query_prompt)
149
+ new_query = parse_xml_and_query(query=query,xml_string=query_response.content)
150
 
151
  retrieved, _ = retrieve(index_name=index_name, query=new_query, embeddings=embeddings, k=k)
152
  if not retrieved: