broadfield-dev commited on
Commit
46fd3e1
·
verified ·
1 Parent(s): bbf6a58

Update rss_processor.py

Browse files
Files changed (1) hide show
  1. rss_processor.py +3 -0
rss_processor.py CHANGED
@@ -160,6 +160,9 @@ def categorize_feed(url):
160
  return "Uncategorized"
161
 
162
  def process_and_store_articles(articles):
 
 
 
163
  vector_db = Chroma(
164
  persist_directory=LOCAL_DB_DIR,
165
  embedding_function=get_embedding_model(),
 
160
  return "Uncategorized"
161
 
162
  def process_and_store_articles(articles):
163
+ if not os.path.exists(LOCAL_DB_DIR):
164
+ os.makedirs(LOCAL_DB_DIR)
165
+
166
  vector_db = Chroma(
167
  persist_directory=LOCAL_DB_DIR,
168
  embedding_function=get_embedding_model(),