@SpringBootTest and Embabel

[bbanfai]

It seems to me that test classes annotated with @SpringBootTest do not seem to execute the code responsible for populating the models available. It also seems to ignore the @TestPropertySource annotation. As my test execution fails due to the missing default model, I have tried to exclude the embabel AutoConfig classes with no success: @EnableAutoConfiguration(exclude= {AgentPlatformAutoConfiguration.class, AgentBedrockAutoConfiguration.class})
Could you please advice.

[alexheifetz]

Hi. Can you share your test code? Tnx.

[bbanfai]

Hi Alex,
Thanks for the follow up. A bit of a background: I am trying to combine Embabel with Spring AI mainly due to the missing feature ( soon to be added) of attaching media to the prompt or PromptRunner to be more specific. Hence my POM contains the following dependencies:

		<dependency>
			<groupId>com.embabel.agent</groupId>
			<artifactId>embabel-agent-starter</artifactId>
		</dependency>
		<dependency>
			<groupId>org.springframework.ai</groupId>
			<artifactId>spring-ai-starter-model-azure-openai</artifactId>
		</dependency>
		<dependency>
			<groupId>org.springframework.ai</groupId>
			<artifactId>spring-ai-starter-model-anthropic</artifactId>
		</dependency>
		<dependency>
			<groupId>org.springframework.ai</groupId>
			<artifactId>spring-ai-starter-mcp-server-webmvc</artifactId>
		</dependency>
		<!-- Dependencies for testing -->
		<dependency>
			<groupId>com.embabel.agent</groupId>
			<artifactId>embabel-agent-test</artifactId>
			<version>0.1.0-SNAPSHOT</version>
			<scope>test</scope>
		</dependency>
	</dependencies>

As for the test, I tried to test a service that does not use Embabel:

@SpringBootTest
public class DocumentValidationServiceTest {
	
	@Autowired
	private DocumentValidationService documentValidationService;
	
	@Autowired
	private ResourceLoader resourceLoader;
	
	@Test
	@Order(1)
	public void loadMediaTest() throws IOException {
		
		List<MediaWithName> mediaList = FileUtils.loadMediaFromFolder(resourceLoader, "docs/KYC");
		assertFalse(mediaList.isEmpty());
		for (MediaWithName mediaWithName : mediaList) {
			System.out.println("Media name: " + mediaWithName.fileName() + " type:" + mediaWithName.media().getMimeType());
		}
	    
	}
...

This fails as Embabel does not seem to pick up the models to pick the default model from during its start.
When I only create context for the Spring UI, I can run my test without Embabel, but I would like to start up the complete app:

@ExtendWith(SpringExtension.class)
@ContextConfiguration(classes = {DocumentValidationService.class, TestConfig.class})
@ImportAutoConfiguration(classes = {AnthropicChatAutoConfiguration.class, ChatMemoryAutoConfiguration.class})
@TestPropertySource("classpath:application_test.properties")

Here is also my config file:

#embabel.models.llms.cheapest=gpt-4.1-nano
embabel.models.defaultLlm=claude-opus-4-20250514

#Use manual control to select the chat client
spring.ai.chat.client.enabled=false
spring.ai.model.chat=anthropic

#Azure OpenAI GPT-35
spring.ai.azure.openai.api-key=${OPENAI_API_KEY}
spring.ai.azure.openai.endpoint=${OPEN_API_ENDPOINT}
spring.ai.azure.openai.chat.options.deployment-name=gpt-35-turbo-16k
spring.ai.azure.openai.chat.options.temperature=0
spring.ai.azure.openai.chat.options.n=1

#Claude
spring.ai.anthropic.api-key=${ANTHROPIC_API_KEY}
spring.ai.anthropic.chat.options.model=claude-opus-4-20250514

logging.level.org.springframework.ai = DEBUG

Thank you in advance.

[alexheifetz]

Hello. Which version of embabel are you using, and can you share log output with Exception you're getting.
If you could share link to your github repo, so I can just run test, that would be even better.

[bbanfai]

Thanks Alex, I managed to run my tests now, also solved my issues with the shell mode by the properties ("spring.shell.interactive.enabled=false",
"spring.shell.noninteractive.enabled=false")

[johnsonr]

Embabel does not pick up Spring AI default models. This is intentional.

I wonder if the WriteAndReviewIntegrationTest in https://github.dev/embabel/java-agent-template might be useful.

[bbanfai]

Hi Rod,
Thanks for your response. I didn't expect Embabel to pick up the Spring Models, it did not pick up its own registered models. I fixed this now so it worked for my 'mixed' purposes (using Spring AI calls within a Embabel workflow as it allows me to submit media) until media can be added to the prompts.

And yes, these properties ("spring.shell.interactive.enabled=false",
"spring.shell.noninteractive.enabled=false") helped to run my @SpringBootTest without the test stopping at the command line.
Thanks again,
Balazs