Lancedb Integration (#1411)

2024-06-21 21:29:22 +05:30
parent f6ddd5ffc5
commit 48b24f6f12
16 changed files with 963 additions and 12 deletions
--- a/embedchain/llm/jina.py
+++ b/embedchain/llm/jina.py
@@ -35,7 +35,8 @@ class JinaLlm(BaseLlm):
        if config.top_p:
            kwargs["model_kwargs"]["top_p"] = config.top_p
        if config.stream:
-            from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
+            from langchain.callbacks.streaming_stdout import \
+                StreamingStdOutCallbackHandler

            chat = JinaChat(**kwargs, streaming=config.stream, callbacks=[StreamingStdOutCallbackHandler()])
        else: