abhivsh commited on
Commit
9091381
·
verified ·
1 Parent(s): c731612

Create setup.py

Browse files
Files changed (1) hide show
  1. setup.py +11 -0
setup.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain_google_genai import GoogleGenerativeAIEmbeddings
2
+ from langchain_community.vectorstores import Chroma
3
+
4
+ # Using Google GenAI Text Embeddings
5
+ embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001", task_type="retrieval_document", google_api_key=GEMINI_API_KEY)
6
+
7
+ # Create Embeddings for Searching the Splits
8
+ persist_directory = './chroma/'
9
+
10
+ vectordb = Chroma.from_documents(documents=splits, persist_directory=persist_directory, embedding=embedding_model)
11
+ vectordb.persist()