Davit6174 commited on
Commit
b8d7142
·
verified ·
1 Parent(s): b35d9f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -44,11 +44,11 @@ class ZephyrAPI:
44
  return "\n".join([f"- {tool.name}: {tool.description}" for tool in tools])
45
 
46
  def __call__(self, question: str) -> str:
47
- prompt = f"<|system|>\n"
48
- f"You are a helpful AI agent. You can use the following tools when needed:\n"
49
- f"{self.tool_descriptions}\n"
50
- f"\nRespond to the user. If a tool is needed, use this format:\n"
51
- f"Action: tool_name\nAction Input: input_for_tool\n"
52
  f"\n<|user|>\n{question}\n<|assistant|>\n"
53
  payload = {
54
  "inputs": prompt,
 
44
  return "\n".join([f"- {tool.name}: {tool.description}" for tool in tools])
45
 
46
  def __call__(self, question: str) -> str:
47
+ prompt = f"<|system|>\n"\
48
+ f"You are a helpful AI agent. You can use the following tools when needed:\n"\
49
+ f"{self.tool_descriptions}\n"\
50
+ f"\nRespond to the user. If a tool is needed, use this format:\n"\
51
+ f"Action: tool_name\nAction Input: input_for_tool\n"\
52
  f"\n<|user|>\n{question}\n<|assistant|>\n"
53
  payload = {
54
  "inputs": prompt,