Spaces:
Sleeping
Sleeping
Commit
·
458a8ea
1
Parent(s):
585c783
feat ✨: Update dependencies for improved functionality
Browse files- chroma_utils.py +8 -6
- requirements.txt +3 -2
chroma_utils.py
CHANGED
|
@@ -1,9 +1,14 @@
|
|
| 1 |
from langchain_chroma import Chroma
|
| 2 |
-
from langchain_ollama import OllamaEmbeddings
|
|
|
|
| 3 |
from langchain_core.documents import Document
|
| 4 |
|
| 5 |
-
embed =
|
| 6 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
|
| 9 |
def save_handbook_to_chroma(handbook_data: list) -> bool:
|
|
@@ -16,9 +21,6 @@ def save_handbook_to_chroma(handbook_data: list) -> bool:
|
|
| 16 |
Returns:
|
| 17 |
bool: True if the handbook is saved correctly, False otherwise.
|
| 18 |
"""
|
| 19 |
-
embeddings = OllamaEmbeddings(
|
| 20 |
-
model="llama3.1",
|
| 21 |
-
)
|
| 22 |
|
| 23 |
documents = []
|
| 24 |
for chapter in handbook_data:
|
|
|
|
| 1 |
from langchain_chroma import Chroma
|
| 2 |
+
# from langchain_ollama import OllamaEmbeddings
|
| 3 |
+
from langchain_huggingface import HuggingFaceEmbeddings
|
| 4 |
from langchain_core.documents import Document
|
| 5 |
|
| 6 |
+
embed = HuggingFaceEmbeddings(
|
| 7 |
+
model_name="sentence-transformers/all-mpnet-base-v2")
|
| 8 |
+
|
| 9 |
+
|
| 10 |
+
# embed = OllamaEmbeddings(
|
| 11 |
+
# model="jina/jina-embeddings-v2-base-es") # Initialize embeddings
|
| 12 |
|
| 13 |
|
| 14 |
def save_handbook_to_chroma(handbook_data: list) -> bool:
|
|
|
|
| 21 |
Returns:
|
| 22 |
bool: True if the handbook is saved correctly, False otherwise.
|
| 23 |
"""
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
documents = []
|
| 26 |
for chapter in handbook_data:
|
requirements.txt
CHANGED
|
@@ -2,7 +2,8 @@ requests==2.32.3
|
|
| 2 |
beautifulsoup4==4.12.3
|
| 3 |
pandas==2.2.3
|
| 4 |
langchain==0.3.6
|
| 5 |
-
langchain-ollama==0.2.0
|
| 6 |
langchain_chroma==0.1.4
|
| 7 |
streamlit==1.39.0
|
| 8 |
-
protobuf==3.20
|
|
|
|
|
|
| 2 |
beautifulsoup4==4.12.3
|
| 3 |
pandas==2.2.3
|
| 4 |
langchain==0.3.6
|
| 5 |
+
# langchain-ollama==0.2.0
|
| 6 |
langchain_chroma==0.1.4
|
| 7 |
streamlit==1.39.0
|
| 8 |
+
protobuf==3.20
|
| 9 |
+
langchain-huggingface==0.1.2
|