Browse Source

feat: ai

master
wangxiang 1 year ago
parent
commit
c8a0e2805f
No known key found for this signature in database
GPG Key ID: 1BA7946AB6B232E4
  1. 3
      kicc-platform/kicc-platform-biz/kicc-system-biz/src/main/java/com/cloud/kicc/system/config/OpenAiConfigProperties.java
  2. 2
      kicc-platform/kicc-platform-biz/kicc-system-biz/src/main/java/com/cloud/kicc/system/controller/ChatGptController.java
  3. 3
      kicc-platform/kicc-platform-biz/kicc-system-biz/src/main/java/com/cloud/kicc/system/service/impl/ImContentServiceImpl.java
  4. 1
      kicc-platform/kicc-platform-biz/kicc-system-biz/src/main/java/com/cloud/kicc/system/util/AiUtil.java

3
kicc-platform/kicc-platform-biz/kicc-system-biz/src/main/java/com/cloud/kicc/system/config/OpenAiConfigProperties.java

@ -57,7 +57,4 @@ public class OpenAiConfigProperties {
/** Context Window Max */ /** Context Window Max */
private String contextMessageMaxCount = "14"; private String contextMessageMaxCount = "14";
/** The maximum number of tokens allowed for the generated answer. By default, the number of tokens the model can return will be (4096 - prompt tokens). */
private Integer completionMaxTokens = 300;
} }

2
kicc-platform/kicc-platform-biz/kicc-system-biz/src/main/java/com/cloud/kicc/system/controller/ChatGptController.java

@ -46,7 +46,7 @@ public class ChatGptController {
@Inner(false) @Inner(false)
@PostMapping("/chat/completions") @PostMapping("/chat/completions")
public R chatCompletions(@Valid @RequestBody ImContent imContent, @RequestParam boolean contextEnable) { public R chatCompletions(@Valid @RequestBody ImContent imContent, @RequestParam(required = false) boolean contextEnable) {
SecurityUtils.openInterfaceTemporaryLoginSession(imContent.getSendUserId()); SecurityUtils.openInterfaceTemporaryLoginSession(imContent.getSendUserId());
iImContentService.rateLimitRequestHandler(); iImContentService.rateLimitRequestHandler();
return R.ok(iImContentService.chatCompletions(imContent, contextEnable)); return R.ok(iImContentService.chatCompletions(imContent, contextEnable));

3
kicc-platform/kicc-platform-biz/kicc-system-biz/src/main/java/com/cloud/kicc/system/service/impl/ImContentServiceImpl.java

@ -1,6 +1,7 @@
package com.cloud.kicc.system.service.impl; package com.cloud.kicc.system.service.impl;
import cn.hutool.core.collection.CollUtil; import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.core.collection.ListUtil; import cn.hutool.core.collection.ListUtil;
import cn.hutool.core.io.FileUtil; import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.BooleanUtil; import cn.hutool.core.util.BooleanUtil;
@ -81,7 +82,7 @@ public class ImContentServiceImpl extends ServiceImpl<ImContentMapper, ImContent
List<ImContent> imContents = super.list(Wrappers.<ImContent>lambdaQuery().eq(ImContent::getSessionId, imContent.getSessionId()) List<ImContent> imContents = super.list(Wrappers.<ImContent>lambdaQuery().eq(ImContent::getSessionId, imContent.getSessionId())
.last(String.format("LIMIT %s", openAiConfigProperties.getContextMessageMaxCount())) .last(String.format("LIMIT %s", openAiConfigProperties.getContextMessageMaxCount()))
.orderByDesc(ImContent::getSendTime)); .orderByDesc(ImContent::getSendTime));
List<ChatMessage> completionMessages = ListUtil.empty(); List<ChatMessage> completionMessages = CollectionUtil.newArrayList();
if (BooleanUtil.isTrue(contextEnable)) { if (BooleanUtil.isTrue(contextEnable)) {
// 历史上下文消息 // 历史上下文消息

1
kicc-platform/kicc-platform-biz/kicc-system-biz/src/main/java/com/cloud/kicc/system/util/AiUtil.java

@ -66,7 +66,6 @@ public class AiUtil {
.n(1) .n(1)
.frequencyPenalty(1.0) .frequencyPenalty(1.0)
.presencePenalty(1.0) .presencePenalty(1.0)
.maxTokens(openAiConfigProperties.getCompletionMaxTokens())
.topP(0.8) .topP(0.8)
.user(SecurityUtils.getCasUser().getId()) .user(SecurityUtils.getCasUser().getId())
.build(); .build();

Loading…
Cancel
Save