-
Notifications
You must be signed in to change notification settings - Fork 24
Expand file tree
/
Copy pathAssistantSimple.java
More file actions
127 lines (114 loc) · 5.63 KB
/
AssistantSimple.java
File metadata and controls
127 lines (114 loc) · 5.63 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
import com.alibaba.dashscope.assistants.Assistant;
import com.alibaba.dashscope.assistants.AssistantParam;
import com.alibaba.dashscope.assistants.Assistants;
import com.alibaba.dashscope.common.GeneralListParam;
import com.alibaba.dashscope.common.ListResult;
import com.alibaba.dashscope.exception.ApiException;
import com.alibaba.dashscope.exception.InputRequiredException;
import com.alibaba.dashscope.exception.InvalidateParameter;
import com.alibaba.dashscope.exception.NoApiKeyException;
import com.alibaba.dashscope.threads.AssistantThread;
import com.alibaba.dashscope.threads.ThreadParam;
import com.alibaba.dashscope.threads.Threads;
import com.alibaba.dashscope.threads.messages.Messages;
import com.alibaba.dashscope.threads.messages.TextMessageParam;
import com.alibaba.dashscope.threads.messages.ThreadMessage;
import com.alibaba.dashscope.threads.runs.Run;
import com.alibaba.dashscope.threads.runs.RunParam;
import com.alibaba.dashscope.threads.runs.Runs;
import com.alibaba.dashscope.tools.search.ToolQuarkSearch;
import java.text.SimpleDateFormat;
import java.util.Date;
public class AssistantSimple {
public static void assistantUsage() throws ApiException, NoApiKeyException, InputRequiredException, InvalidateParameter, InterruptedException {
Assistants assistants = new Assistants();
// build assistant parameters
AssistantParam param = AssistantParam.builder()
// 此处以qwen-max为例，可按需更换模型名称。模型列表：https://help.aliyun.com/zh/model-studio/getting-started/models
.model("qwen-max")
.name("intelligent guide")
.description("a smart guide")
.instructions("You are a helpful assistant.")
.tool(ToolQuarkSearch.builder().build())
.topP(0.8f)
.topK(8)
.temperature(0.8f)
.maxTokens(2048)
.build();
// Log before creating assistant with millisecond precision
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
long startTime = System.currentTimeMillis();
System.out.println(sdf.format(new Date(startTime)) +
" - Starting assistant creation");
Assistant assistant = assistants.create(param);
// Log after creating assistant with duration
long endTime = System.currentTimeMillis();
System.out.println(sdf.format(new Date(endTime)) +
" - Assistant creation completed, duration: " +
(endTime - startTime) + "ms");
System.out.println(assistant);
// sleep for 1 second
// Log before creating assistant with millisecond precision
startTime = System.currentTimeMillis();
System.out.println(sdf.format(new Date(startTime)) +
" - Starting assistant creation");
assistant = assistants.create(param);
// Log after creating assistant with duration
endTime = System.currentTimeMillis();
System.out.println(sdf.format(new Date(endTime)) +
" - Assistant creation completed, duration: " +
(endTime - startTime) + "ms");
System.out.println(assistant);
// update assistant parameters
// param = AssistantParam.builder()
// // 此处以qwen-max为例，可按需更换模型名称。模型列表：https://help.aliyun.com/zh/model-studio/getting-started/models
// .model("qwen-plus")
// .name("intelligent guide")
// .description("a smart guide")
// .instructions("You are a helpful assistant.")
// .topP(0.7)
// .topK(9)
// .temperature(0.7)
// .maxTokens(1024)
// .build();
// assistant = assistants.update(assistant.getId(), param);
// System.out.println(assistant);
// create a thread
Threads threads = new Threads();
AssistantThread assistantThread = threads.create(ThreadParam.builder().build());
// create a message to thread
Messages messages = new Messages();
ThreadMessage message = messages.create(assistantThread.getId(), TextMessageParam.builder().role("user").content("如何做出美味的牛肉炖土豆？").build());
System.out.println(message);
// create run
Runs runs = new Runs();
RunParam runParam = RunParam.builder()
.assistantId(assistant.getId())
.topP(0.7f)
.topK(7)
.temperature(0.7f)
.maxTokens(1024)
.build();
Run run = runs.create(assistantThread.getId(), runParam);
System.out.println(run);
// wait for run completed
while(true){
if(run.getStatus().equals(Run.Status.CANCELLED) ||
run.getStatus().equals(Run.Status.COMPLETED) ||
run.getStatus().equals(Run.Status.FAILED) ||
run.getStatus().equals(Run.Status.REQUIRES_ACTION)||
run.getStatus().equals(Run.Status.EXPIRED)){
break;
}else{
Thread.sleep(1000);
}
run = runs.retrieve(assistantThread.getId(), run.getId());
System.out.println(run);
}
ListResult<ThreadMessage> threadMessages = messages.list(assistantThread.getId(), GeneralListParam.builder().build());
System.out.println(threadMessages);
}
public static void main(String[] args) throws ApiException, NoApiKeyException, InputRequiredException, InvalidateParameter, InterruptedException {
assistantUsage();
}
}