Skip to content
LlamaIndex: Adding Personal Data to LLMs
  • AI Chat
  • Code
  • Report
  • %%capture
    %pip install llama-index openai pypdf

    Loading Data and Creating the Index

    from llama_index import TreeIndex, SimpleDirectoryReader
    
    resume = SimpleDirectoryReader("Private-Data").load_data()
    new_index = TreeIndex.from_documents(resume)

    Running Query on text-davinci-003

    query_engine = new_index.as_query_engine()
    response = query_engine.query("When did Abid graduated?")
    print(response)
    response = query_engine.query("What is the name of certification that Abid received?")
    print(response)

    Saving and Loading the Context

    new_index.storage_context.persist()
    from llama_index import StorageContext, load_index_from_storage
    
    storage_context = StorageContext.from_defaults(persist_dir="./storage")
    index = load_index_from_storage(storage_context)
    
    query_engine = index.as_query_engine()
    response = query_engine.query("What is Abid's job title?")
    print(response)

    Chatbot

    query_engine = index.as_chat_engine()
    response = query_engine.chat("What is the job title of Abid in 2021?")
    print(response)
    response = query_engine.chat("What else he do during that time?")
    print(response)