Spaces:
Sleeping
Sleeping
Removed history
Browse files
app.py
CHANGED
|
@@ -51,7 +51,7 @@ llm = pipeline(
|
|
| 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,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
|
| 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
|