|
24 | 24 | from maf_starter.devui_overrides import OLD_GD_RENDERER |
25 | 25 | from maf_starter.devui_overrides import _patch_devui_bundle |
26 | 26 | from maf_starter.devui_overrides import install_devui_ui_overrides |
| 27 | +import maf_starter.provider_fallback as provider_fallback |
27 | 28 | from maf_starter.provider_fallback import ( |
28 | 29 | ResponseStream, |
29 | 30 | _is_fallback_worthy_error, |
30 | 31 | _merge_route_metadata, |
31 | 32 | _response_from_updates, |
32 | 33 | _resolve_run_scope, |
33 | | - _sanitize_messages, |
34 | 34 | _wrap_stream_with_fallback, |
35 | 35 | ) |
36 | 36 | from maf_starter.routing_policy import RoutingPlan, build_routing_plan |
@@ -182,6 +182,9 @@ def test_build_repo_tools_contains_expected_tool_names(self) -> None: |
182 | 182 | "search_repo", |
183 | 183 | "request_human_approval", |
184 | 184 | "apply_repo_write_plan", |
| 185 | + "delegate_task_to_claude", |
| 186 | + "delegate_task_to_codex", |
| 187 | + "delegate_task_to_antigravity", |
185 | 188 | }, |
186 | 189 | ) |
187 | 190 |
|
@@ -348,7 +351,7 @@ def test_sanitize_messages_strips_unsupported_reasoning_properties(self) -> None |
348 | 351 | Message(role="user", text="final"), |
349 | 352 | ] |
350 | 353 |
|
351 | | - sanitized = _sanitize_messages(messages) |
| 354 | + sanitized = provider_fallback._sanitize_messages(messages) |
352 | 355 |
|
353 | 356 | self.assertEqual(len(sanitized), 2) |
354 | 357 | self.assertNotIn("reasoning_details", sanitized[0].additional_properties) |
@@ -618,3 +621,4 @@ def test_env_example_documents_ollama(self) -> None: |
618 | 621 |
|
619 | 622 | if __name__ == "__main__": |
620 | 623 | unittest.main() |
| 624 | + |
0 commit comments