abhivsh commited on
Commit
61c450e
·
verified ·
1 Parent(s): 95c952d

Update initialize.py

Browse files
Files changed (1) hide show
  1. initialize.py +2 -2
initialize.py CHANGED
@@ -13,9 +13,9 @@ OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
13
  # Using Google GenAI Text Embeddings
14
  # embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001", task_type="retrieval_query", google_api_key=GEMINI_API_KEY) ###
15
 
16
- #embedding_model = OpenAIEmbeddings(api_key=OPENAI_API_KEY, model="text-embedding-3-small")
17
 
18
- embedding_model = OllamaEmbeddings(model="mxbai-embed-large")
19
 
20
  # Create Embeddings for Searching the Splits
21
  persist_directory = './chroma/'
 
13
  # Using Google GenAI Text Embeddings
14
  # embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001", task_type="retrieval_query", google_api_key=GEMINI_API_KEY) ###
15
 
16
+ embedding_model = OpenAIEmbeddings(api_key=OPENAI_API_KEY, model="text-embedding-3-small")
17
 
18
+ #embedding_model = OllamaEmbeddings(model="mxbai-embed-large")
19
 
20
  # Create Embeddings for Searching the Splits
21
  persist_directory = './chroma/'