From 87a8ac92280b3c87f2f523c83143c811ff339e70 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Mon, 1 Jun 2026 06:23:21 +0000 Subject: [PATCH] fix: match parent class type annotations for _prepare_completion_params MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: João --- .../llms/providers/openai_compatible/completion.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/lib/crewai/src/crewai/llms/providers/openai_compatible/completion.py b/lib/crewai/src/crewai/llms/providers/openai_compatible/completion.py index b5771a52b..2dc915f99 100644 --- a/lib/crewai/src/crewai/llms/providers/openai_compatible/completion.py +++ b/lib/crewai/src/crewai/llms/providers/openai_compatible/completion.py @@ -16,11 +16,16 @@ from dataclasses import dataclass, field import json import logging import os -from typing import Any +from typing import TYPE_CHECKING, Any from pydantic import BaseModel, model_validator from crewai.llms.providers.openai.completion import OpenAICompletion +from crewai.utilities.types import LLMMessage + + +if TYPE_CHECKING: + from crewai.tools.base_tool import BaseTool @dataclass(frozen=True) @@ -265,8 +270,8 @@ class OpenAICompatibleCompletion(OpenAICompletion): def _prepare_completion_params( self, - messages: list[dict[str, Any]], - tools: list[dict[str, Any]] | None = None, + messages: list[LLMMessage], + tools: list[dict[str, BaseTool]] | None = None, ) -> dict[str, Any]: """Prepare params, stripping json_schema response_format if unsupported.""" params = super()._prepare_completion_params(messages, tools)