Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
KnowledgePlanet
road-map
chatglm-sdk-java
提交
58438b86
chatglm-sdk-java
项目概览
KnowledgePlanet
/
road-map
/
chatglm-sdk-java
通知
253
Star
20
Fork
10
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
chatglm-sdk-java
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
提交
58438b86
编写于
3月 30, 2024
作者:
小傅哥
⛹
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
docs:v2.1 发布 - 补充单测
上级
a6014d59
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
25 addition
and
3 deletion
+25
-3
src/test/java/cn/bugstack/chatglm/test/ApiTest.java
src/test/java/cn/bugstack/chatglm/test/ApiTest.java
+25
-3
未找到文件。
src/test/java/cn/bugstack/chatglm/test/ApiTest.java
浏览文件 @
58438b86
...
@@ -40,7 +40,7 @@ public class ApiTest {
...
@@ -40,7 +40,7 @@ public class ApiTest {
// 1. 配置文件
// 1. 配置文件
Configuration
configuration
=
new
Configuration
();
Configuration
configuration
=
new
Configuration
();
configuration
.
setApiHost
(
"https://open.bigmodel.cn/"
);
configuration
.
setApiHost
(
"https://open.bigmodel.cn/"
);
configuration
.
setApiSecretKey
(
"
62ddec38b1d0b9a7b0fddaf271e6ed90.HpD0SUBUlvqd05ey
"
);
configuration
.
setApiSecretKey
(
"
74e0b7fff9a578e2137300bcd58ae062.q0ABClJJS50rHQnn
"
);
configuration
.
setLevel
(
HttpLoggingInterceptor
.
Level
.
BODY
);
configuration
.
setLevel
(
HttpLoggingInterceptor
.
Level
.
BODY
);
// 2. 会话工厂
// 2. 会话工厂
OpenAiSessionFactory
factory
=
new
DefaultOpenAiSessionFactory
(
configuration
);
OpenAiSessionFactory
factory
=
new
DefaultOpenAiSessionFactory
(
configuration
);
...
@@ -125,7 +125,7 @@ public class ApiTest {
...
@@ -125,7 +125,7 @@ public class ApiTest {
// 入参;模型、请求信息
// 入参;模型、请求信息
ChatCompletionRequest
request
=
new
ChatCompletionRequest
();
ChatCompletionRequest
request
=
new
ChatCompletionRequest
();
request
.
setModel
(
Model
.
GLM_
3_5_TURBO
);
// GLM_3_5_TURBO、GLM_4
request
.
setModel
(
Model
.
GLM_
4
);
// GLM_3_5_TURBO、GLM_4
request
.
setIsCompatible
(
false
);
request
.
setIsCompatible
(
false
);
// 24年1月发布的 glm-3-turbo、glm-4 支持函数、知识库、联网功能
// 24年1月发布的 glm-3-turbo、glm-4 支持函数、知识库、联网功能
request
.
setTools
(
new
ArrayList
<
ChatCompletionRequest
.
Tool
>()
{
request
.
setTools
(
new
ArrayList
<
ChatCompletionRequest
.
Tool
>()
{
...
@@ -280,7 +280,7 @@ public class ApiTest {
...
@@ -280,7 +280,7 @@ public class ApiTest {
* 同步请求
* 同步请求
*/
*/
@Test
@Test
public
void
test_completions_sync
()
throws
Exception
{
public
void
test_completions_sync
_01
()
throws
Exception
{
// 入参;模型、请求信息
// 入参;模型、请求信息
ChatCompletionRequest
request
=
new
ChatCompletionRequest
();
ChatCompletionRequest
request
=
new
ChatCompletionRequest
();
request
.
setModel
(
Model
.
GLM_3_5_TURBO
);
// chatGLM_6b_SSE、chatglm_lite、chatglm_lite_32k、chatglm_std、chatglm_pro
request
.
setModel
(
Model
.
GLM_3_5_TURBO
);
// chatGLM_6b_SSE、chatglm_lite、chatglm_lite_32k、chatglm_std、chatglm_pro
...
@@ -312,6 +312,28 @@ public class ApiTest {
...
@@ -312,6 +312,28 @@ public class ApiTest {
log
.
info
(
"测试结果:{}"
,
JSON
.
toJSONString
(
response
));
log
.
info
(
"测试结果:{}"
,
JSON
.
toJSONString
(
response
));
}
}
@Test
public
void
test_completions_sync_02
()
throws
Exception
{
// 入参;模型、请求信息
ChatCompletionRequest
request
=
new
ChatCompletionRequest
();
request
.
setModel
(
Model
.
GLM_3_5_TURBO
);
// chatGLM_6b_SSE、chatglm_lite、chatglm_lite_32k、chatglm_std、chatglm_pro
request
.
setPrompt
(
new
ArrayList
<
ChatCompletionRequest
.
Prompt
>()
{
private
static
final
long
serialVersionUID
=
-
7988151926241837899L
;
{
add
(
ChatCompletionRequest
.
Prompt
.
builder
()
.
role
(
Role
.
user
.
getCode
())
.
content
(
"1+1"
)
.
build
());
}
});
ChatCompletionSyncResponse
response
=
openAiSession
.
completionsSync
(
request
);
log
.
info
(
"测试结果:{}"
,
JSON
.
toJSONString
(
response
));
System
.
out
.
println
(
response
.
getChoices
().
get
(
0
).
getMessage
().
getContent
());
}
@Test
@Test
public
void
test_genImages
()
throws
Exception
{
public
void
test_genImages
()
throws
Exception
{
ImageCompletionRequest
request
=
new
ImageCompletionRequest
();
ImageCompletionRequest
request
=
new
ImageCompletionRequest
();
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录