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/'
|