修改:AI大模型接入-sdk接入
This commit is contained in:
parent
5f993c453a
commit
68f3f0d050
@ -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;
|
||||||
|
}
|
@ -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())
|
@ -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());
|
||||||
|
Loading…
Reference in New Issue
Block a user