Skip to content

Commit dba0b55

Browse files
author
maebahesioru
committed
fix: increase submit button wait timeout and add fuzzy model id matching
- click_element for Submit Button: internal_timeout 2000ms -> 10000ms - request_processor: if model id not in parsed_model_list, try fuzzy match (e.g. gemini-2.5-flash -> gemini-2.5-flash-preview-04-17)
1 parent 474190b commit dba0b55

2 files changed

Lines changed: 8 additions & 2 deletions

File tree

src/api/request_processor.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,13 @@ async def _analyze_model_requirements(req_id: str, context: dict, request: ChatC
114114
if parsed_model_list:
115115
valid_model_ids = [m.get('id') for m in parsed_model_list]
116116
if requested_model_id not in valid_model_ids:
117-
raise HTTPException(status_code=400, detail=f"[{req_id}] Invalid model '{requested_model_id}'. Available models: {', '.join(valid_model_ids)}")
117+
# fuzzy match: find model whose id contains the requested id or vice versa
118+
fuzzy = next((mid for mid in valid_model_ids if requested_model_id in mid or mid.startswith(requested_model_id.split('-preview')[0])), None)
119+
if fuzzy:
120+
logger.info(f'[{req_id}] 模型 "{requested_model_id}" 不在列表中,自动映射到 "{fuzzy}"')
121+
requested_model_id = fuzzy
122+
else:
123+
raise HTTPException(status_code=400, detail=f"[{req_id}] Invalid model '{requested_model_id}'. Available models: {', '.join(valid_model_ids)}")
118124
context['model_id_to_use'] = requested_model_id
119125
if current_ai_studio_model_id != requested_model_id:
120126
context['needs_model_switching'] = True

src/browser/page_controller.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1079,7 +1079,7 @@ async def submit_prompt(self, prompt: str, image_list: List, check_client_discon
10791079
submitted_successfully = await self._try_shortcut_submit(prompt_textarea_locator, check_client_disconnected)
10801080
if not submitted_successfully:
10811081
self.logger.info(f'[{self.req_id}] 快捷键提交失败,尝试点击提交按钮...')
1082-
await click_element(self.page, submit_button_locator, 'Submit Button', self.req_id)
1082+
await click_element(self.page, submit_button_locator, 'Submit Button', self.req_id, internal_timeout=10000)
10831083
self.logger.info(f'[{self.req_id}] 提交按钮点击完成。')
10841084
await self._check_disconnect(check_client_disconnected, '提交后')
10851085

0 commit comments

Comments
 (0)