fix incompatible BaseLLM.call method signature

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

View File

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

View File

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