[Feature] Add support for custom streaming callback (#971)

2023-11-22 01:06:33 -08:00
parent 798d3fcc5a
commit f6b80e01a1
82 changed files with 162 additions and 84 deletions
--- a/embedchain/embedder/base.py
+++ b/embedchain/embedder/base.py
@@ -3,12 +3,12 @@ from typing import Any, Callable, Optional
 from embedchain.config.embedder.base import BaseEmbedderConfig

 try:
-    from chromadb.api.types import Embeddings, Embeddable, EmbeddingFunction
+    from chromadb.api.types import Embeddable, EmbeddingFunction, Embeddings
 except RuntimeError:
    from embedchain.utils import use_pysqlite3

    use_pysqlite3()
-    from chromadb.api.types import Embeddings, Embeddable, EmbeddingFunction
+    from chromadb.api.types import Embeddable, EmbeddingFunction, Embeddings


 class EmbeddingFunc(EmbeddingFunction):