Skip to content

feat(gcp): migrate AutoML to Vertex AI + implement LlmProvider#55

Open
Priyanshu-rgbb wants to merge 1 commit intoc2siorg:mainfrom
Priyanshu-rgbb:main-fix-the-legacy-GCP-AutoML-issue-and-migrate-it-to-Vertex-AI
Open

feat(gcp): migrate AutoML to Vertex AI + implement LlmProvider#55
Priyanshu-rgbb wants to merge 1 commit intoc2siorg:mainfrom
Priyanshu-rgbb:main-fix-the-legacy-GCP-AutoML-issue-and-migrate-it-to-Vertex-AI

Conversation

@Priyanshu-rgbb
Copy link

@Priyanshu-rgbb Priyanshu-rgbb commented Mar 13, 2026

closes #53

Summary--->

Migrates GCP AI module from deprecated AutoML to Vertex AI and implements the LlmProvider trait with Gemini model support.

Problem--->

Google is sunsetting standalone AutoML in favor of Vertex AI. Our gcp_automl.rs module uses deprecated endpoints that will stop working.

Solution

  • Created VertexAI struct implementing full LlmProvider trait
  • Migrated all dataset/model management methods to Vertex AI endpoints
  • Added Gemini support: text generation, streaming, embeddings, function calling
  • Marked old AutoML struct as #[deprecated] (non-breaking)

Changes--->

New files:

  • gcp_vertex_ai.rs - Core implementation with LlmProvider trait
  • gcp_vertex_ai_types.rs - 19 serde types for Vertex AI/Gemini API
  • gcp_vertex_ai_operations.rs - 16 test cases

Modified:

  • Cargo.toml - Added reqwest json/stream features
  • mod.rs files - Module registration
  • gcp_automl.rs - Added deprecation warning

Checklist--->

  • Implements LlmProvider trait for GCP
    -Migrates AutoML methods to Vertex AI
    -Includes test coverage
  • Non-breaking (old code compiles with warning)
    -Passes cargo fmt

…ex AI, ensuring compatibility with the latest GCP services and enhancing the functionality of the RustCloud library. This includes updating the relevant modules, adding new types for Vertex AI, and modifying existing tests to reflect the changes.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

feat: implement LlmProvider for GCP Vertex AI (Gemini)

1 participant