klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[chore]: Rename modules for better readability and maintainability (#587)

Browse Source
This commit is contained in:
Deshraj Yadav
2023-09-10 18:31:40 -07:00
committed by GitHub
parent 6fed75bb45
commit 79f5a1d052
65 changed files with 109 additions and 108 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
embedchain/llm/antrophic_llm.py → embedchain/llm/antrophic.py
View File

@@ -2,8 +2,8 @@ import logging
from typing import Optional
from embedchain.config import BaseLlmConfig
from embedchain.helper_classes.json_serializable import register_deserializable
from embedchain.llm.base_llm import BaseLlm
from embedchain.helper.json_serializable import register_deserializable
from embedchain.llm.base import BaseLlm
@register_deserializable

8
embedchain/llm/azure_openai_llm.py → embedchain/llm/azure_openai.py
View File

@@ -2,17 +2,17 @@ import logging
from typing import Optional
from embedchain.config import BaseLlmConfig
from embedchain.helper_classes.json_serializable import register_deserializable
from embedchain.llm.base_llm import BaseLlm
from embedchain.helper.json_serializable import register_deserializable
from embedchain.llm.base import BaseLlm
@register_deserializable
class AzureOpenAiLlm(BaseLlm):
class AzureOpenAILlm(BaseLlm):
def __init__(self, config: Optional[BaseLlmConfig] = None):
super().__init__(config=config)
def get_llm_model_answer(self, prompt):
return AzureOpenAiLlm._get_azure_openai_answer(prompt=prompt, config=self.config)
return AzureOpenAILlm._get_azure_openai_answer(prompt=prompt, config=self.config)
@staticmethod
def _get_azure_openai_answer(prompt: str, config: BaseLlmConfig) -> str:

2
embedchain/llm/base_llm.py → embedchain/llm/base.py
View File

@@ -8,7 +8,7 @@ from embedchain.config import BaseLlmConfig
from embedchain.config.llm.base_llm_config import (
DEFAULT_PROMPT, DEFAULT_PROMPT_WITH_HISTORY_TEMPLATE,
DOCS_SITE_PROMPT_TEMPLATE)
from embedchain.helper_classes.json_serializable import JSONSerializable
from embedchain.helper.json_serializable import JSONSerializable
class BaseLlm(JSONSerializable):

4
embedchain/llm/gpt4all_llm.py → embedchain/llm/gpt4all.py
View File

@@ -1,8 +1,8 @@
from typing import Iterable, Optional, Union
from embedchain.config import BaseLlmConfig
from embedchain.helper_classes.json_serializable import register_deserializable
from embedchain.llm.base_llm import BaseLlm
from embedchain.helper.json_serializable import register_deserializable
from embedchain.llm.base import BaseLlm
@register_deserializable

4
embedchain/llm/llama2_llm.py → embedchain/llm/llama2.py
View File

@@ -4,8 +4,8 @@ from typing import Optional
from langchain.llms import Replicate
from embedchain.config import BaseLlmConfig
from embedchain.helper_classes.json_serializable import register_deserializable
from embedchain.llm.base_llm import BaseLlm
from embedchain.helper.json_serializable import register_deserializable
from embedchain.llm.base import BaseLlm
@register_deserializable

6
embedchain/llm/openai_llm.py → embedchain/llm/openai.py
View File

@@ -3,12 +3,12 @@ from typing import Optional
import openai
from embedchain.config import BaseLlmConfig
from embedchain.helper_classes.json_serializable import register_deserializable
from embedchain.llm.base_llm import BaseLlm
from embedchain.helper.json_serializable import register_deserializable
from embedchain.llm.base import BaseLlm
@register_deserializable
class OpenAiLlm(BaseLlm):
class OpenAILlm(BaseLlm):
def __init__(self, config: Optional[BaseLlmConfig] = None):
super().__init__(config=config)

4
embedchain/llm/vertex_ai_llm.py → embedchain/llm/vertex_ai.py
View File

@@ -2,8 +2,8 @@ import logging
from typing import Optional
from embedchain.config import BaseLlmConfig
from embedchain.helper_classes.json_serializable import register_deserializable
from embedchain.llm.base_llm import BaseLlm
from embedchain.helper.json_serializable import register_deserializable
from embedchain.llm.base import BaseLlm
@register_deserializable