Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
KnowledgePlanet
road-map
chatglm-sdk-java
提交
07f0aad3
chatglm-sdk-java
项目概览
KnowledgePlanet
/
road-map
/
chatglm-sdk-java
通知
254
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看板
提交
07f0aad3
编写于
11月 10, 2023
作者:
小傅哥
⛹
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
feat:增加同步请求,便于有写场景需要一次处理结果
上级
8699cadf
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
82 addition
and
3 deletion
+82
-3
src/main/java/cn/bugstack/chatglm/session/OpenAiSession.java
src/main/java/cn/bugstack/chatglm/session/OpenAiSession.java
+5
-0
src/main/java/cn/bugstack/chatglm/session/defaults/DefaultOpenAiSession.java
...gstack/chatglm/session/defaults/DefaultOpenAiSession.java
+49
-3
src/test/java/cn/bugstack/chatglm/test/ApiTest.java
src/test/java/cn/bugstack/chatglm/test/ApiTest.java
+28
-0
未找到文件。
src/main/java/cn/bugstack/chatglm/session/OpenAiSession.java
浏览文件 @
07f0aad3
package
cn.bugstack.chatglm.session
;
import
cn.bugstack.chatglm.model.ChatCompletionRequest
;
import
cn.bugstack.chatglm.model.ChatCompletionResponse
;
import
com.fasterxml.jackson.core.JsonProcessingException
;
import
okhttp3.sse.EventSource
;
import
okhttp3.sse.EventSourceListener
;
import
java.util.concurrent.CompletableFuture
;
/**
* @author 小傅哥,微信:fustack
* @description 会话服务接口
...
...
@@ -15,4 +18,6 @@ public interface OpenAiSession {
EventSource
completions
(
ChatCompletionRequest
chatCompletionRequest
,
EventSourceListener
eventSourceListener
)
throws
JsonProcessingException
;
CompletableFuture
<
String
>
completions
(
ChatCompletionRequest
chatCompletionRequest
)
throws
InterruptedException
;
}
src/main/java/cn/bugstack/chatglm/session/defaults/DefaultOpenAiSession.java
浏览文件 @
07f0aad3
...
...
@@ -2,14 +2,23 @@ package cn.bugstack.chatglm.session.defaults;
import
cn.bugstack.chatglm.IOpenAiApi
;
import
cn.bugstack.chatglm.model.ChatCompletionRequest
;
import
cn.bugstack.chatglm.model.ChatCompletionResponse
;
import
cn.bugstack.chatglm.model.EventType
;
import
cn.bugstack.chatglm.session.Configuration
;
import
cn.bugstack.chatglm.session.OpenAiSession
;
import
com.alibaba.fastjson.JSON
;
import
com.fasterxml.jackson.core.JsonProcessingException
;
import
lombok.extern.slf4j.Slf4j
;
import
okhttp3.MediaType
;
import
okhttp3.Request
;
import
okhttp3.RequestBody
;
import
okhttp3.Response
;
import
okhttp3.sse.EventSource
;
import
okhttp3.sse.EventSourceListener
;
import
org.jetbrains.annotations.Nullable
;
import
java.util.concurrent.CompletableFuture
;
import
java.util.concurrent.CountDownLatch
;
/**
* @author 小傅哥,微信:fustack
...
...
@@ -17,6 +26,7 @@ import okhttp3.sse.EventSourceListener;
* @github https://github.com/fuzhengwei
* @Copyright 公众号:bugstack虫洞栈 | 博客:https://bugstack.cn - 沉淀、分享、成长,让自己和他人都能有所收获!
*/
@Slf4j
public
class
DefaultOpenAiSession
implements
OpenAiSession
{
/**
...
...
@@ -28,11 +38,8 @@ public class DefaultOpenAiSession implements OpenAiSession {
*/
private
final
EventSource
.
Factory
factory
;
private
IOpenAiApi
openAiApi
;
public
DefaultOpenAiSession
(
Configuration
configuration
)
{
this
.
configuration
=
configuration
;
this
.
openAiApi
=
configuration
.
getOpenAiApi
();
this
.
factory
=
configuration
.
createRequestFactory
();
}
...
...
@@ -48,4 +55,43 @@ public class DefaultOpenAiSession implements OpenAiSession {
return
factory
.
newEventSource
(
request
,
eventSourceListener
);
}
@Override
public
CompletableFuture
<
String
>
completions
(
ChatCompletionRequest
chatCompletionRequest
)
throws
InterruptedException
{
// 用于执行异步任务并获取结果
CompletableFuture
<
String
>
future
=
new
CompletableFuture
<>();
StringBuffer
dataBuffer
=
new
StringBuffer
();
// 构建请求信息
Request
request
=
new
Request
.
Builder
()
.
url
(
configuration
.
getApiHost
().
concat
(
IOpenAiApi
.
v3_completions
).
replace
(
"{model}"
,
chatCompletionRequest
.
getModel
().
getCode
()))
.
post
(
RequestBody
.
create
(
MediaType
.
parse
(
"application/json"
),
chatCompletionRequest
.
toString
()))
.
build
();
// 异步响应请求
factory
.
newEventSource
(
request
,
new
EventSourceListener
()
{
@Override
public
void
onEvent
(
EventSource
eventSource
,
@Nullable
String
id
,
@Nullable
String
type
,
String
data
)
{
ChatCompletionResponse
response
=
JSON
.
parseObject
(
data
,
ChatCompletionResponse
.
class
);
// type 消息类型,add 增量,finish 结束,error 错误,interrupted 中断
if
(
EventType
.
add
.
getCode
().
equals
(
type
))
{
dataBuffer
.
append
(
response
.
getData
());
}
else
if
(
EventType
.
finish
.
getCode
().
equals
(
type
))
{
future
.
complete
(
dataBuffer
.
toString
());
}
}
@Override
public
void
onClosed
(
EventSource
eventSource
)
{
future
.
completeExceptionally
(
new
RuntimeException
(
"Request closed before completion"
));
}
@Override
public
void
onFailure
(
EventSource
eventSource
,
@Nullable
Throwable
t
,
@Nullable
Response
response
)
{
future
.
completeExceptionally
(
new
RuntimeException
(
"Request closed before completion"
));
}
});
return
future
;
}
}
src/test/java/cn/bugstack/chatglm/test/ApiTest.java
浏览文件 @
07f0aad3
...
...
@@ -17,7 +17,9 @@ import org.junit.Before;
import
org.junit.Test
;
import
java.util.ArrayList
;
import
java.util.concurrent.CompletableFuture
;
import
java.util.concurrent.CountDownLatch
;
import
java.util.concurrent.ExecutionException
;
/**
* @author 小傅哥,微信:fustack
...
...
@@ -108,6 +110,32 @@ public class ApiTest {
new
CountDownLatch
(
1
).
await
();
}
/**
* 同步请求
*/
@Test
public
void
test_completions_future
()
throws
ExecutionException
,
InterruptedException
{
// 入参;模型、请求信息
ChatCompletionRequest
request
=
new
ChatCompletionRequest
();
request
.
setModel
(
Model
.
CHATGLM_LITE
);
// 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
(
"写个java冒泡排序"
)
.
build
());
}
});
CompletableFuture
<
String
>
future
=
openAiSession
.
completions
(
request
);
String
response
=
future
.
get
();
log
.
info
(
"测试结果:{}"
,
response
);
}
@Test
public
void
test_curl
()
{
// 1. 配置文件
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录