Spaces:
Runtime error
Runtime error
Create setup.py
Browse files
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()
|