Spaces:
Running
Running
Daniel Foley
commited on
Commit
·
fea960e
1
Parent(s):
1ba93a0
Updated xml parse call for query alignment
Browse files
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:
|