karthikvarunn commited on
Commit
c1dfb09
·
verified ·
1 Parent(s): 88244a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ embeddings = VoyageAIEmbeddings(voyage_api_key=voyage_api_key, model="voyage-law
26
  # 🔹 Query Expansion using GPT-4
27
  def expand_query(query):
28
  llm = ChatOpenAI(model="gpt-4", openai_api_key=openai.api_key, temperature=0.3)
29
- prompt = f"Rewrite this vague query for searching a document into a more specific one:\nQuery: {query}\nSpecific Query:"
30
  refined_query = llm([HumanMessage(content=prompt)]).content.strip()
31
  return refined_query if refined_query else query
32
 
 
26
  # 🔹 Query Expansion using GPT-4
27
  def expand_query(query):
28
  llm = ChatOpenAI(model="gpt-4", openai_api_key=openai.api_key, temperature=0.3)
29
+ prompt = f"Rewrite this vague query for searching a document into a more specific one:\nQuery: {query}\nSpecific Query: If it is just 'docs'/'documents' do not rewrite"
30
  refined_query = llm([HumanMessage(content=prompt)]).content.strip()
31
  return refined_query if refined_query else query
32