07Codex07 commited on
Commit
19e0262
·
1 Parent(s): 9f40f90

Ensure HF dataset downloads on startup

Browse files
Files changed (1) hide show
  1. main_api.py +3 -2
main_api.py CHANGED
@@ -13,12 +13,13 @@ import uvicorn
13
 
14
  # import your existing modules (assumed in same directory)
15
  from memory_store import init_db, save_message, get_last_messages, clear_user_memory, build_gradio_history # :contentReference[oaicite:4]{index=4}
16
- from chatbot_retriever import build_or_load_indexes, hybrid_retrieve, retrieve_node_from_rows, load_all_docs # :contentReference[oaicite:5]{index=5}
17
  from chatbot_graph import SYSTEM_PROMPT, call_llm, _extract_answer_from_response # :contentReference[oaicite:6]{index=6}
18
 
 
19
  # ----------------- CORS SETUP -----------------
20
  from fastapi.middleware.cors import CORSMiddleware
21
-
22
  app = FastAPI(title="RAG Chat Backend", version="1.0")
23
 
24
  from fastapi.middleware.cors import CORSMiddleware
 
13
 
14
  # import your existing modules (assumed in same directory)
15
  from memory_store import init_db, save_message, get_last_messages, clear_user_memory, build_gradio_history # :contentReference[oaicite:4]{index=4}
16
+ from chatbot_retriever import build_or_load_indexes, hybrid_retrieve, retrieve_node_from_rows, load_all_docs, ensure_data_dir # :contentReference[oaicite:5]{index=5}
17
  from chatbot_graph import SYSTEM_PROMPT, call_llm, _extract_answer_from_response # :contentReference[oaicite:6]{index=6}
18
 
19
+ ensure_data_dir()
20
  # ----------------- CORS SETUP -----------------
21
  from fastapi.middleware.cors import CORSMiddleware
22
+ ensure_data_dir()
23
  app = FastAPI(title="RAG Chat Backend", version="1.0")
24
 
25
  from fastapi.middleware.cors import CORSMiddleware