comment
Browse files- model/custom_agent.py +3 -3
model/custom_agent.py
CHANGED
|
@@ -137,8 +137,8 @@ class CustomHfAgent(Agent):
|
|
| 137 |
"content": prompt,
|
| 138 |
}
|
| 139 |
]
|
| 140 |
-
|
| 141 |
-
|
| 142 |
else:
|
| 143 |
prompt = self.run_prompt_template.replace("<<all_tools>>", description)
|
| 144 |
prompt = prompt.replace("<<prompt>>", task)
|
|
@@ -148,5 +148,5 @@ class CustomHfAgent(Agent):
|
|
| 148 |
"content": prompt,
|
| 149 |
}
|
| 150 |
]
|
| 151 |
-
|
| 152 |
return prompt
|
|
|
|
| 137 |
"content": prompt,
|
| 138 |
}
|
| 139 |
]
|
| 140 |
+
# prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
|
| 141 |
+
## prompt
|
| 142 |
else:
|
| 143 |
prompt = self.run_prompt_template.replace("<<all_tools>>", description)
|
| 144 |
prompt = prompt.replace("<<prompt>>", task)
|
|
|
|
| 148 |
"content": prompt,
|
| 149 |
}
|
| 150 |
]
|
| 151 |
+
# prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
|
| 152 |
return prompt
|