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

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

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;
}

View File

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

View File

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