andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-24 23:58:15 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix incompatible BaseLLM.call method signature

Browse Source
This commit is contained in:
Lucas Gomide
2025-06-30 17:36:41 -03:00
parent f8a8d63ae0
commit 2f7d9bf061
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/crewai/llms/third_party/ai_suite.py vendored
View File

@@ -16,6 +16,7 @@ class AISuiteLLM(BaseLLM):
tools: Optional[List[dict]] = None, tools: Optional[List[dict]] = None,
callbacks: Optional[List[Any]] = None, callbacks: Optional[List[Any]] = None,
available_functions: Optional[Dict[str, Any]] = None, available_functions: Optional[Dict[str, Any]] = None,
from_task: Optional[Task] = None,
) -> Union[str, Any]: ) -> Union[str, Any]:
completion_params = self._prepare_completion_params(messages, tools) completion_params = self._prepare_completion_params(messages, tools)
response = self.client.chat.completions.create(**completion_params) response = self.client.chat.completions.create(**completion_params)

1
tests/custom_llm_test.py
View File

@@ -1,5 +1,4 @@
from typing import Any, Dict, List, Optional, Union from typing import Any, Dict, List, Optional, Union
from unittest.mock import Mock
import pytest import pytest