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

[refactor] Use pipeline for bots instead of App (#936)

Browse Source
This commit is contained in:
Sidharth Mohanty
2023-11-10 23:43:21 +05:30
committed by GitHub
parent b314e5e080
commit c45413969a
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
embedchain/bots/base.py
View File

@@ -1,10 +1,9 @@
from typing import Any from typing import Any
from embedchain import App from embedchain import Pipeline as App
from embedchain.config import AddConfig, AppConfig, BaseLlmConfig from embedchain.config import AddConfig, PipelineConfig, BaseLlmConfig
from embedchain.embedder.openai import OpenAIEmbedder from embedchain.embedder.openai import OpenAIEmbedder
from embedchain.helper.json_serializable import (JSONSerializable, from embedchain.helper.json_serializable import JSONSerializable, register_deserializable
register_deserializable)
from embedchain.llm.openai import OpenAILlm from embedchain.llm.openai import OpenAILlm
from embedchain.vectordb.chroma import ChromaDB from embedchain.vectordb.chroma import ChromaDB
@@ -12,7 +11,7 @@ from embedchain.vectordb.chroma import ChromaDB
@register_deserializable @register_deserializable
class BaseBot(JSONSerializable): class BaseBot(JSONSerializable):
def __init__(self): def __init__(self):
self.app = App(config=AppConfig(), llm=OpenAILlm(), db=ChromaDB(), embedder=OpenAIEmbedder()) self.app = App(config=PipelineConfig(), llm=OpenAILlm(), db=ChromaDB(), embedding_model=OpenAIEmbedder())
def add(self, data: Any, config: AddConfig = None): def add(self, data: Any, config: AddConfig = None):
""" """