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 str(response)