(
body: dict[str, Any],
*,
default_mode: RoutingMode = RoutingMode.AUTO,
)
| 2489 | |
| 2490 | |
| 2491 | def _normalize_selector_body( |
| 2492 | body: dict[str, Any], |
| 2493 | *, |
| 2494 | default_mode: RoutingMode = RoutingMode.AUTO, |
| 2495 | ) -> tuple[dict[str, Any] | None, str | None]: |
| 2496 | payload = dict(body) |
| 2497 | model = str(payload.get("model") or "").strip().lower() |
| 2498 | mode_value = payload.get("mode") |
| 2499 | if mode_value is not None and not model: |
| 2500 | try: |
| 2501 | model = VIRTUAL_MODEL_IDS[_parse_mode_value(str(mode_value))] |
| 2502 | except ValueError: |
| 2503 | return None, "Invalid mode" |
| 2504 | payload["model"] = model |
| 2505 | if payload.get("messages"): |
| 2506 | if not payload.get("model"): |
| 2507 | payload["model"] = VIRTUAL_MODEL_IDS[default_mode] |
| 2508 | return payload, None |
| 2509 | prompt = payload.get("prompt") |
| 2510 | if not isinstance(prompt, str) or not prompt.strip(): |
| 2511 | return None, "Requires either messages or prompt" |
| 2512 | system_prompt = payload.get("system_prompt") |
| 2513 | messages: list[dict[str, Any]] = [] |
| 2514 | if isinstance(system_prompt, str) and system_prompt.strip(): |
| 2515 | messages.append({"role": "system", "content": system_prompt}) |
| 2516 | messages.append({"role": "user", "content": prompt}) |
| 2517 | payload["messages"] = messages |
| 2518 | if not payload.get("model"): |
| 2519 | payload["model"] = VIRTUAL_MODEL_IDS[default_mode] |
| 2520 | return payload, None |
| 2521 | |
| 2522 | |
| 2523 | def _flatten_tool_result(content: Any) -> str: |
no test coverage detected