#函数调用基础

// 使用向量搜索查找FUNCTION类型的函数
EmbeddingSearchResult<TextSegment> searchResult = EmbeddingProvider.search(
    chatMessageDTO.getContent(),
    metadataKey(EmbedBizTypeEnums.Fields.type)
        .isEqualTo(EmbedBizTypeEnums.FUNCTION.getType())
);

// 如果未找到匹配函数,返回提示信息
if (Objects.isNull(searchResult) || searchResult.matches().isEmpty()) {
    return Flux.just(new AiMessageResultDTO("未找到相关功能,请点击下方+按钮选择目标功能"));
}

// 获取最匹配的函数名并更新到上下文
String functionName = searchResult.matches().get(0).embedded().metadata().getString(TEMP_ID);
chatMessageDTO.getExtDetails().setFuncName(functionName);
ChatMessageContextHolder.set(chatMessageDTO);

// 获取对应模型的AI函数助手
AiFunctionAssistantService aiFunctionAssistant = modelProvider
    .getAiFunctionAssistant(chatMessageDTO.getModelName())
    .getValue();

// 流式处理用户请求并返回结果
return aiFunctionAssistant.chatFlux(chatMessageDTO.getContent())
    .map(AiMessageResultDTO::new);

@Data
@JsonInclude(JsonInclude.Include.NON_NULL)
@JsonClassDescription("学生信息")
public class StudentRequest extends BaseAiRequest {
    @FieldPrompt("学生姓名")
    private String studentName;
}

@Component
@RequiredArgsConstructor
public class MathScoreFunctionCalling implements FunctionCalling<StudentRequest> {

    /**
     * 功能描述,多个函数时AI根据描述选择调用哪个函数
     */
    @Override
    public String functionDesc() {
        return "学生成绩查询助手: 能根据您的描述帮您查询学生的数学成绩";
    }

    /**
     * 参数校验,大模型返回的数据不一定稳定,必须进行业务校验
     */
    @Override
    public R checkParams(StudentRequest studentRequest, PigxUser userDetails,
                         ChatMessageDTO.ExtDetails extDetails) {
        if (StrUtil.isBlank(studentRequest.getStudentName())) {
            return R.failed("学生姓名不能为空");
        }
        return R.ok();
    }

    /**
     * 业务处理逻辑
     */
    @Override
    public R<String> handle(StudentRequest studentRequest, PigxUser userDetails,
                            ChatMessageDTO.ExtDetails extDetails) {
        // 模拟根据学生姓名查询数学成绩
        if ("小明".equals(studentRequest.getStudentName())) {
            return R.ok("98");
        }
        return R.failed("查询失败,学生信息不存在");
    }
}