Skip to content

Commit f4c8f93

Browse files
committed
fix tool code generation in case of multiline descriptions
1 parent a17f915 commit f4c8f93

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/smolagents/tools.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -235,10 +235,10 @@ def save(self, output_dir):
235235
from typing import Optional
236236
237237
class {class_name}(Tool):
238-
name = "{self.name}"
239-
description = "{self.description}"
238+
name = {json.dumps(self.name)}
239+
description = {json.dumps(textwrap.dedent(self.description).strip())}
240240
inputs = {json.dumps(self.inputs, separators=(",", ":"))}
241-
output_type = "{self.output_type}"
241+
output_type = {json.dumps(self.output_type)}
242242
"""
243243
).strip()
244244
import re

0 commit comments

Comments
 (0)