fix: update FAISS import

WillBooster · Nov 6, 2023 · 928dc10 · 928dc10
1 parent dd2e09e
commit 928dc10
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/app_builder/builders/fix.py b/app_builder/builders/fix.py
@@ -1,9 +1,9 @@
 import logging
 import os
 
-from langchain import FAISS
 from langchain.chat_models import ChatOpenAI
 from langchain.prompts import load_prompt
+from langchain.vectorstores import FAISS
 from parsers.strict_pydantic_output_parser import StrictPydanticOutputParser
 from schema import (
     PublicInterfaceDocument,

diff --git a/app_builder/builders/source_code.py b/app_builder/builders/source_code.py
@@ -2,11 +2,11 @@
 import logging
 import os
 
-from langchain import FAISS
 from langchain.chat_models import ChatOpenAI
 from langchain.document_loaders import DirectoryLoader, TextLoader
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.prompts import load_prompt
+from langchain.vectorstores import FAISS
 from parsers.code_output_parser import CodeOutputParser
 from schema import PublicInterfaceDocument
 from util import (