huangge1199/long-ai-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改：AI大模型接入-sdk接入

Browse Source
This commit is contained in:
huangge1199 2025-05-14 13:31:44 +08:00
parent 5f993c453a
commit 68f3f0d050
3 changed files with 26 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/main/java/com/huangge1199/aiagent/Service/InvokeService.java Normal file
View File

@ -0,0 +1,15 @@
package com.huangge1199.aiagent.Service;
import com.alibaba.dashscope.aigc.generation.GenerationResult;
import com.alibaba.dashscope.exception.InputRequiredException;
import com.alibaba.dashscope.exception.NoApiKeyException;
/**
* InvokeService
*
* @author huangge1199
* @since 2025/5/14 13:27:08
*/
public interface InvokeService {
GenerationResult callWithMessage() throws NoApiKeyException, InputRequiredException;
}

16
src/main/java/com/huangge1199/aiagent/demo/invoke/SdkAiInvoke.java → src/main/java/com/huangge1199/aiagent/Service/impl/InvokeServiceImpl.java
View File

@ -1,18 +1,17 @@
package com.huangge1199.aiagent.demo.invoke;
package com.huangge1199.aiagent.Service.impl;
import java.util.Arrays;
import java.lang.System;
import com.alibaba.dashscope.aigc.generation.Generation;
import com.alibaba.dashscope.aigc.generation.GenerationParam;
import com.alibaba.dashscope.aigc.generation.GenerationResult;
import com.alibaba.dashscope.common.Message;
import com.alibaba.dashscope.common.Role;
import com.alibaba.dashscope.exception.ApiException;
import com.alibaba.dashscope.exception.InputRequiredException;
import com.alibaba.dashscope.exception.NoApiKeyException;
import com.alibaba.dashscope.utils.JsonUtils;
import com.huangge1199.aiagent.Service.InvokeService;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.stereotype.Service;
/**
* SdkAiInvoke
@ -20,13 +19,14 @@ import org.springframework.stereotype.Component;
* @author huangge1199
* @since 2025/5/14 12:27:47
*/
@Component
public class SdkAiInvoke {
@Service
public class InvokeServiceImpl implements InvokeService {
@Value("${bailian.API-KEY}")
private String baiLianKey;
public GenerationResult callWithMessage() throws ApiException, NoApiKeyException, InputRequiredException {
@Override
public GenerationResult callWithMessage() throws NoApiKeyException, InputRequiredException {
Generation gen = new Generation();
Message systemMsg = Message.builder()
.role(Role.SYSTEM.getValue())

6
src/main/java/com/huangge1199/aiagent/controller/invoke/InvokeController.java
View File

@ -5,8 +5,8 @@ import com.alibaba.dashscope.exception.ApiException;
import com.alibaba.dashscope.exception.InputRequiredException;
import com.alibaba.dashscope.exception.NoApiKeyException;
import com.alibaba.dashscope.utils.JsonUtils;
import com.huangge1199.aiagent.Service.InvokeService;
import com.huangge1199.aiagent.common.R;
import com.huangge1199.aiagent.demo.invoke.SdkAiInvoke;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
@ -26,13 +26,13 @@ import org.springframework.web.bind.annotation.RestController;
public class InvokeController {
@Resource
private SdkAiInvoke sdkAiInvoke;
private InvokeService invokeService;
@PostMapping("/sdk")
@Operation(summary = "sdk接入")
public R<String> sdkAiInvoke() {
try {
GenerationResult result = sdkAiInvoke.callWithMessage();
GenerationResult result = invokeService.callWithMessage();
return R.ok(JsonUtils.toJson(result));
} catch (ApiException | NoApiKeyException | InputRequiredException e) {
return R.fail(e.getMessage());