diff --git a/mindsql/llms/__init__.py b/mindsql/llms/__init__.py index e713c24..9aff339 100644 --- a/mindsql/llms/__init__.py +++ b/mindsql/llms/__init__.py @@ -1,6 +1,6 @@ +from .illm import ILlm from .anthropic import AnthropicAi from .googlegenai import GoogleGenAi from .huggingface import HuggingFace -from .illm import ILlm from .llama import LlamaCpp from .open_ai import OpenAi diff --git a/mindsql/llms/anthropic.py b/mindsql/llms/anthropic.py index 9156f60..4f37984 100644 --- a/mindsql/llms/anthropic.py +++ b/mindsql/llms/anthropic.py @@ -1,6 +1,6 @@ from anthropic import Anthropic -from . import ILlm +from .illm import ILlm from .._utils.constants import ANTHROPIC_VALUE_ERROR, PROMPT_EMPTY_EXCEPTION diff --git a/mindsql/llms/googlegenai.py b/mindsql/llms/googlegenai.py index 9e80582..80723c9 100644 --- a/mindsql/llms/googlegenai.py +++ b/mindsql/llms/googlegenai.py @@ -1,7 +1,7 @@ import google.generativeai as genai from .._utils.constants import GOOGLE_GEN_AI_VALUE_ERROR, GOOGLE_GEN_AI_APIKEY_ERROR -from . import ILlm +from .illm import ILlm class GoogleGenAi(ILlm): diff --git a/mindsql/llms/open_ai.py b/mindsql/llms/open_ai.py index 18ec443..b9bd4f9 100644 --- a/mindsql/llms/open_ai.py +++ b/mindsql/llms/open_ai.py @@ -1,6 +1,6 @@ from openai import OpenAI -from . import ILlm +from .illm import ILlm from .._utils.constants import OPENAI_VALUE_ERROR, PROMPT_EMPTY_EXCEPTION