Skip to content

Commit f57053c

Browse files
committed
fix minor compatibility issues
1 parent 6fee42f commit f57053c

3 files changed

Lines changed: 6 additions & 8 deletions

File tree

example.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from dotenv import load_dotenv
22

3-
from swarms import Agent
3+
from swarms.structs.agent import Agent
44

55
load_dotenv()
66

@@ -25,7 +25,7 @@
2525
max_loops=1,
2626
top_p=None,
2727
temperature=None,
28-
reasoning_effort="medium",
28+
reasoning_effort=None,
2929
persistent_memory=False,
3030
)
3131

swarms/structs/agent.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,8 @@
5454
from swarms.prompts.react_base_prompt import REACT_SYS_PROMPT
5555
from swarms.prompts.safety_prompt import SAFETY_PROMPT
5656
from swarms.schemas.agent_errors import (
57-
AgentError,
5857
AgentInitializationError,
5958
AgentLLMError,
60-
AgentLLMInitializationError,
6159
AgentRunError,
6260
AgentToolExecutionError,
6361
)

swarms/utils/litellm_wrapper.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1206,10 +1206,10 @@ def _build_completion_params(
12061206
if self.modalities and len(self.modalities) >= 2:
12071207
completion_params["modalities"] = self.modalities
12081208

1209-
if (
1210-
self.reasoning_effort is not None
1211-
and _model_supports_reasoning(self.model_name)
1212-
):
1209+
# Forward an explicitly requested effort level even when LiteLLM's
1210+
# local capability registry does not yet recognize a newly released
1211+
# model. `drop_params` handles providers that do not accept it.
1212+
if self.reasoning_effort is not None:
12131213
completion_params["reasoning_effort"] = (
12141214
self.reasoning_effort
12151215
)

0 commit comments

Comments
 (0)