Spaces:
Running
Running
Update app.py
Browse files
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 |
|