Browse Source

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

Sidharth Mohanty 1 year ago
parent
commit
c45413969a
1 changed files with 4 additions and 5 deletions
  1. 4 5
      embedchain/bots/base.py

+ 4 - 5
embedchain/bots/base.py

@@ -1,10 +1,9 @@
 from typing import Any
 
-from embedchain import App
-from embedchain.config import AddConfig, AppConfig, BaseLlmConfig
+from embedchain import Pipeline as App
+from embedchain.config import AddConfig, PipelineConfig, BaseLlmConfig
 from embedchain.embedder.openai import OpenAIEmbedder
-from embedchain.helper.json_serializable import (JSONSerializable,
-                                                 register_deserializable)
+from embedchain.helper.json_serializable import JSONSerializable, register_deserializable
 from embedchain.llm.openai import OpenAILlm
 from embedchain.vectordb.chroma import ChromaDB
 
@@ -12,7 +11,7 @@ from embedchain.vectordb.chroma import ChromaDB
 @register_deserializable
 class BaseBot(JSONSerializable):
     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):
         """