Feat/serialize deserialize (#508)

Co-authored-by: Taranjeet Singh <reachtotj@gmail.com>
This commit is contained in:
cachho
2023-09-03 21:50:18 +02:00
committed by GitHub
parent 2aa25a5169
commit 0d4ad07d7b
42 changed files with 345 additions and 8 deletions

View File

@@ -4,8 +4,10 @@ import openai
from embedchain.config import AppConfig, ChatConfig
from embedchain.embedchain import EmbedChain
from embedchain.helper_classes.json_serializable import register_deserializable
@register_deserializable
class App(EmbedChain):
"""
The EmbedChain app.

View File

@@ -5,9 +5,11 @@ from langchain.schema import BaseMessage
from embedchain.config import ChatConfig, CustomAppConfig
from embedchain.embedchain import EmbedChain
from embedchain.helper_classes.json_serializable import register_deserializable
from embedchain.models import Providers
@register_deserializable
class CustomApp(EmbedChain):
"""
The custom EmbedChain app.

View File

@@ -3,10 +3,12 @@ from typing import Iterable, Union, Optional
from embedchain.config import ChatConfig, OpenSourceAppConfig
from embedchain.embedchain import EmbedChain
from embedchain.helper_classes.json_serializable import register_deserializable
gpt4all_model = None
@register_deserializable
class OpenSourceApp(EmbedChain):
"""
The OpenSource app.

View File

@@ -6,8 +6,10 @@ from embedchain.config import ChatConfig, QueryConfig
from embedchain.config.apps.BaseAppConfig import BaseAppConfig
from embedchain.config.QueryConfig import (DEFAULT_PROMPT,
DEFAULT_PROMPT_WITH_HISTORY)
from embedchain.helper_classes.json_serializable import register_deserializable
@register_deserializable
class EmbedChainPersonApp:
"""
Base class to create a person bot.
@@ -50,6 +52,7 @@ class EmbedChainPersonApp:
return config
@register_deserializable
class PersonApp(EmbedChainPersonApp, App):
"""
The Person app.
@@ -65,6 +68,7 @@ class PersonApp(EmbedChainPersonApp, App):
return super().chat(input_query, config, dry_run)
@register_deserializable
class PersonOpenSourceApp(EmbedChainPersonApp, OpenSourceApp):
"""
The Person app.