pdf-qa-server/src/main/java/com/supervision/pdfqaserver/config/OllamaChatModelAspect.java

package com.supervision.pdfqaserver.config;

import cn.hutool.core.util.StrUtil;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

@Aspect
@Component
public class OllamaChatModelAspect {

    @Value("${spring.ai.ollama.chat.model}")
    private String model;

    private String callStringMessage = "String org.springframework.ai.chat.model.ChatModel.call(String)";


    /**
     * 修改ollama的call方法，添加/no_think参数，返回结果去掉think标签
     * @param joinPoint joinPoint
     * @return Object
     * @throws Throwable
     */
    @Around("execution(* org.springframework.ai.chat.model.ChatModel.call(..))")
    public Object aroundMethodExecution(ProceedingJoinPoint joinPoint) throws Throwable {
        String signature = joinPoint.getSignature().toString();
        if (StrUtil.equals(model,"qwen3:30b-a3b") && StrUtil.equals(signature, callStringMessage)) {
            Object[] args = joinPoint.getArgs();
            if (args.length > 0) {
                String arg = (String) args[0];
                args[0] = arg + "\n /no_think";
            }
        }
        // 执行原方法
        Object result = joinPoint.proceed();

        if (StrUtil.equals(model,"qwen3:30b-a3b") && StrUtil.equals(signature, callStringMessage)) {
            result = ((String) result).replaceAll("(?is)<think\\b[^>]*>(.*?)</think>", "").trim();
        }

        return result;
    }
}