Merge pull request #18 from cyrannano/illm-import-fix

Reorganize and clarify ILlm imports across llms
Mindinventory · Aug 23, 2024 · 8c6a90a · 8c6a90a
2 parents 6215f58 + 9b9a66e
commit 8c6a90a
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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