RalphThings commited on
Commit
03d7954
·
verified ·
1 Parent(s): 4ea9079

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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: {q}\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=self.system_prompt,
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)