Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;

import org.springframework.ai.chat.memory.repository.jdbc.JdbcChatMemoryRepository;
import org.springframework.ai.chat.messages.AssistantMessage;
Expand All @@ -35,21 +34,17 @@
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.context.ApplicationContext;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.test.context.junit.jupiter.SpringExtension;

import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.fail;

@ExtendWith(SpringExtension.class)
@SpringBootTest(classes = JdbcChatMemoryRepositoryHsqldbAutoConfigurationIT.TestConfig.class,
properties = { "spring.datasource.url=jdbc:hsqldb:mem:chat_memory_auto_configuration_test;DB_CLOSE_DELAY=-1",
"spring.datasource.username=sa", "spring.datasource.password=",
"spring.datasource.driver-class-name=org.hsqldb.jdbcDriver",
"spring.ai.chat.memory.repository.jdbc.initialize-schema=always", "spring.sql.init.mode=always",
"spring.jpa.hibernate.ddl-auto=none", "spring.jpa.defer-datasource-initialization=true",
"spring.sql.init.continue-on-error=true", "spring.sql.init.schema-locations=classpath:schema.sql",
"logging.level.org.springframework.jdbc=DEBUG",
"logging.level.org.springframework.boot.sql.init=DEBUG" })
@SpringBootTest({ "spring.datasource.url=jdbc:hsqldb:mem:chat_memory_auto_configuration_test;DB_CLOSE_DELAY=-1",
"spring.datasource.username=sa", "spring.datasource.password=",
"spring.datasource.driver-class-name=org.hsqldb.jdbcDriver",
"spring.ai.chat.memory.repository.jdbc.initialize-schema=always", "spring.sql.init.mode=always",
"spring.jpa.hibernate.ddl-auto=none", "spring.jpa.defer-datasource-initialization=true",
"spring.sql.init.continue-on-error=true", "spring.sql.init.schema-locations=classpath:schema.sql",
"logging.level.org.springframework.jdbc=DEBUG", "logging.level.org.springframework.boot.sql.init=DEBUG" })
@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.ANY)
@ImportAutoConfiguration({ org.springframework.ai.model.chat.memory.autoconfigure.ChatMemoryAutoConfiguration.class,
JdbcChatMemoryRepositoryAutoConfiguration.class, JdbcTemplateAutoConfiguration.class,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@
* @author Utkarsh Srivastava
* @author Christian Tzolov
*/
@SpringBootTest(properties = { "spring.ai.postgresml.embedding.options.metadata-mode=all",
@SpringBootTest({ "spring.ai.postgresml.embedding.options.metadata-mode=all",
"spring.ai.postgresml.embedding.options.kwargs.key1=value1",
"spring.ai.postgresml.embedding.options.kwargs.key2=value2",
"spring.ai.postgresml.embedding.options.transformer=abc123" })
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
* @author Yanming Zhou
* @author Eddú Meléndez
*/
@SpringBootTest(properties = "spring.datasource.url=jdbc:tc:sqlserver:2022-latest:///")
@SpringBootTest("spring.datasource.url=jdbc:tc:sqlserver:2022-latest:///")
@Sql(scripts = "classpath:org/springframework/ai/chat/memory/repository/jdbc/schema-sqlserver.sql",
executionPhase = Sql.ExecutionPhase.BEFORE_TEST_CLASS)
class JdbcChatMemoryRepositorySqlServerIT extends AbstractJdbcChatMemoryRepositoryIT {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@
*
* @author Łukasz Jernaś
*/
@SpringBootTest(classes = MongoChatMemoryRepositoryIT.TestConfiguration.class)
@SpringBootTest
public class MongoChatMemoryRepositoryIT {

@Autowired
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@
/**
* @author Christian Tzolov
*/
@SpringBootTest(classes = AnthropicChatModelAdditionalHttpHeadersIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "ANTHROPIC_API_KEY", matches = ".+")
public class AnthropicChatModelAdditionalHttpHeadersIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@

import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest(classes = AnthropicChatModelIT.Config.class, properties = "spring.ai.retry.on-http-codes=429")
@SpringBootTest("spring.ai.retry.on-http-codes=429")
@EnabledIfEnvironmentVariable(named = "ANTHROPIC_API_KEY", matches = ".+")
class AnthropicChatModelIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,7 @@
* @author Thomas Vitale
* @author Alexandros Pappas
*/
@SpringBootTest(classes = AnthropicChatModelObservationIT.Config.class,
properties = "spring.ai.retry.on-http-codes=429")
@SpringBootTest("spring.ai.retry.on-http-codes=429")
@EnabledIfEnvironmentVariable(named = "ANTHROPIC_API_KEY", matches = ".+")
public class AnthropicChatModelObservationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@
* @author Soby Chacko
* @since 1.1.0
*/
@SpringBootTest(classes = AnthropicCitationIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "ANTHROPIC_API_KEY", matches = ".+")
class AnthropicCitationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@
*
* @author Piotr Olaszewski
*/
@SpringBootTest(classes = AzureOpenAiAudioTranscriptionModelIT.TestConfiguration.class)
@SpringBootTest
@EnabledIfEnvironmentVariables({
@EnabledIfEnvironmentVariable(named = "AZURE_OPENAI_TRANSCRIPTION_API_KEY", matches = ".+"),
@EnabledIfEnvironmentVariable(named = "AZURE_OPENAI_TRANSCRIPTION_ENDPOINT", matches = ".+") })
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@
/**
* @author Soby Chacko
*/
@SpringBootTest(classes = AzureOpenAiChatClientIT.TestConfiguration.class)
@SpringBootTest
@RequiresAzureCredentials
public class AzureOpenAiChatClientIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@

import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest(classes = AzureOpenAiChatModelIT.TestConfiguration.class)
@SpringBootTest
@RequiresAzureCredentials
class AzureOpenAiChatModelIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@
/**
* @author Soby Chacko
*/
@SpringBootTest(classes = AzureOpenAiChatModelObservationIT.TestConfiguration.class)
@SpringBootTest
@RequiresAzureCredentials
class AzureOpenAiChatModelObservationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@
*
* @author Christian Tzolov
*/
@SpringBootTest(classes = AzureOpenAiEmbeddingModelObservationIT.Config.class)
@SpringBootTest
@RequiresAzureCredentials
public class AzureOpenAiEmbeddingModelObservationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@

import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest(classes = AzureOpenAiChatModelFunctionCallIT.TestConfiguration.class)
@SpringBootTest
@RequiresAzureCredentials
class AzureOpenAiChatModelFunctionCallIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@
/**
* NOTE: use deployment ID dall-e-3
*/
@SpringBootTest(classes = AzureOpenAiImageModelIT.TestConfiguration.class)
@SpringBootTest
@EnabledIfEnvironmentVariables({ @EnabledIfEnvironmentVariable(named = "AZURE_OPENAI_IMAGE_API_KEY", matches = ".+"),
@EnabledIfEnvironmentVariable(named = "AZURE_OPENAI_IMAGE_ENDPOINT", matches = ".+") })
public class AzureOpenAiImageModelIT {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -48,8 +48,7 @@
*
* @author Christian Tzolov
*/
@SpringBootTest(classes = BedrockProxyChatModelObservationIT.Config.class,
properties = "spring.ai.retry.on-http-codes=429")
@SpringBootTest("spring.ai.retry.on-http-codes=429")
@RequiresAwsCredentials
public class BedrockProxyChatModelObservationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@
* @author Christian Tzolov
*/
// @Disabled
@SpringBootTest(classes = BedrockNovaChatClientIT.Config.class)
@SpringBootTest
@RequiresAwsCredentials
public class BedrockNovaChatClientIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@
*
* @author Geng Rong
*/
@SpringBootTest(classes = DeepSeekChatModelObservationIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "DEEPSEEK_API_KEY", matches = ".+")
public class DeepSeekChatModelObservationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@
* @author Christian Tzolov
* @author Dan Dobrin
*/
@SpringBootTest(classes = GoogleGenAiTextEmbeddingModelIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "GOOGLE_CLOUD_PROJECT", matches = ".*")
@EnabledIfEnvironmentVariable(named = "GOOGLE_CLOUD_LOCATION", matches = ".*")
class GoogleGenAiTextEmbeddingModelIT {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@
* @author Christian Tzolov
* @author Dan Dobrin
*/
@SpringBootTest(classes = GoogleGenAiTextEmbeddingModelObservationIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "GOOGLE_CLOUD_PROJECT", matches = ".*")
@EnabledIfEnvironmentVariable(named = "GOOGLE_CLOUD_LOCATION", matches = ".*")
public class GoogleGenAiTextEmbeddingModelObservationIT {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@
*
* @author Geng Rong
*/
@SpringBootTest(classes = MiniMaxChatModelObservationIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "MINIMAX_API_KEY", matches = ".+")
public class MiniMaxChatModelObservationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@
*
* @author Geng Rong
*/
@SpringBootTest(classes = MiniMaxEmbeddingModelObservationIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "MINIMAX_API_KEY", matches = ".+")
public class MiniMaxEmbeddingModelObservationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@
* @author Alexandros Pappas
* @author Jason Smith
*/
@SpringBootTest(classes = MistralAiChatModelObservationIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "MISTRAL_AI_API_KEY", matches = ".+")
public class MistralAiChatModelObservationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@
* @author Thomas Vitale
* @author Jason Smith
*/
@SpringBootTest(classes = MistralAiEmbeddingModelObservationIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "MISTRAL_AI_API_KEY", matches = ".+")
public class MistralAiEmbeddingModelObservationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@

import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest(classes = OllamaChatModelFunctionCallingIT.Config.class)
@SpringBootTest
class OllamaChatModelFunctionCallingIT extends BaseOllamaIT {

private static final Logger logger = LoggerFactory.getLogger(OllamaChatModelFunctionCallingIT.class);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@
*
* @author Sun Yuhan
*/
@SpringBootTest(classes = OllamaChatModelMetadataTests.Config.class)
@SpringBootTest
class OllamaChatModelMetadataTests extends BaseOllamaIT {

private static final String MODEL = OllamaModel.QWEN_3_06B.getName();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -50,7 +50,7 @@
* @author Thomas Vitale
* @author Alexandros Pappas
*/
@SpringBootTest(classes = OllamaChatModelObservationIT.Config.class)
@SpringBootTest
public class OllamaChatModelObservationIT extends BaseOllamaIT {

private static final String MODEL = OllamaModel.QWEN_2_5_3B.getName();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
*
* @author Thomas Vitale
*/
@SpringBootTest(classes = OllamaEmbeddingModelObservationIT.Config.class)
@SpringBootTest
public class OllamaEmbeddingModelObservationIT extends BaseOllamaIT {

private static final String MODEL = OllamaModel.NOMIC_EMBED_TEXT.getName();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@
* @author Ilayaperumal Gopinathan
* @author Jonghoon Park
*/
@SpringBootTest(classes = OpenAiAudioModelNoOpApiKeysIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "OPENAI_API_KEY", matches = ".+")
public class OpenAiAudioModelNoOpApiKeysIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@
/**
* @author Christian Tzolov
*/
@SpringBootTest(classes = OpenAiChatModelAdditionalHttpHeadersIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "OPENAI_API_KEY", matches = ".+")
public class OpenAiChatModelAdditionalHttpHeadersIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@

import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest(classes = OpenAiChatModelFunctionCallingIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "OPENAI_API_KEY", matches = ".+")
class OpenAiChatModelFunctionCallingIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@
/**
* @author Ilayaperumal Gopinathan
*/
@SpringBootTest(classes = OpenAiChatModelNoOpApiKeysIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "OPENAI_API_KEY", matches = ".+")
public class OpenAiChatModelNoOpApiKeysIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@
*
* @author Thomas Vitale
*/
@SpringBootTest(classes = OpenAiChatModelObservationIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "OPENAI_API_KEY", matches = ".+")
public class OpenAiChatModelObservationIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@
* @author Christian Tzolov
* @author Ilayaperumal Gopinathan
*/
@SpringBootTest(classes = OpenAiChatModelResponseFormatIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "OPENAI_API_KEY", matches = ".+")
public class OpenAiChatModelResponseFormatIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@
* For more information on DeepSeek behavior, refer to its API documentation:
* <a href="https://api-docs.deepseek.com/">DeepSeek API</a>
*/
@SpringBootTest(classes = DeepSeekWithOpenAiChatModelIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "DEEPSEEK_API_KEY", matches = ".+")
@Disabled("Requires DeepSeek credits")
class DeepSeekWithOpenAiChatModelIT {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@
* @since 1.0.0
*/
@Testcontainers
@SpringBootTest(classes = DockerModelRunnerWithOpenAiChatModelIT.Config.class)
@SpringBootTest
@Disabled("Requires Docker Model Runner enabled. See https://docs.docker.com/desktop/features/model-runner/")
class DockerModelRunnerWithOpenAiChatModelIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@

import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest(classes = GroqWithOpenAiChatModelIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "GROQ_API_KEY", matches = ".+")
// @Disabled("Due to rate limiting it is hard to run it in one go")
class GroqWithOpenAiChatModelIT {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@

import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest(classes = MistralWithOpenAiChatModelIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "MISTRAL_AI_API_KEY", matches = ".+")
class MistralWithOpenAiChatModelIT {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@

import static org.assertj.core.api.Assertions.assertThat;

@SpringBootTest(classes = MultiOpenAiClientIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "GROQ_API_KEY", matches = ".+")
@EnabledIfEnvironmentVariable(named = "OPENAI_API_KEY", matches = ".+")
@ActiveProfiles("logging-test")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@
* @author Christian Tzolov
* @since 1.0.0
*/
@SpringBootTest(classes = NvidiaWithOpenAiChatModelIT.Config.class)
@SpringBootTest
@EnabledIfEnvironmentVariable(named = "NVIDIA_API_KEY", matches = ".+")
@Disabled("Requires NVIDIA credits")
class NvidiaWithOpenAiChatModelIT {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@

@Disabled("For manual smoke testing only.")
@Testcontainers
@SpringBootTest(classes = OllamaWithOpenAiChatModelIT.Config.class)
@SpringBootTest
class OllamaWithOpenAiChatModelIT {

private static final Logger logger = LoggerFactory.getLogger(OllamaWithOpenAiChatModelIT.class);
Expand Down
Loading