Spaces:
Runtime error
Runtime error
Embedding Model changed
Browse files- initialize.py +5 -1
initialize.py
CHANGED
@@ -1,12 +1,16 @@
|
|
1 |
from langchain_google_genai import GoogleGenerativeAIEmbeddings
|
|
|
2 |
from langchain_community.vectorstores import Chroma
|
3 |
import os
|
4 |
import gen_splits
|
5 |
|
6 |
GEMINI_API_KEY = os.environ.get('GEMINI_API_KEY')
|
|
|
7 |
|
8 |
# Using Google GenAI Text Embeddings
|
9 |
-
embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001", task_type="retrieval_query", google_api_key=GEMINI_API_KEY)
|
|
|
|
|
10 |
|
11 |
# Create Embeddings for Searching the Splits
|
12 |
persist_directory = './chroma/'
|
|
|
1 |
from langchain_google_genai import GoogleGenerativeAIEmbeddings
|
2 |
+
from langchain.embeddings.openai import OpenAIEmbeddings
|
3 |
from langchain_community.vectorstores import Chroma
|
4 |
import os
|
5 |
import gen_splits
|
6 |
|
7 |
GEMINI_API_KEY = os.environ.get('GEMINI_API_KEY')
|
8 |
+
OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
|
9 |
|
10 |
# Using Google GenAI Text Embeddings
|
11 |
+
# embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001", task_type="retrieval_query", google_api_key=GEMINI_API_KEY) ###
|
12 |
+
|
13 |
+
embedding_model = OpenAIEmbeddings(api_key=OPENAI_API_KEY)
|
14 |
|
15 |
# Create Embeddings for Searching the Splits
|
16 |
persist_directory = './chroma/'
|