Update app.py
Browse files
app.py
CHANGED
|
@@ -49,7 +49,7 @@ class Agent1:
|
|
| 49 |
|
| 50 |
return questions
|
| 51 |
|
| 52 |
-
def process(self, user_input: str) -> Dict[str, List[Dict[str, str]]]:
|
| 53 |
queries = self.rephrase_and_split(user_input)
|
| 54 |
print("Identified queries:", queries)
|
| 55 |
results = {}
|
|
@@ -217,7 +217,7 @@ def ask_question(question, temperature, top_p, repetition_penalty, web_search):
|
|
| 217 |
|
| 218 |
model = get_model(temperature, top_p, repetition_penalty)
|
| 219 |
embed = get_embeddings()
|
| 220 |
-
agent1 = Agent1(model
|
| 221 |
|
| 222 |
if os.path.exists("faiss_database"):
|
| 223 |
database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
|
|
|
|
| 49 |
|
| 50 |
return questions
|
| 51 |
|
| 52 |
+
def process(self, user_input: str) -> tuple[List[str], Dict[str, List[Dict[str, str]]]]:
|
| 53 |
queries = self.rephrase_and_split(user_input)
|
| 54 |
print("Identified queries:", queries)
|
| 55 |
results = {}
|
|
|
|
| 217 |
|
| 218 |
model = get_model(temperature, top_p, repetition_penalty)
|
| 219 |
embed = get_embeddings()
|
| 220 |
+
agent1 = Agent1() # Create Agent1 without passing a model
|
| 221 |
|
| 222 |
if os.path.exists("faiss_database"):
|
| 223 |
database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
|