This commit is contained in:
Shuchang Zheng 2025-04-30 00:06:59 +08:00 committed by GitHub
parent f4ea546ae4
commit 53f4c23ef9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions

View file

@ -739,9 +739,9 @@ if settings.ENABLE_NOVITA:
# Set the value of my_vertex_credentials as the environment variable VERTEX_CREDENTIALS # Set the value of my_vertex_credentials as the environment variable VERTEX_CREDENTIALS
if settings.ENABLE_VERTEX_AI: if settings.ENABLE_VERTEX_AI:
LLMConfigRegistry.register_config( LLMConfigRegistry.register_config(
"VERTEX_GEMINI_2.5_PRO_EXP_03_25", "VERTEX_GEMINI_2.5_FLASH_PREVIEW_04_17",
LLMConfig( LLMConfig(
"vertex_ai/gemini-2.5-pro-preview-03-25", "vertex_ai/gemini-2.5-flash-preview-04-17",
["VERTEX_CREDENTIALS"], ["VERTEX_CREDENTIALS"],
supports_vision=True, supports_vision=True,
add_assistant_prefix=False, add_assistant_prefix=False,

View file

@ -2464,7 +2464,7 @@ async def select_from_emerging_elements(
task_id=task.task_id, task_id=task.task_id,
) )
action_type_str: str = json_response.get("action_type", "") action_type_str: str = json_response.get("action_type", "") or ""
action_type = ActionType(action_type_str.lower()) action_type = ActionType(action_type_str.lower())
element_id: str | None = json_response.get("id", None) element_id: str | None = json_response.get("id", None)
if not element_id or action_type not in [ActionType.CLICK, ActionType.INPUT_TEXT]: if not element_id or action_type not in [ActionType.CLICK, ActionType.INPUT_TEXT]:
@ -2590,7 +2590,7 @@ async def select_from_dropdown(
task_id=task.task_id, task_id=task.task_id,
) )
action_type: str = json_response.get("action_type", "") action_type: str = json_response.get("action_type", "") or ""
action_type = action_type.lower() action_type = action_type.lower()
single_select_result.action_type = ActionType(action_type) single_select_result.action_type = ActionType(action_type)
element_id: str | None = json_response.get("id", None) element_id: str | None = json_response.get("id", None)