timothytzkung commited on
Commit
f129043
·
verified ·
1 Parent(s): 049816d

Removed history

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,7 +51,7 @@ llm = pipeline(
51
  token=hf_token
52
  )
53
 
54
- def clean_query_with_llm(query, history):
55
  prompt_content = f"""
56
  Below is a new question asked by the user that needs to be answered by searching in a knowledge base.
57
  You have access to SFU IT Knowledge Base index with 100's of chunked documents.
@@ -99,7 +99,7 @@ def generate_with_rag(query, top_k=5):
99
  query = query.replace("gosfu", "goSFU")
100
 
101
  # Retrieve
102
- search_query = clean_query_with_llm(query, history)
103
  results = retrieve_with_pandas(search_query)
104
 
105
  # Turn the Series into a single string of text
 
51
  token=hf_token
52
  )
53
 
54
+ def clean_query_with_llm(query):
55
  prompt_content = f"""
56
  Below is a new question asked by the user that needs to be answered by searching in a knowledge base.
57
  You have access to SFU IT Knowledge Base index with 100's of chunked documents.
 
99
  query = query.replace("gosfu", "goSFU")
100
 
101
  # Retrieve
102
+ search_query = clean_query_with_llm(query)
103
  results = retrieve_with_pandas(search_query)
104
 
105
  # Turn the Series into a single string of text