utils/util.py

import requests


def validate(token: str):
    api_endpoint = "https://api.openai.com/v1/chat/completions"
    api_key = token

    headers = {
        "Content-Type" : "application/json",
        "Authorization": f"Bearer {api_key}"
    }

    messages = [
        {"role": "user", "content": "Say this is a test!"}
    ]

    data = {
        "model": "gpt-3.5-turbo",
        "messages": messages
    }

    response = requests.post(api_endpoint, json=data, headers=headers)
    return response

def create_index(model):
    index = model.initialize_index("bloomLlama")
    return index

def get_response(vector_index, query_str):
    print("query_str: ", query_str)
    query_engine = vector_index.as_query_engine()
    response = query_engine.query(query_str)
    return response