reab5555 commited on
Commit
94700b8
·
verified ·
1 Parent(s): 78ed617

Update processing.py

Browse files
Files changed (1) hide show
  1. processing.py +2 -2
processing.py CHANGED
@@ -1,6 +1,6 @@
1
  from langchain.schema import HumanMessage
2
  from output_parser import attachment_parser, bigfive_parser, personality_parser
3
- from langchain_community.embeddings import OpenAIEmbeddings
4
  from langchain_community.vectorstores import FAISS
5
  from llm_loader import load_model # Import the function to load the model
6
  from config import openai_api_key # Import the API key from config.py
@@ -8,7 +8,7 @@ from langchain.chains import RetrievalQA
8
  import os
9
 
10
  # Initialize embeddings and FAISS index
11
- embedding_model = OpenAIEmbeddings(openai_api_key)
12
 
13
  # Path to the knowledge files
14
  knowledge_files = {
 
1
  from langchain.schema import HumanMessage
2
  from output_parser import attachment_parser, bigfive_parser, personality_parser
3
+ from langchain_openai import OpenAIEmbeddings
4
  from langchain_community.vectorstores import FAISS
5
  from llm_loader import load_model # Import the function to load the model
6
  from config import openai_api_key # Import the API key from config.py
 
8
  import os
9
 
10
  # Initialize embeddings and FAISS index
11
+ embedding_model = OpenAIEmbeddings(model="text-embedding-3-large", openai_api_key=openai_api_key)
12
 
13
  # Path to the knowledge files
14
  knowledge_files = {