| 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 |