abhivsh commited on
Commit
ecef4cf
·
verified ·
1 Parent(s): 5f99ca5

Embedding Model changed

Browse files
Files changed (1) hide show
  1. 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/'