diff --git a/g4f/local/__init__.py b/g4f/local/__init__.py index cc678dc6..c9d3d74a 100644 --- a/g4f/local/__init__.py +++ b/g4f/local/__init__.py @@ -1,8 +1,8 @@ from ..typing import Union, Iterator, Messages from ..stubs import ChatCompletion, ChatCompletionChunk -from .core.engine import LocalProvider -from .core.models import models -from ..client import iter_response, filter_none, IterResponse +from ._engine import LocalProvider +from ._models import models +from ..client import iter_response, filter_none, IterResponse class LocalClient(): def __init__(self, **kwargs) -> None: diff --git a/g4f/local/core/engine.py b/g4f/local/_engine.py similarity index 94% rename from g4f/local/core/engine.py rename to g4f/local/_engine.py index 920ed9b4..5a701894 100644 --- a/g4f/local/core/engine.py +++ b/g4f/local/_engine.py @@ -1,7 +1,7 @@ import os -from gpt4all import GPT4All -from .models import models +from gpt4all import GPT4All +from ._models import models class LocalProvider: @staticmethod diff --git a/g4f/local/core/models.py b/g4f/local/_models.py similarity index 100% rename from g4f/local/core/models.py rename to g4f/local/_models.py