31 lines
1.2 KiB
Java
31 lines
1.2 KiB
Java
package com.baoshi.conf;
|
|
|
|
import com.alibaba.cloud.ai.autoconfigure.dashscope.DashScopeChatProperties;
|
|
import com.alibaba.cloud.ai.dashscope.chat.DashScopeChatModel;
|
|
import com.alibaba.cloud.ai.dashscope.chat.DashScopeChatOptions;
|
|
import org.springframework.ai.chat.client.ChatClient;
|
|
import org.springframework.ai.model.ollama.autoconfigure.OllamaChatProperties;
|
|
import org.springframework.ai.ollama.OllamaChatModel;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.context.annotation.Configuration;
|
|
|
|
@Configuration
|
|
public class AiConfig {
|
|
|
|
@Bean
|
|
public ChatClient dashScope(@Autowired DashScopeChatModel dashScopeChatModel,
|
|
@Autowired DashScopeChatProperties properties) {
|
|
return ChatClient.builder(dashScopeChatModel)
|
|
//.defaultOptions(DashScopeChatOptions.builder().build())
|
|
.build();
|
|
}
|
|
|
|
@Bean
|
|
public ChatClient ollama(@Autowired OllamaChatModel ollamaChatModel,
|
|
@Autowired OllamaChatProperties properties) {
|
|
return ChatClient.builder(ollamaChatModel).defaultOptions(properties.getOptions()).build();
|
|
}
|
|
|
|
}
|