Add missing configuration and implementation

Author: pacphi

.gitignore

@@ -43,7 +43,7 @@ out/
 ### Miscellaneous ###
 data/
 tmp/
-config/
+**/creds*.yml
 logs/
 
 

src/main/java/me/pacphi/config/Chat.java

@@ -0,0 +1,49 @@
+package me.pacphi.config;
+
+import org.springframework.ai.autoconfigure.openai.OpenAiChatProperties;
+import org.springframework.ai.autoconfigure.openai.OpenAiConnectionProperties;
+import org.springframework.ai.model.function.FunctionCallbackResolver;
+import org.springframework.ai.openai.OpenAiChatModel;
+import org.springframework.ai.openai.api.OpenAiApi;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Profile;
+import org.springframework.http.HttpHeaders;
+import org.springframework.retry.support.RetryTemplate;
+import org.springframework.web.client.ResponseErrorHandler;
+import org.springframework.web.client.RestClient;
+import org.springframework.web.reactive.function.client.WebClient;
+
+@Configuration
+// @see https://github.com/spring-projects/spring-ai/issues/372#issuecomment-2242650500
+public class Chat {
+
+    @Bean
+    public OpenAiChatModel chatModel(
+            OpenAiConnectionProperties connectionProperties,
+            OpenAiChatProperties chatProperties,
+            WebClient.Builder webClientBuilder,
+            RetryTemplate retryTemplate,
+            FunctionCallbackResolver functionCallbackResolver,
+            ResponseErrorHandler responseErrorHandler,
+            RestClient.Builder restClientBuilder
+    ) {
+        restClientBuilder.defaultHeaders(headers -> headers.set(HttpHeaders.ACCEPT_ENCODING, "gzip, deflate"));
+
+        OpenAiApi openAiApi = new OpenAiApi(
+                chatProperties.getBaseUrl() != null ? chatProperties.getBaseUrl() : connectionProperties.getBaseUrl(),
+                chatProperties.getApiKey() != null ? chatProperties.getApiKey() : connectionProperties.getApiKey(),
+                restClientBuilder,
+                webClientBuilder,
+                responseErrorHandler
+        );
+
+        return new OpenAiChatModel(
+                openAiApi,
+                chatProperties.getOptions(),
+                functionCallbackResolver,
+                retryTemplate
+        );
+    }
+
+}

src/main/java/me/pacphi/config/MultiChat.java

@@ -0,0 +1,69 @@
+package me.pacphi.config;
+
+import org.springframework.ai.autoconfigure.openai.OpenAiChatProperties;
+import org.springframework.ai.autoconfigure.openai.OpenAiConnectionProperties;
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;
+import org.springframework.ai.model.function.FunctionCallbackResolver;
+import org.springframework.ai.openai.OpenAiChatModel;
+import org.springframework.ai.openai.OpenAiChatOptions;
+import org.springframework.ai.openai.api.OpenAiApi;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Profile;
+import org.springframework.http.HttpHeaders;
+import org.springframework.retry.support.RetryTemplate;
+import org.springframework.web.client.ResponseErrorHandler;
+import org.springframework.web.client.RestClient;
+import org.springframework.web.reactive.function.client.WebClient;
+
+import java.util.Map;
+import java.util.stream.Collectors;
+
+@Configuration
+public class MultiChat {
+
+    @Bean
+    public Map<String, ChatClient> chatClients(
+            OpenAiConnectionProperties connectionProperties,
+            OpenAiChatProperties chatProperties,
+            WebClient.Builder webClientBuilder,
+            RetryTemplate retryTemplate,
+            FunctionCallbackResolver functionCallbackResolver,
+            ResponseErrorHandler responseErrorHandler,
+            MultiChatProperties multiChatProperties
+    ) {
+        RestClient.Builder restClientBuilder = RestClient.builder()
+                .defaultHeaders(headers -> headers.set(HttpHeaders.ACCEPT_ENCODING, "gzip, deflate"));
+
+        OpenAiApi openAiApi = new OpenAiApi(
+                chatProperties.getBaseUrl() != null ? chatProperties.getBaseUrl() : connectionProperties.getBaseUrl(),
+                chatProperties.getApiKey() != null ? chatProperties.getApiKey() : connectionProperties.getApiKey(),
+                restClientBuilder,
+                webClientBuilder,
+                responseErrorHandler
+        );
+
+        return multiChatProperties.getOptions().getModels().stream().collect(
+                Collectors.toMap(
+                    model -> model,
+                    model -> {
+                        OpenAiChatOptions chatOptions = OpenAiChatOptions.fromOptions(chatProperties.getOptions());
+                        chatOptions.setModel(model);
+                        OpenAiChatModel openAiChatModel = new OpenAiChatModel(
+                                openAiApi,
+                                chatOptions,
+                                functionCallbackResolver,
+                                retryTemplate
+                        );
+                        // Create ChatClient with similar configuration to original service
+                        return ChatClient.builder(openAiChatModel)
+                                .defaultAdvisors(
+                                        new SimpleLoggerAdvisor())
+                                .build();
+                    }
+                )
+        );
+    }
+
+}

src/main/java/me/pacphi/config/MultiChatProperties.java

@@ -0,0 +1,6 @@
+package me.pacphi.config;
+
+public interface MultiChatProperties {
+    Options getOptions();
+    void setOptions(Options options);
+}

src/main/java/me/pacphi/config/OpenRouterAiChatProperties.java

@@ -0,0 +1,20 @@
+package me.pacphi.config;
+
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.stereotype.Component;
+
+@Component
+@ConfigurationProperties(prefix = "spring.ai.openrouter.chat")
+public class OpenRouterAiChatProperties implements MultiChatProperties {
+
+    private Options options = new Options();
+
+    public Options getOptions() {
+        return options;
+    }
+
+    public void setOptions(Options options) {
+        this.options = options;
+    }
+
+}

src/main/java/me/pacphi/config/Options.java

@@ -0,0 +1,16 @@
+package me.pacphi.config;
+
+import java.util.Set;
+
+public class Options {
+
+    private Set<String> models;
+
+    public Set<String> getModels() {
+        return models;
+    }
+
+    public void setModels(Set<String> models) {
+        this.models = models;
+    }
+}