danielRamon commited on
Commit
458a8ea
·
1 Parent(s): 585c783

feat ✨: Update dependencies for improved functionality

Browse files
Files changed (2) hide show
  1. chroma_utils.py +8 -6
  2. 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 = OllamaEmbeddings(
6
- model="jina/jina-embeddings-v2-base-es") # Initialize embeddings
 
 
 
 
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