Added logRequests/logResponses properties to Ollama SB starter langch…

…ain4j/langchain4j#662
langchain4j · Jun 11, 2024 · a3d5a36 · a3d5a36
1 parent 20a79ef
commit a3d5a36
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/...4j-ollama-spring-boot-starter/src/main/java/dev/langchain4j/ollama/spring/AutoConfig.java b/...4j-ollama-spring-boot-starter/src/main/java/dev/langchain4j/ollama/spring/AutoConfig.java
@@ -30,6 +30,8 @@ OllamaChatModel ollamaChatModel(Properties properties) {
                 .timeout(chatModelProperties.getTimeout())
                 .maxRetries(chatModelProperties.getMaxRetries())
                 .customHeaders(chatModelProperties.getCustomHeaders())
+                .logRequests(chatModelProperties.getLogRequests())
+                .logResponses(chatModelProperties.getLogResponses())
                 .build();
     }
 
@@ -50,6 +52,8 @@ OllamaStreamingChatModel ollamaStreamingChatModel(Properties properties) {
                 .format(chatModelProperties.getFormat())
                 .timeout(chatModelProperties.getTimeout())
                 .customHeaders(chatModelProperties.getCustomHeaders())
+                .logRequests(chatModelProperties.getLogRequests())
+                .logResponses(chatModelProperties.getLogResponses())
                 .build();
     }
 
@@ -71,6 +75,8 @@ OllamaLanguageModel ollamaLanguageModel(Properties properties) {
                 .timeout(languageModelProperties.getTimeout())
                 .maxRetries(languageModelProperties.getMaxRetries())
                 .customHeaders(languageModelProperties.getCustomHeaders())
+                .logRequests(languageModelProperties.getLogRequests())
+                .logResponses(languageModelProperties.getLogResponses())
                 .build();
     }
 
@@ -91,6 +97,8 @@ OllamaStreamingLanguageModel ollamaStreamingLanguageModel(Properties properties)
                 .format(languageModelProperties.getFormat())
                 .timeout(languageModelProperties.getTimeout())
                 .customHeaders(languageModelProperties.getCustomHeaders())
+                .logRequests(languageModelProperties.getLogRequests())
+                .logResponses(languageModelProperties.getLogResponses())
                 .build();
     }
 
@@ -104,6 +112,8 @@ OllamaEmbeddingModel ollamaEmbeddingModel(Properties properties) {
                 .timeout(embeddingModelProperties.getTimeout())
                 .maxRetries(embeddingModelProperties.getMaxRetries())
                 .customHeaders(embeddingModelProperties.getCustomHeaders())
+                .logRequests(embeddingModelProperties.getLogRequests())
+                .logResponses(embeddingModelProperties.getLogResponses())
                 .build();
     }
 }
diff --git a/...-spring-boot-starter/src/main/java/dev/langchain4j/ollama/spring/ChatModelProperties.java b/...-spring-boot-starter/src/main/java/dev/langchain4j/ollama/spring/ChatModelProperties.java
@@ -24,4 +24,6 @@ class ChatModelProperties {
     Duration timeout;
     Integer maxRetries;
     Map<String, String> customHeaders;
+    Boolean logRequests;
+    Boolean logResponses;
 }
diff --git a/...ng-boot-starter/src/main/java/dev/langchain4j/ollama/spring/EmbeddingModelProperties.java b/...ng-boot-starter/src/main/java/dev/langchain4j/ollama/spring/EmbeddingModelProperties.java
@@ -15,4 +15,6 @@ class EmbeddingModelProperties {
     Duration timeout;
     Integer maxRetries;
     Map<String, String> customHeaders;
+    Boolean logRequests;
+    Boolean logResponses;
 }
diff --git a/...ing-boot-starter/src/main/java/dev/langchain4j/ollama/spring/LanguageModelProperties.java b/...ing-boot-starter/src/main/java/dev/langchain4j/ollama/spring/LanguageModelProperties.java
@@ -24,4 +24,6 @@ class LanguageModelProperties {
     Duration timeout;
     Integer maxRetries;
     Map<String, String> customHeaders;
+    Boolean logRequests;
+    Boolean logResponses;
 }