29 lines
973 B
Java
29 lines
973 B
Java
package com.ai.config;
|
|
|
|
import com.ai.function.MyCalculator;
|
|
import com.ai.service.OllamaAssist;
|
|
import dev.langchain4j.memory.chat.MessageWindowChatMemory;
|
|
import dev.langchain4j.model.chat.ChatLanguageModel;
|
|
import dev.langchain4j.model.ollama.OllamaChatModel;
|
|
import dev.langchain4j.service.AiServices;
|
|
import org.springframework.context.annotation.Bean;
|
|
import org.springframework.context.annotation.Configuration;
|
|
|
|
@Configuration
|
|
public class OllamaConfig {
|
|
|
|
@Bean
|
|
public OllamaAssist ollamaAssist() {
|
|
ChatLanguageModel ollamaModel = OllamaChatModel.builder()
|
|
.baseUrl("http://localhost:11434")
|
|
.modelName("llama3.2:3b")
|
|
.build();
|
|
return AiServices.builder(OllamaAssist.class)
|
|
.chatMemoryProvider(memoryId -> MessageWindowChatMemory.withMaxMessages(10))
|
|
.chatLanguageModel(ollamaModel)
|
|
.tools(new MyCalculator())
|
|
.build();
|
|
}
|
|
|
|
}
|