File tree 1 file changed +53
-0
lines changed
1 file changed +53
-0
lines changed Original file line number Diff line number Diff line change
1
+ import openai
2
+ import time
3
+
4
+ # gets API Key from environment variable OPENAI_API_KEY
5
+ client = openai .OpenAI ()
6
+
7
+ assistant = client .beta .assistants .create (
8
+ name = "Math Tutor" ,
9
+ instructions = "You are a personal math tutor. Write and run code to answer math questions." ,
10
+ tools = [{"type" : "code_interpreter" }],
11
+ model = "gpt-4-1106-preview" ,
12
+ )
13
+
14
+ thread = client .beta .threads .create ()
15
+
16
+ message = client .beta .threads .messages .create (
17
+ thread_id = thread .id ,
18
+ role = "user" ,
19
+ content = "I need to solve the equation `3x + 11 = 14`. Can you help me?"
20
+ )
21
+
22
+ run = client .beta .threads .runs .create (
23
+ thread_id = thread .id ,
24
+ assistant_id = assistant .id ,
25
+ instructions = "Please address the user as Jane Doe. The user has a premium account."
26
+ )
27
+
28
+ print ("checking assistant status. " )
29
+ while True :
30
+ run = client .beta .threads .runs .retrieve (
31
+ thread_id = thread .id ,
32
+ run_id = run .id
33
+ )
34
+
35
+ if run .status == "completed" :
36
+ print ("done!" )
37
+ messages = client .beta .threads .messages .list (
38
+ thread_id = thread .id
39
+ )
40
+
41
+ print ("messages: " )
42
+ for message in messages :
43
+ print ({
44
+ "role" : message .role ,
45
+ "message" : message .content [0 ].text .value
46
+ })
47
+
48
+ client .beta .assistants .delete (assistant .id )
49
+
50
+ break
51
+ else :
52
+ print ("in progress..." )
53
+ time .sleep (5 )
You can’t perform that action at this time.
0 commit comments