Update app.py
Browse files
app.py
CHANGED
|
@@ -149,13 +149,13 @@ class BasicAgent:
|
|
| 149 |
|
| 150 |
# --- Core dispatcher/fallback ---
|
| 151 |
def __call__(self, question: str) -> str:
|
| 152 |
-
prompt = f"{self.system_prompt}Q: {
|
| 153 |
#out = self.generator(prompt, max_new_tokens=16, return_full_text=False)
|
| 154 |
#return out[0]["generated_text"].strip()
|
| 155 |
agent = AgentExecutor.from_llm_and_tools(
|
| 156 |
llm=self.llm,
|
| 157 |
tools=TOOLS,
|
| 158 |
-
system_message=
|
| 159 |
verbose=True,
|
| 160 |
)
|
| 161 |
#agent = AgentExecutor(agent=self.llm, tools=TOOLS, prompt=prompt, verbose=False, return_intermediate_steps=False)
|
|
|
|
| 149 |
|
| 150 |
# --- Core dispatcher/fallback ---
|
| 151 |
def __call__(self, question: str) -> str:
|
| 152 |
+
prompt = f"{self.system_prompt}Q: {question}\nA:"
|
| 153 |
#out = self.generator(prompt, max_new_tokens=16, return_full_text=False)
|
| 154 |
#return out[0]["generated_text"].strip()
|
| 155 |
agent = AgentExecutor.from_llm_and_tools(
|
| 156 |
llm=self.llm,
|
| 157 |
tools=TOOLS,
|
| 158 |
+
system_message=prompt,
|
| 159 |
verbose=True,
|
| 160 |
)
|
| 161 |
#agent = AgentExecutor(agent=self.llm, tools=TOOLS, prompt=prompt, verbose=False, return_intermediate_steps=False)
|