File size: 762 Bytes
552d350
 
2d413a5
552d350
2d413a5
 
9d2bec8
5e4e457
2d413a5
5e4e457
 
 
 
2d413a5
 
5e4e457
 
 
 
 
2d413a5
5e4e457
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from llama_index.core import SimpleDirectoryReader, VectorStoreIndex
from llama_index.core import Settings
import logging, sys

logging.basicConfig(stream=sys.stdout, level=logging.INFO, force=True)
logger = logging.getLogger(__name__)

def create_symptom_index():
    logger.info("build_symptom_index: Loading documents from data directory…")
    documents = SimpleDirectoryReader(
        input_dir="data",
        filename_as_id=True
    ).load_data()

    logger.info(f"build_symptom_index: Creating vector index from {len(documents)} documents…")
    symptom_index = VectorStoreIndex.from_documents(
        documents,
        show_progress=True
    )

    logger.info("build_symptom_index: Symptom index created successfully")
    return symptom_index