com.agentsflex.spring.boot.llm.openai.OpenAiAutoConfiguration Maven / Gradle / Ivy
package com.agentsflex.spring.boot.llm.openai;
import com.agentsflex.llm.openai.OpenAiLlm;
import com.agentsflex.llm.openai.OpenAiLlmConfig;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
/**
* Agents-Flex 大语言模型自动配置。
*
* @author 王帅
* @since 2024-04-10
*/
@ConditionalOnClass(OpenAiLlm.class)
@Configuration(proxyBeanMethods = false)
@EnableConfigurationProperties(OpenAiProperties.class)
public class OpenAiAutoConfiguration {
@Bean
@ConditionalOnMissingBean
public OpenAiLlm openAiLlm(OpenAiProperties properties) {
OpenAiLlmConfig config = new OpenAiLlmConfig();
config.setApiKey(properties.getApiKey());
config.setApiSecret(properties.getApiSecret());
config.setEndpoint(properties.getEndpoint());
config.setModel(properties.getModel());
return new OpenAiLlm(config);
}
}