fix: patch failing gemini unit tests (#786)

This commit is contained in:
Charles Packer
2025-01-26 15:52:02 -08:00
committed by GitHub
parent 2b8023dd7b
commit 4a4f734f1a
2 changed files with 24 additions and 12 deletions

View File

@@ -437,6 +437,7 @@ def generate_tool_schema_for_composio(
name: str,
description: str,
append_heartbeat: bool = True,
strict: bool = False,
) -> Dict[str, Any]:
properties_json = {}
required_fields = parameters_model.required or []
@@ -473,14 +474,26 @@ def generate_tool_schema_for_composio(
required_fields.append("request_heartbeat")
# Return the final schema
return {
"name": name,
"description": description,
"strict": True,
"parameters": {
"type": "object",
"properties": properties_json,
"additionalProperties": False,
"required": required_fields,
},
}
if strict:
# https://platform.openai.com/docs/guides/function-calling#strict-mode
return {
"name": name,
"description": description,
"strict": True, # NOTE
"parameters": {
"type": "object",
"properties": properties_json,
"additionalProperties": False, # NOTE
"required": required_fields,
},
}
else:
return {
"name": name,
"description": description,
"parameters": {
"type": "object",
"properties": properties_json,
"required": required_fields,
},
}

View File

@@ -191,7 +191,6 @@ def compile_system_message(
# render the variables using the built-in templater
try:
formatted_prompt = safe_format(system_prompt, variables)
print(f"Formatted system prompt:\n{formatted_prompt}")
except Exception as e:
raise ValueError(f"Failed to format system prompt - {str(e)}. System prompt value:\n{system_prompt}")