From 6e91d7233261bc5f21deb4fce4888462c5329ddd Mon Sep 17 00:00:00 2001
From: Google Team Member <genai-sdk-bot@google.com>
Date: Tue, 30 Jun 2026 15:11:00 -0700
Subject: [PATCH] fix: Support ref and def for memory schemas

PiperOrigin-RevId: 940680657
---
 .../types/StructuredMemorySchemaConfig.java   | 22 +++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/src/main/java/com/google/cloud/agentplatform/types/StructuredMemorySchemaConfig.java b/src/main/java/com/google/cloud/agentplatform/types/StructuredMemorySchemaConfig.java
index f079af8..8e92458 100644
--- a/src/main/java/com/google/cloud/agentplatform/types/StructuredMemorySchemaConfig.java
+++ b/src/main/java/com/google/cloud/agentplatform/types/StructuredMemorySchemaConfig.java
@@ -49,6 +49,10 @@ public abstract class StructuredMemorySchemaConfig extends JsonSerializable {
   @JsonProperty("memoryType")
   public abstract Optional<MemoryType> memoryType();
 
+  /** Required. Represents the OpenAPI schema of the structured memories. */
+  @JsonProperty("memoryJsonSchema")
+  public abstract Optional<Object> memoryJsonSchema();
+
   /** Instantiates a builder for StructuredMemorySchemaConfig. */
   @ExcludeFromGeneratedCoverageReport
   public static Builder builder() {
@@ -147,6 +151,24 @@ public Builder memoryType(String memoryType) {
       return memoryType(new MemoryType(memoryType));
     }
 
+    /**
+     * Setter for memoryJsonSchema.
+     *
+     * <p>memoryJsonSchema: Required. Represents the OpenAPI schema of the structured memories.
+     */
+    @JsonProperty("memoryJsonSchema")
+    public abstract Builder memoryJsonSchema(Object memoryJsonSchema);
+
+    @ExcludeFromGeneratedCoverageReport
+    abstract Builder memoryJsonSchema(Optional<Object> memoryJsonSchema);
+
+    /** Clears the value of memoryJsonSchema field. */
+    @ExcludeFromGeneratedCoverageReport
+    @CanIgnoreReturnValue
+    public Builder clearMemoryJsonSchema() {
+      return memoryJsonSchema(Optional.empty());
+    }
+
     public abstract StructuredMemorySchemaConfig build();
   }