Zwea Htet commited on
Commit
2b31c7d
1 Parent(s): ef2a3f4
Files changed (1) hide show
  1. utils/util.py +33 -0
utils/util.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import requests
2
+
3
+
4
+ def validate(token: str):
5
+ api_endpoint = "https://api.openai.com/v1/chat/completions"
6
+ api_key = token
7
+
8
+ headers = {
9
+ "Content-Type" : "application/json",
10
+ "Authorization": f"Bearer {api_key}"
11
+ }
12
+
13
+ messages = [
14
+ {"role": "user", "content": "Say this is a test!"}
15
+ ]
16
+
17
+ data = {
18
+ "model": "gpt-3.5-turbo",
19
+ "messages": messages
20
+ }
21
+
22
+ response = requests.post(api_endpoint, json=data, headers=headers)
23
+ return response
24
+
25
+ def create_index(model):
26
+ index = model.initialize_index("bloomLlama")
27
+ return index
28
+
29
+ def get_response(vector_index, query_str):
30
+ print("query_str: ", query_str)
31
+ query_engine = vector_index.as_query_engine()
32
+ response = query_engine.query(query_str)
33
+ return response