andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-07-02 13:48:09 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

fix: widen max_tokens to int | float | None + apply ruff format

Browse Source
This commit is contained in:
Lucas Gomide
2026-05-27 18:23:54 -03:00
parent c25543f3fb
commit fe3453294d
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/crewai/src/crewai/events/types/llm_events.py
View File

@@ -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

8
lib/crewai/src/crewai/llms/base_llm.py
View File

@@ -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: