diff --git a/src/askui/agent_base.py b/src/askui/agent_base.py index a1e4439e..bb72357c 100644 --- a/src/askui/agent_base.py +++ b/src/askui/agent_base.py @@ -107,7 +107,7 @@ def _init_model_choice( "locate": model_choice.get("locate", ModelName.ASKUI), } - @telemetry.record_call(exclude={"goal", "on_message"}) + @telemetry.record_call(exclude={"goal", "on_message", "settings"}) @validate_call def act( self, diff --git a/src/askui/android_agent.py b/src/askui/android_agent.py index 141adca9..6fcb6d35 100644 --- a/src/askui/android_agent.py +++ b/src/askui/android_agent.py @@ -105,6 +105,7 @@ messages=MessageSettings( model=ModelName.ANTHROPIC__CLAUDE__3_5__SONNET__20241022.value, system=_SYSTEM_PROMPT, + betas=[], ), ) @@ -113,6 +114,7 @@ model=ModelName.CLAUDE__SONNET__4__20250514.value, system=_SYSTEM_PROMPT, thinking={"type": "enabled", "budget_tokens": 2048}, + betas=[], ), ) diff --git a/src/askui/models/anthropic/messages_api.py b/src/askui/models/anthropic/messages_api.py index 6866f7e9..5418984b 100644 --- a/src/askui/models/anthropic/messages_api.py +++ b/src/askui/models/anthropic/messages_api.py @@ -139,7 +139,9 @@ def create_message( tools=tools.to_params() if tools else NOT_GIVEN, max_tokens=max_tokens or self._settings.messages.max_tokens, model=ANTHROPIC_MODEL_MAPPING[model], - betas=betas or self._settings.messages.betas, + betas=betas + if not isinstance(betas, NotGiven) + else self._settings.messages.betas, system=system or self._settings.messages.system, thinking=thinking or self._settings.messages.thinking, tool_choice=tool_choice or self._settings.messages.tool_choice, diff --git a/src/askui/models/askui/inference_api.py b/src/askui/models/askui/inference_api.py index ab480f5b..e01d3e6e 100644 --- a/src/askui/models/askui/inference_api.py +++ b/src/askui/models/askui/inference_api.py @@ -230,7 +230,9 @@ def create_message( "tools": tools.to_params() if tools else NOT_GIVEN, "max_tokens": max_tokens or self._settings.messages.max_tokens, "model": model, - "betas": betas or self._settings.messages.betas, + "betas": betas + if not isinstance(betas, NotGiven) + else self._settings.messages.betas, "system": system or self._settings.messages.system, "thinking": thinking or self._settings.messages.thinking, "tool_choice": tool_choice or self._settings.messages.tool_choice,