Fix CI/CD (#1492)

2024-07-17 23:40:12 -07:00
parent 7441f1462d
commit fb5a3bfd95
10 changed files with 146 additions and 20 deletions
@@ -12,7 +12,8 @@ class GPT4AllEmbedder(BaseEmbedder):
        from langchain_community.embeddings import GPT4AllEmbeddings as LangchainGPT4AllEmbeddings

        model_name = self.config.model or "all-MiniLM-L6-v2-f16.gguf"
-        embeddings = LangchainGPT4AllEmbeddings(model_name=model_name)
+        gpt4all_kwargs = {'allow_download': 'True'}
+        embeddings = LangchainGPT4AllEmbeddings(model_name=model_name, gpt4all_kwargs=gpt4all_kwargs)
        embedding_fn = BaseEmbedder._langchain_default_concept(embeddings)
        self.set_embedding_fn(embedding_fn=embedding_fn)