From ecd46fd6147717bf6d1f02805e86d29c6be0fa2f Mon Sep 17 00:00:00 2001
From: suhas-koheda <sharmasuhas450@gmail.com>
Date: Wed, 8 Jan 2025 22:13:37 +0530
Subject: [PATCH] -

---
 .../java/dev/langchain4j/googleaigemini/spring/AutoConfig.java  | 2 +-
 .../langchain4j/googleaigemini/spring/ChatModelProperties.java  | 2 --
 .../googleaigemini/spring/EmbeddingModelProperties.java         | 2 +-
 .../googleaigemini/spring/GeminiFunctionCallingConfig.java      | 2 +-
 4 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/AutoConfig.java b/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/AutoConfig.java
index 050fa116..72400ca6 100644
--- a/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/AutoConfig.java
+++ b/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/AutoConfig.java
@@ -13,7 +13,7 @@
 
 @AutoConfiguration
 @EnableConfigurationProperties(Properties.class)
-public class AutoConfig {
+class AutoConfig {
 
     @Bean
     @ConditionalOnProperty({
diff --git a/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/ChatModelProperties.java b/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/ChatModelProperties.java
index 89f7def0..add914bc 100644
--- a/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/ChatModelProperties.java
+++ b/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/ChatModelProperties.java
@@ -1,8 +1,6 @@
 package dev.langchain4j.googleaigemini.spring;
 
 import dev.langchain4j.model.chat.request.ResponseFormat;
-import dev.langchain4j.model.googleai.GeminiHarmBlockThreshold;
-import dev.langchain4j.model.googleai.GeminiHarmCategory;
 
 import java.time.Duration;
 import java.util.Map;
diff --git a/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/EmbeddingModelProperties.java b/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/EmbeddingModelProperties.java
index b6d9ff29..d5e3b90c 100644
--- a/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/EmbeddingModelProperties.java
+++ b/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/EmbeddingModelProperties.java
@@ -4,7 +4,7 @@
 
 import java.time.Duration;
 
-public record EmbeddingModelProperties(
+record EmbeddingModelProperties(
         String apiKey,
         String modelName,
         String titleMetadataKey,
diff --git a/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/GeminiFunctionCallingConfig.java b/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/GeminiFunctionCallingConfig.java
index a5bc5090..32f6c3b4 100644
--- a/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/GeminiFunctionCallingConfig.java
+++ b/langchain4j-google-ai-gemini-spring-boot-starter/src/main/java/dev/langchain4j/googleaigemini/spring/GeminiFunctionCallingConfig.java
@@ -4,7 +4,7 @@
 
 import java.util.List;
 
-public record GeminiFunctionCallingConfig(
+record GeminiFunctionCallingConfig(
         GeminiMode geminiMode,
         List<String> allowedFunctionNames
 ) {