Spaces:
Runtime error
Runtime error
Update processing.py
Browse files- 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
|
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 = {
|