diff --git a/neon_llm_core/llm.py b/neon_llm_core/llm.py index 37235eb..5d48f91 100644 --- a/neon_llm_core/llm.py +++ b/neon_llm_core/llm.py @@ -28,7 +28,6 @@ from typing import List, Optional, Tuple, Union from neon_data_models.models.api import LLMRequest, LLMResponse -from openai import OpenAI from ovos_utils.log import log_deprecation @@ -69,7 +68,7 @@ def tokenizer_model_name(self) -> str: @property @abstractmethod - def model(self) -> OpenAI: + def model(self) -> object: """ Get an OpenAI client object to send requests to. """