From 9b9a66e947e957666a0ae055b85222463726d920 Mon Sep 17 00:00:00 2001 From: Szymon Cyranik Date: Thu, 22 Aug 2024 14:34:47 +0200 Subject: [PATCH] refactor(imports): reorganize and clarify ILlm imports across llms module --- mindsql/llms/__init__.py | 2 +- mindsql/llms/anthropic.py | 2 +- mindsql/llms/googlegenai.py | 2 +- mindsql/llms/open_ai.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) 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