console.py

from kb import index_knowledge_base, ask_knowledge_base

kb.index_knowledge_base('./datasets') # dossier avec des pdfs ou d'autres documents
kb.ask_knowledge_base('Quels sont les 3 techniques qui fonctionnent pour éliminer les moustiques d un marécage ?')

kb.py

from typing import Optional

import os

from llama_index.core import VectorStoreIndex, SimpleDirectoryReader

os.environ["OPENAI_API_KEY"] = 'sk-xxxxxxxxx'

vector_store: Optional[VectorStoreIndex] = None

def index_knowledge_base(path: str):
    global vector_store

    documents = SimpleDirectoryReader(path).load_data()
    vector_store = VectorStoreIndex.from_documents(documents)


def ask_knowledge_base(question: str) -> str:
    query_str = question
    query_engine = vector_store.as_query_engine()
    response = query_engine.query(query_str)
    return response.response