Add memory bank to modules/agent-engine (#3844)
* feat(agent-engine): add support for memory bank configuration * refactor(agent-engine): remove source_path_override and revert to standard source_path * provider version upgrade from 7.17 to 7.27 * docs(agent-engine): fix README validation and update tables --------- Co-authored-by: Hemanand <hemr@google.com>
This commit is contained in:
@@ -15,6 +15,7 @@
|
||||
*/
|
||||
|
||||
resource "google_vertex_ai_reasoning_engine" "unmanaged" {
|
||||
provider = google-beta
|
||||
count = var.managed ? 0 : 1
|
||||
display_name = var.name
|
||||
project = local.project_id
|
||||
@@ -148,6 +149,71 @@ resource "google_vertex_ai_reasoning_engine" "unmanaged" {
|
||||
}
|
||||
}
|
||||
|
||||
dynamic "context_spec" {
|
||||
for_each = var.memory_bank_config != null ? { 1 = 1 } : {}
|
||||
|
||||
content {
|
||||
memory_bank_config {
|
||||
disable_memory_revisions = var.memory_bank_config.disable_memory_revisions
|
||||
|
||||
dynamic "generation_config" {
|
||||
for_each = (
|
||||
var.memory_bank_config.generation_config != null ? { 1 = 1 } : {}
|
||||
)
|
||||
content {
|
||||
model = lookup(
|
||||
local.ctx.models,
|
||||
var.memory_bank_config.generation_config.model,
|
||||
var.memory_bank_config.generation_config.model
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
dynamic "similarity_search_config" {
|
||||
for_each = (
|
||||
var.memory_bank_config.similarity_search_config != null ? { 1 = 1 } : {}
|
||||
)
|
||||
content {
|
||||
embedding_model = lookup(
|
||||
local.ctx.models,
|
||||
var.memory_bank_config.similarity_search_config.embedding_model,
|
||||
var.memory_bank_config.similarity_search_config.embedding_model
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
dynamic "ttl_config" {
|
||||
for_each = (
|
||||
var.memory_bank_config.ttl_config != null ? { 1 = 1 } : {}
|
||||
)
|
||||
content {
|
||||
default_ttl = var.memory_bank_config.ttl_config.default_ttl
|
||||
memory_revision_default_ttl = var.memory_bank_config.ttl_config.memory_revision_default_ttl
|
||||
|
||||
dynamic "granular_ttl_config" {
|
||||
for_each = (
|
||||
var.memory_bank_config.ttl_config.granular_ttl_config != null
|
||||
? { 1 = 1 }
|
||||
: {}
|
||||
)
|
||||
content {
|
||||
create_ttl = (
|
||||
var.memory_bank_config.ttl_config.granular_ttl_config.create_ttl
|
||||
)
|
||||
generate_created_ttl = (
|
||||
var.memory_bank_config.ttl_config.granular_ttl_config.generate_created_ttl
|
||||
)
|
||||
generate_updated_ttl = (
|
||||
var.memory_bank_config.ttl_config.granular_ttl_config.generate_updated_ttl
|
||||
)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
lifecycle {
|
||||
ignore_changes = [
|
||||
spec[0].package_spec,
|
||||
|
||||
Reference in New Issue
Block a user