Revert tool validation

This commit is contained in:
Brandon Hancock
2025-01-09 16:55:23 -05:00
parent dfb2b4b47a
commit b1960ecb4c
2 changed files with 9 additions and 4 deletions

View File

@@ -409,9 +409,15 @@ class ToolUsage:
def _validate_tool_input(self, tool_input: str) -> Dict[str, Any]:
try:
# Replace 'None', 'True', 'False' with their JSON equivalents
tool_input = tool_input.replace('"None"', "null")
tool_input = tool_input.replace("True", "true").replace("False", "false")
# Replace Python literals with JSON equivalents
replacements = {
r"'": '"',
r"None": "null",
r"True": "true",
r"False": "false",
}
for pattern, replacement in replacements.items():
tool_input = re.sub(pattern, replacement, tool_input)
arguments = json.loads(tool_input)
except json.JSONDecodeError:

View File

@@ -12,7 +12,6 @@ class InternalAgentTool(BaseAgentTool):
return "Test response"
# new comment
@pytest.mark.parametrize(
"role_name,should_match",
[