diff --git a/application/llm/base.py b/application/llm/base.py index cbce4ffd..bef3e11f 100644 --- a/application/llm/base.py +++ b/application/llm/base.py @@ -30,9 +30,9 @@ class BaseLLM(ABC): and self.fallback_model_name ): try: - from llm.llm_creator import LLMCreator + from application.llm.llm_creator import LLMCreator - self._fallback_llm = LLMCreator( + self._fallback_llm = LLMCreator.create_llm( self.fallback_provider, self.fallback_llm_api_key, None,