llama_index examples

query-notion-page.py

import logging
import sys
import openai

logging.basicConfig(stream=sys.stdout, level=logging.INFO)
logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))

from llama_index.llms import OpenAI
from llama_index import ListIndex, NotionPageReader, ServiceContext
import os

os.environ["OPENAI_API_KEY"] = ""
openai.api_key = os.environ["OPENAI_API_KEY"]

chatgpt = OpenAI(temperature=0, model="gpt-3.5-turbo")
service_context = ServiceContext.from_defaults(llm=chatgpt, chunk_size=1024)

integration_token = ""
page_ids = [""]
documents = NotionPageReader(integration_token=integration_token).load_data(
    page_ids=page_ids
)

index = ListIndex.from_documents(documents)

# set Logging to DEBUG for more detailed outputs
query_engine = index.as_query_engine()
response = query_engine.query("query here")

print(response)

summarize.py

import os
import logging
import sys
import openai

os.environ["OPENAI_API_KEY"] = ''
openai.api_key = os.environ["OPENAI_API_KEY"]

from llama_index.llms import OpenAI

from llama_index import (
    SimpleDirectoryReader,
    ListIndex,
    ServiceContext,
)

chatgpt = OpenAI(temperature=0, model="gpt-3.5-turbo")
service_context = ServiceContext.from_defaults(llm=chatgpt, chunk_size=1024)

documents = SimpleDirectoryReader("./data").load_data()

index = ListIndex.from_documents(documents)

prompt = "Summarize..."

response = index.as_query_engine(response_mode="tree_summarize").query(prompt)

print(response)