Skip to content

Enable Memory Bank Configurations in Vertex AI Agent Engine (prev Reasoning Engine) #26522

@heyanand

Description

@heyanand

Community Note

  • Please vote on this issue by adding a 👍 reaction to the original issue to help the community and maintainers prioritize this request.
  • Please do not leave +1 or me too comments, they generate extra noise for issue followers and do not help prioritize the request.
  • If you are interested in working on this issue or have submitted a pull request, please leave a comment.
  • If an issue is assigned to a user, that user is claiming responsibility for the issue.
  • Customers working with a Google Technical Account Manager or Customer Engineer can ask them to reach out internally to expedite investigation and resolution of this issue.

Description

The documentation available at https://docs.cloud.google.com/vertex-ai/docs/reference/rest/v1beta1/projects.locations.reasoningEngines#MemoryBankConfig contains the configurations for setting up Memory Bank with Agent Engine.

This feature is needed for enterprises to be able to automate their infra setup when building their Agentic AI apps on GCP and MemoryBank is a crucial feature for building agent.

New or Affected Resource(s)

  • google_vertex_ai_reasoning_engine

Potential Terraform Configuration

resource "google_vertex_ai_reasoning_engine" "test_engine" {
  provider     = google-beta
  display_name = "test-mem-bank-engine"
  description  = "Test engine with Memory Bank Config"
  force_destroy = true
  region       = var.region

  spec {
    agent_framework = "google-adk"
    source_code_spec {
      inline_source {
        source_archive = filebase64(data.archive_file.source.output_path)
      }
      python_spec {
        entrypoint_module = var.source_config.entrypoint_module
        entrypoint_object = var.source_config.entrypoint_object
        requirements_file = var.source_config.requirements_path
        version           = "3.12" # Specify your Python version
      }
    }
  }

  context_spec {
    memory_bank_config {
      generation_config {
        model = "projects/${var.project_id}/locations/${var.region}/publishers/google/models/gemini-2.5-flash"
      }
      similarity_search_config {
        embedding_model = "projects/${var.project_id}/locations/${var.region}/publishers/google/models/text-embedding-005"
      }
      disable_memory_revisions = false
    }
  }
}

References

No response

b/495458451

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions