From f61213cd3f6ae9dfd7f41e0894db51ba3062ebe5 Mon Sep 17 00:00:00 2001 From: Tianqi Xu <40522713+dandansamax@users.noreply.github.com> Date: Wed, 18 Sep 2024 23:22:07 +0300 Subject: [PATCH] Update crab/agents/backend_models/gemini_model.py Co-authored-by: Isaac Jin --- crab/agents/backend_models/gemini_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/crab/agents/backend_models/gemini_model.py b/crab/agents/backend_models/gemini_model.py index 4a25b56..c297410 100644 --- a/crab/agents/backend_models/gemini_model.py +++ b/crab/agents/backend_models/gemini_model.py @@ -168,7 +168,7 @@ def _convert_action_to_schema(action_space: list[Action] | None) -> list[Tool] | return actions -def _clear_schema(schema_dict: dict): +def _clear_schema(schema_dict: dict) -> None: schema_dict.pop("title", None) p_type = schema_dict.pop("type", None) for prop in schema_dict.get("properties", {}).values():