From fe3453294d418e946e5cb5cf2a1329a3a9040eb3 Mon Sep 17 00:00:00 2001 From: Lucas Gomide Date: Wed, 27 May 2026 18:23:54 -0300 Subject: [PATCH] fix: widen max_tokens to int | float | None + apply ruff format --- lib/crewai/src/crewai/events/types/llm_events.py | 2 +- lib/crewai/src/crewai/llms/base_llm.py | 8 +++----- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/lib/crewai/src/crewai/events/types/llm_events.py b/lib/crewai/src/crewai/events/types/llm_events.py index 13585d495..b1be8c3d5 100644 --- a/lib/crewai/src/crewai/events/types/llm_events.py +++ b/lib/crewai/src/crewai/events/types/llm_events.py @@ -52,7 +52,7 @@ class LLMCallStartedEvent(LLMEventBase): # All optional so legacy emitters keep working unchanged. temperature: float | None = None top_p: float | None = None - max_tokens: int | None = None + max_tokens: int | float | None = None stream: bool | None = None seed: int | None = None stop_sequences: list[str] | None = None diff --git a/lib/crewai/src/crewai/llms/base_llm.py b/lib/crewai/src/crewai/llms/base_llm.py index aaaca1cd0..b2d66273f 100644 --- a/lib/crewai/src/crewai/llms/base_llm.py +++ b/lib/crewai/src/crewai/llms/base_llm.py @@ -151,7 +151,7 @@ class BaseLLM(BaseModel, ABC): model: str temperature: float | None = None top_p: float | None = None - max_tokens: int | None = None + max_tokens: int | float | None = None stream: bool | None = None seed: int | None = None frequency_penalty: float | None = None @@ -481,7 +481,7 @@ class BaseLLM(BaseModel, ABC): from_agent: BaseAgent | None = None, temperature: float | None = None, top_p: float | None = None, - max_tokens: int | None = None, + max_tokens: int | float | None = None, stream: bool | None = None, seed: int | None = None, stop_sequences: list[str] | None = None, @@ -489,9 +489,7 @@ class BaseLLM(BaseModel, ABC): presence_penalty: float | None = None, n: int | None = None, ) -> None: - """Emit LLM call started event. - - """ + """Emit LLM call started event.""" from crewai.utilities.serialization import to_serializable if temperature is None: