Spring AI项目中如何正确禁用Gemini和Vertex AI服务

在Spring AI项目的实际开发过程中，测试环境的配置往往需要与生产环境有所区别。特别是在使用云服务如Google Vertex AI时，测试环境可能无法访问真实的云服务资源。本文将详细介绍如何在Spring AI项目中正确禁用Gemini和Vertex AI相关功能。

问题背景

当开发者在Spring AI项目中集成了Vertex AI的Gemini和Embedding服务后，在测试环境中运行时会遇到一个常见问题：即使已经在配置文件中设置了禁用相关服务，系统仍然会尝试连接Vertex AI服务，导致测试失败并抛出"Vertex AI project-id must be set!"的错误提示。

解决方案

经过实践验证，正确的配置方式应该是使用以下属性：

spring.ai.model.chat=none
spring.ai.model.embedding.text=none
spring.ai.model.embedding.multimodal=none

这三个配置项分别控制着：

1. 聊天模型服务的启用/禁用
2. 文本嵌入服务的启用/禁用
3. 多模态嵌入服务的启用/禁用

配置详解

1. 聊天模型禁用

spring.ai.model.chat=none配置项会完全禁用Spring AI中的聊天模型功能。当设置为"none"时，系统不会尝试初始化任何聊天模型服务，包括Vertex AI的Gemini服务。

2. 文本嵌入服务禁用

spring.ai.model.embedding.text=none配置项专门用于禁用文本嵌入服务。这对于测试环境中不需要文本向量化功能的场景特别有用。

3. 多模态嵌入服务禁用

spring.ai.model.embedding.multimodal=none配置项则负责禁用多模态嵌入服务，确保系统不会尝试连接任何多模态模型服务。

最佳实践建议

1. 环境隔离：建议在application-test.properties或application-test.yml中专门配置这些属性，确保测试环境与开发/生产环境隔离。

2. 配置优先级：了解Spring Boot的配置优先级，确保测试配置能够覆盖主配置。测试配置通常具有更高的优先级。

3. 完整测试配置示例：

spring:
  ai:
    model:
      chat: 'none'
      embedding:
        text: 'none'
        multimodal: 'none'

4. 版本兼容性：虽然本文基于Spring AI 1.0.0版本，但这些配置在后续版本中应该保持兼容性。不过建议查阅对应版本的官方文档确认。

常见误区

1. 错误地使用vertex.ai前缀：如问题描述中所示，使用spring.ai.vertex.ai.embedding.text=none这样的配置是无效的，正确的配置路径应该是spring.ai.model开头。

2. 部分禁用：有些开发者可能只禁用聊天模型而忘记禁用嵌入服务，或者反之，这都会导致测试失败。需要确保所有相关服务都被正确禁用。

3. 配置位置错误：确保配置放在正确的配置文件中，并且该配置文件确实被测试环境加载。

总结

在Spring AI项目中正确禁用云AI服务是保证测试环境稳定运行的关键。通过本文介绍的配置方法，开发者可以轻松地在测试环境中禁用Vertex AI的Gemini和Embedding服务，避免不必要的云服务连接尝试，提高测试效率和稳定性。记住使用spring.ai.model作为配置前缀，而不是特定实现的vertex.ai前缀，这是解决问题的关键所在。