diff --git a/src/askui/chat/api/runs/runner/runner.py b/src/askui/chat/api/runs/runner/runner.py index f21b9d2c..944b173c 100644 --- a/src/askui/chat/api/runs/runner/runner.py +++ b/src/askui/chat/api/runs/runner/runner.py @@ -35,6 +35,7 @@ from askui.chat.api.runs.runner.events.message_events import MessageEvent from askui.chat.api.runs.runner.events.run_events import RunEvent from askui.custom_agent import CustomAgent +from askui.models.models import ModelName from askui.models.shared.agent_message_param import ( Base64ImageSourceParam, ImageBlockParam, @@ -327,10 +328,12 @@ def _run_agent_inner() -> None: custom_agent = CustomAgent() custom_agent.act( messages, + model=ModelName.ASKUI, on_message=on_message, tools=_tools, settings=ActSettings( messages=MessageSettings( + model=ModelName.CLAUDE__SONNET__4__20250514, system=self._assistant.system or anthropic.NOT_GIVEN, thinking={"type": "enabled", "budget_tokens": 2048}, ),