wulin/qiuguo-iot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

阿里千问优化

Browse Source
This commit is contained in:
wulin 2023-10-12 09:09:31 +08:00
parent 6ef3d637be
commit af64a3caaa
2 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
iot-common/iot-third/src/main/java/com/qiuguo/iot/third/nlp/AliYunQianWen.java
View File

@ -28,13 +28,7 @@ public class AliYunQianWen {
/**
* 阿里千问配置参数
*/
static final QwenParam qwenParam = QwenParam.builder().model(Generation.Models.QWEN_PLUS)
.resultFormat(QwenParam.ResultFormat.MESSAGE)
.topP(0.8)
.apiKey(SpringUtil.getProperty("Ali.qianwen"))
.enableSearch(true)
.incrementalOutput(true) // get streaming output incrementally
.build();;
QwenParam qwenParam;
/**
* 消息管理，上下文。默认最多10个
@ -45,6 +39,16 @@ public class AliYunQianWen {
GenerationResult lastGenerationResult = null;
public AliYunQianWen(String key) {
qwenParam = QwenParam.builder().model(Generation.Models.QWEN_PLUS)
.resultFormat(QwenParam.ResultFormat.MESSAGE)
.topP(0.8)
.apiKey(key)
.enableSearch(true)
.incrementalOutput(true) // get streaming output incrementally
.build();
}
public void sendMessage(String msg,
Consumer<? super String> onNext,

3
iot-common/iot-third/src/main/java/com/qiuguo/iot/third/service/QWenService.java
View File

@ -1,5 +1,6 @@
package com.qiuguo.iot.third.service;
import cn.hutool.extra.spring.SpringUtil;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.qiuguo.iot.base.constans.RedisConstans;
@ -39,7 +40,7 @@ public class QWenService {
public Mono<QWenReplyResponse> communication(TongYiCommunicationRest rest, Consumer<? super String> onNext){
AliYunQianWen aliQianWen = null;
if (!qianwenGroup.containsKey(rest.getOnlyId())) {
aliQianWen = new AliYunQianWen();
aliQianWen = new AliYunQianWen(SpringUtil.getProperty("Ali.qianwen"));
qianwenGroup.put(rest.getOnlyId(), aliQianWen);
} else {
aliQianWen = qianwenGroup.get(rest.getOnlyId());