Parcourir la source

update: import gpt4all locally (#146)

Import gpt4all locally to avoid error in case its not present.
Rohit Gupta il y a 2 ans
Parent
commit
0e74688145
1 fichiers modifiés avec 2 ajouts et 1 suppressions
  1. 2 1
      embedchain/embedchain.py

+ 2 - 1
embedchain/embedchain.py

@@ -3,7 +3,6 @@ import os
 
 from chromadb.utils import embedding_functions
 from dotenv import load_dotenv
-from gpt4all import GPT4All
 from langchain.docstore.document import Document
 from langchain.embeddings.openai import OpenAIEmbeddings
 
@@ -292,6 +291,8 @@ class OpenSourceApp(EmbedChain):
         super().__init__(db, ef)
 
     def get_llm_model_answer(self, prompt):
+        from gpt4all import GPT4All
+    
         global gpt4all_model
         if gpt4all_model is None:
             gpt4all_model = GPT4All("orca-mini-3b.ggmlv3.q4_0.bin")