broadfield-dev commited on
Commit
9a08a0f
·
verified ·
1 Parent(s): 522b0df

Update rss_processor.py

Browse files
Files changed (1) hide show
  1. rss_processor.py +2 -2
rss_processor.py CHANGED
@@ -1,6 +1,6 @@
1
  import os
2
  import feedparser
3
- from chromadb import Client, Documents
4
  from langchain.embeddings import HuggingFaceEmbeddings
5
  from langchain.docstore.document import Document
6
  import logging
@@ -163,7 +163,7 @@ def process_and_store_articles(articles):
163
  if not os.path.exists(LOCAL_DB_DIR):
164
  os.makedirs(LOCAL_DB_DIR)
165
 
166
- client = Client(persist_directory=LOCAL_DB_DIR)
167
  collection = client.get_or_create_collection(name=COLLECTION_NAME)
168
 
169
  try:
 
1
  import os
2
  import feedparser
3
+ from chromadb import PersistentClient
4
  from langchain.embeddings import HuggingFaceEmbeddings
5
  from langchain.docstore.document import Document
6
  import logging
 
163
  if not os.path.exists(LOCAL_DB_DIR):
164
  os.makedirs(LOCAL_DB_DIR)
165
 
166
+ client = PersistentClient(path=LOCAL_DB_DIR)
167
  collection = client.get_or_create_collection(name=COLLECTION_NAME)
168
 
169
  try: