Преглед изворни кода

update: import gpt4all locally (#146)

Import gpt4all locally to avoid error in case its not present.
Rohit Gupta пре 2 година
родитељ
комит
0e74688145
1 измењених фајлова са 2 додато и 1 уклоњено
  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")