abhishekbhakat/airflow-wingman
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix tools schema conversion for openai

Browse Source
This commit is contained in:
abhishekbhakat
2025-03-02 19:24:46 +00:00
parent 02a6f89bc7
commit dfa5e8c25d
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/airflow_wingman/tools/conversion.py
View File

@@ -66,6 +66,15 @@ def convert_to_openai_tools(airflow_tools: list) -> list:
# Add default value if available # Add default value if available
if "default" in param_info and param_info["default"] is not None: if "default" in param_info and param_info["default"] is not None:
param_def["default"] = param_info["default"] param_def["default"] = param_info["default"]
# Add items property for array types
if param_def.get("type") == "array" and "items" not in param_def:
# If items is defined in the original schema, use it
if "items" in param_info:
param_def["items"] = param_info["items"]
else:
# Otherwise, default to string items
param_def["items"] = {"type": "string"}
# Add to properties # Add to properties
openai_tool["function"]["parameters"]["properties"][param_name] = param_def openai_tool["function"]["parameters"]["properties"][param_name] = param_def
@@ -141,3 +150,7 @@ def _handle_schema_construct(param_def: dict[str, Any], param_info: dict[str, An
# If no type was found, default to string # If no type was found, default to string
if "type" not in param_def: if "type" not in param_def:
param_def["type"] = "string" param_def["type"] = "string"
# Add items property for array types
if param_def.get("type") == "array" and "items" not in param_def:
param_def["items"] = {"type": "string"}