Skip to content

Commit 79f5a1d

Browse files
authored
[chore]: Rename modules for better readability and maintainability (#587)
1 parent 6fed75b commit 79f5a1d

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

65 files changed

+109
-108
lines changed

docs/advanced/app_types.mdx

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,10 @@ app = OpenSourceApp()
7171
from embedchain import CustomApp
7272
from embedchain.config import (CustomAppConfig, ElasticsearchDBConfig,
7373
EmbedderConfig, LlmConfig)
74-
from embedchain.embedder.vertexai_embedder import VertexAiEmbedder
75-
from embedchain.llm.vertex_ai_llm import VertexAiLlm
74+
from embedchain.embedder.vertexai import VertexAiEmbedder
75+
from embedchain.llm.vertex_ai import VertexAiLlm
7676
from embedchain.models import EmbeddingFunctions, Providers
77-
from embedchain.vectordb.elasticsearch_db import Elasticsearch
77+
from embedchain.vectordb.elasticsearch import Elasticsearch
7878

7979
# short
8080
app = CustomApp(llm=VertexAiLlm(), db=Elasticsearch(), embedder=VertexAiEmbedder())

embedchain/__init__.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,6 @@
66
from embedchain.apps.CustomApp import CustomApp # noqa: F401
77
from embedchain.apps.Llama2App import Llama2App # noqa: F401
88
from embedchain.apps.OpenSourceApp import OpenSourceApp # noqa: F401
9-
from embedchain.apps.PersonApp import PersonApp, PersonOpenSourceApp # noqa: F401
10-
from embedchain.vectordb.chroma_db import ChromaDB # noqa: F401
9+
from embedchain.apps.PersonApp import (PersonApp, # noqa: F401
10+
PersonOpenSourceApp)
11+
from embedchain.vectordb.chroma import ChromaDB # noqa: F401

embedchain/apps/App.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@
33
from embedchain.config import (AppConfig, BaseEmbedderConfig, BaseLlmConfig,
44
ChromaDbConfig)
55
from embedchain.embedchain import EmbedChain
6-
from embedchain.embedder.openai_embedder import OpenAiEmbedder
7-
from embedchain.helper_classes.json_serializable import register_deserializable
8-
from embedchain.llm.openai_llm import OpenAiLlm
9-
from embedchain.vectordb.chroma_db import ChromaDB
6+
from embedchain.embedder.openai import OpenAiEmbedder
7+
from embedchain.helper.json_serializable import register_deserializable
8+
from embedchain.llm.openai import OpenAILlm
9+
from embedchain.vectordb.chroma import ChromaDB
1010

1111

1212
@register_deserializable
@@ -47,7 +47,7 @@ def __init__(
4747
if config is None:
4848
config = AppConfig()
4949

50-
llm = OpenAiLlm(config=llm_config)
50+
llm = OpenAILlm(config=llm_config)
5151
embedder = OpenAiEmbedder(config=BaseEmbedderConfig(model="text-embedding-ada-002"))
5252
database = ChromaDB(config=chromadb_config)
5353

embedchain/apps/CustomApp.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@
22

33
from embedchain.config import CustomAppConfig
44
from embedchain.embedchain import EmbedChain
5-
from embedchain.embedder.base_embedder import BaseEmbedder
6-
from embedchain.helper_classes.json_serializable import register_deserializable
7-
from embedchain.llm.base_llm import BaseLlm
8-
from embedchain.vectordb.base_vector_db import BaseVectorDB
5+
from embedchain.embedder.base import BaseEmbedder
6+
from embedchain.helper.json_serializable import register_deserializable
7+
from embedchain.llm.base import BaseLlm
8+
from embedchain.vectordb.base import BaseVectorDB
99

1010

1111
@register_deserializable
@@ -35,7 +35,7 @@ def __init__(
3535
:param config: Config for the app instance. This is the most basic configuration,
3636
that does not fall into the LLM, database or embedder category, defaults to None
3737
:type config: Optional[CustomAppConfig], optional
38-
:param llm: LLM Class instance. example: `from embedchain.llm.openai_llm import OpenAiLlm`, defaults to None
38+
:param llm: LLM Class instance. example: `from embedchain.llm.openai import OpenAILlm`, defaults to None
3939
:type llm: BaseLlm
4040
:param db: The database to use for storing and retrieving embeddings,
4141
example: `from embedchain.vectordb.chroma_db import ChromaDb`, defaults to None

embedchain/apps/Llama2App.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,10 @@
22

33
from embedchain.apps.CustomApp import CustomApp
44
from embedchain.config import CustomAppConfig
5-
from embedchain.embedder.openai_embedder import OpenAiEmbedder
6-
from embedchain.helper_classes.json_serializable import register_deserializable
7-
from embedchain.llm.llama2_llm import Llama2Llm
8-
from embedchain.vectordb.chroma_db import ChromaDB
5+
from embedchain.embedder.openai import OpenAiEmbedder
6+
from embedchain.helper.json_serializable import register_deserializable
7+
from embedchain.llm.llama2 import Llama2Llm
8+
from embedchain.vectordb.chroma import ChromaDB
99

1010

1111
@register_deserializable

embedchain/apps/OpenSourceApp.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,10 @@
44
from embedchain.config import (BaseEmbedderConfig, BaseLlmConfig,
55
ChromaDbConfig, OpenSourceAppConfig)
66
from embedchain.embedchain import EmbedChain
7-
from embedchain.embedder.gpt4all_embedder import GPT4AllEmbedder
8-
from embedchain.helper_classes.json_serializable import register_deserializable
9-
from embedchain.llm.gpt4all_llm import GPT4ALLLlm
10-
from embedchain.vectordb.chroma_db import ChromaDB
7+
from embedchain.embedder.gpt4all import GPT4AllEmbedder
8+
from embedchain.helper.json_serializable import register_deserializable
9+
from embedchain.llm.gpt4all import GPT4ALLLlm
10+
from embedchain.vectordb.chroma import ChromaDB
1111

1212
gpt4all_model = None
1313

embedchain/apps/PersonApp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from embedchain.config.apps.BaseAppConfig import BaseAppConfig
77
from embedchain.config.llm.base_llm_config import (DEFAULT_PROMPT,
88
DEFAULT_PROMPT_WITH_HISTORY)
9-
from embedchain.helper_classes.json_serializable import register_deserializable
9+
from embedchain.helper.json_serializable import register_deserializable
1010

1111

1212
@register_deserializable

embedchain/bots/base.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2,17 +2,17 @@
22

33
from embedchain import CustomApp
44
from embedchain.config import AddConfig, CustomAppConfig, LlmConfig
5-
from embedchain.embedder.openai_embedder import OpenAiEmbedder
6-
from embedchain.helper_classes.json_serializable import (
7-
JSONSerializable, register_deserializable)
8-
from embedchain.llm.openai_llm import OpenAiLlm
9-
from embedchain.vectordb.chroma_db import ChromaDB
5+
from embedchain.embedder.openai import OpenAiEmbedder
6+
from embedchain.helper.json_serializable import (JSONSerializable,
7+
register_deserializable)
8+
from embedchain.llm.openai import OpenAILlm
9+
from embedchain.vectordb.chroma import ChromaDB
1010

1111

1212
@register_deserializable
1313
class BaseBot(JSONSerializable):
1414
def __init__(self):
15-
self.app = CustomApp(config=CustomAppConfig(), llm=OpenAiLlm(), db=ChromaDB(), embedder=OpenAiEmbedder())
15+
self.app = CustomApp(config=CustomAppConfig(), llm=OpenAILlm(), db=ChromaDB(), embedder=OpenAiEmbedder())
1616

1717
def add(self, data: Any, config: AddConfig = None):
1818
"""

embedchain/bots/discord.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from discord import app_commands
77
from discord.ext import commands
88

9-
from embedchain.helper_classes.json_serializable import register_deserializable
9+
from embedchain.helper.json_serializable import register_deserializable
1010

1111
from .base import BaseBot
1212

embedchain/bots/poe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
from fastapi_poe import PoeBot, run
77

8-
from embedchain.helper_classes.json_serializable import register_deserializable
8+
from embedchain.helper.json_serializable import register_deserializable
99

1010
from .base import BaseBot
1111

0 commit comments

Comments
 (0)