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
|