diff --git a/litellm/files/main.py b/litellm/files/main.py index 666ad1f4aa..fbdf6206d2 100644 --- a/litellm/files/main.py +++ b/litellm/files/main.py @@ -15,7 +15,7 @@ import httpx import litellm from litellm import client, get_secret_str -from litellm.llms.files_apis.azure import AzureOpenAIFilesAPI +from litellm.llms.azure.files.handler import AzureOpenAIFilesAPI from litellm.llms.openai.openai import FileDeleted, FileObject, OpenAIFilesAPI from litellm.llms.vertex_ai_and_google_ai_studio.files.handler import ( VertexAIFilesHandler, diff --git a/litellm/fine_tuning/main.py b/litellm/fine_tuning/main.py index a4bdcadadd..2d97eb21f4 100644 --- a/litellm/fine_tuning/main.py +++ b/litellm/fine_tuning/main.py @@ -18,9 +18,9 @@ import httpx import litellm from litellm._logging import verbose_logger -from litellm.llms.fine_tuning_apis.azure import AzureOpenAIFineTuningAPI +from litellm.llms.azure.fine_tuning.handler import AzureOpenAIFineTuningAPI from litellm.llms.openai.fine_tuning.handler import OpenAIFineTuningAPI, FineTuningJob, FineTuningJobCreate -from litellm.llms.fine_tuning_apis.vertex_ai import VertexFineTuningAPI +from litellm.llms.vertex_ai_and_google_ai_studio.fine_tuning.handler import VertexFineTuningAPI from litellm.secret_managers.main import get_secret_str from litellm.types.llms.openai import Hyperparameters from litellm.types.router import * diff --git a/litellm/llms/files_apis/azure.py b/litellm/llms/azure/files/handler.py similarity index 100% rename from litellm/llms/files_apis/azure.py rename to litellm/llms/azure/files/handler.py diff --git a/litellm/llms/fine_tuning_apis/azure.py b/litellm/llms/azure/fine_tuning/handler.py similarity index 98% rename from litellm/llms/fine_tuning_apis/azure.py rename to litellm/llms/azure/fine_tuning/handler.py index 3e9c335e16..f336efa331 100644 --- a/litellm/llms/fine_tuning_apis/azure.py +++ b/litellm/llms/azure/fine_tuning/handler.py @@ -7,7 +7,7 @@ from openai.types.fine_tuning import FineTuningJob from litellm._logging import verbose_logger from litellm.llms.base import BaseLLM -from litellm.llms.files_apis.azure import get_azure_openai_client +from litellm.llms.azure.files.handler import get_azure_openai_client from litellm.types.llms.openai import FineTuningJobCreate diff --git a/litellm/llms/fine_tuning_apis/vertex_ai.py b/litellm/llms/vertex_ai_and_google_ai_studio/fine_tuning/handler.py similarity index 99% rename from litellm/llms/fine_tuning_apis/vertex_ai.py rename to litellm/llms/vertex_ai_and_google_ai_studio/fine_tuning/handler.py index fd418103e7..f82d781c55 100644 --- a/litellm/llms/fine_tuning_apis/vertex_ai.py +++ b/litellm/llms/vertex_ai_and_google_ai_studio/fine_tuning/handler.py @@ -9,7 +9,6 @@ import litellm from litellm._logging import verbose_logger from litellm.llms.base import BaseLLM from litellm.llms.custom_httpx.http_handler import ( - AsyncHTTPHandler, HTTPHandler, get_async_httpx_client, ) diff --git a/tests/local_testing/test_fine_tuning_api.py b/tests/local_testing/test_fine_tuning_api.py index 938c1b2951..e231117a19 100644 --- a/tests/local_testing/test_fine_tuning_api.py +++ b/tests/local_testing/test_fine_tuning_api.py @@ -20,7 +20,7 @@ from test_gcs_bucket import load_vertex_ai_credentials from litellm import create_fine_tuning_job from litellm._logging import verbose_logger -from litellm.llms.fine_tuning_apis.vertex_ai import ( +from litellm.llms.vertex_ai_and_google_ai_studio.fine_tuning.handler import ( FineTuningJobCreate, VertexFineTuningAPI, )