Merge remote-tracking branch 'upstream/main' into delete-minimal-catalog

Author: josemontespg

.github/PULL_REQUEST_TEMPLATE.md

@@ -21,5 +21,5 @@ If you need help, consider asking for advice on the [discussion board].
 [CHANGELOG]: ../CHANGELOG.md
 [CLA]: https://cla.developers.google.com/
 [Contributors Guide]: ../CONTRIBUTING.md
-[discussion board]: https://github.com/google/A2UI/discussions
+[discussion board]: https://github.com/a2ui-project/a2ui/discussions
 [Style Guide]: ../CONTRIBUTING.md#coding-style

.github/workflows/docs.yml

@@ -39,7 +39,7 @@ jobs:
       contents: write
       actions: read
 
-    if: github.repository == 'google/A2UI'
+    if: github.repository == 'a2ui-project/a2ui'
 
     steps:
       - name: Checkout Code

.github/workflows/e2e_test.yaml

@@ -15,7 +15,7 @@ jobs:
   e2e_test:
     # Do not run on forked branches,
     # because the test does not have access to secrets in forks.
-    if: github.repository == 'google/a2ui'
+    if: github.repository == 'a2ui-project/a2ui'
     runs-on: ubuntu-latest
     permissions:
       contents: read

README.md

@@ -114,8 +114,8 @@ Pick the path that matches where you want to start:
 Prerequisites: Node.js 18+, [uv](https://docs.astral.sh/uv/), and a [Gemini API key](https://aistudio.google.com/apikey).
 
 ```bash
-git clone https://github.com/google/A2UI.git
-cd A2UI
+git clone https://github.com/a2ui-project/a2ui.git
+cd a2ui
 export GEMINI_API_KEY="your_gemini_api_key"
 cd samples/client/lit
 npm run demo:restaurant

agent_sdks/agent_sdk_guide.md

@@ -262,7 +262,7 @@ Create the helper utilities to wrap JSON in transport Parts (if needed for your
 
 ### Step 5: Sample Applications
 
-Create a simple sample (like a command-line agent or local server) to verify that the SDK works end-to-end. Refer to the reference Python samples (e.g., `samples/agent/adk/contact_lookup`) for inspiration.
+Create a simple sample (like a command-line agent or local server) to verify that the SDK works end-to-end. Refer to the reference Python samples (e.g., `samples/agent/adk/restaurant_finder`) for inspiration.
 
 > [!IMPORTANT]
 > Keep the SDK idiomatic to your language. Don't force Python-isms if it doesn't make sense (e.g., use builder patterns in Java/Kotlin or macros in C++ if they are more ergonomic).

agent_sdks/kotlin/build.gradle.kts

@@ -39,7 +39,7 @@ repositories {
 
 dependencies {
   api("org.jetbrains.kotlinx:kotlinx-serialization-json:1.6.3")
-  implementation("com.networknt:json-schema-validator:1.5.1")
+  implementation("com.networknt:json-schema-validator:2.0.1")
   implementation("com.fasterxml.jackson.core:jackson-databind:2.17.2")
 
   // Core Dependencies

agent_sdks/kotlin/src/main/kotlin/com/google/a2ui/schema/Validator.kt

@@ -16,11 +16,11 @@
 
 package com.google.a2ui.schema
 
-import com.fasterxml.jackson.databind.ObjectMapper
-import com.networknt.schema.JsonSchema
-import com.networknt.schema.JsonSchemaFactory
-import com.networknt.schema.SchemaValidatorsConfig
-import com.networknt.schema.SpecVersion
+import com.networknt.schema.InputFormat
+import com.networknt.schema.Schema
+import com.networknt.schema.SchemaRegistry
+import com.networknt.schema.SchemaRegistryConfig
+import com.networknt.schema.dialect.Dialects
 import kotlinx.serialization.json.Json
 import kotlinx.serialization.json.JsonArray
 import kotlinx.serialization.json.JsonElement
@@ -44,21 +44,18 @@ constructor(
   private val catalog: A2uiCatalog,
   private val schemaMappings: Map<String, String> = emptyMap(),
 ) {
-  private val validator: JsonSchema = buildValidator()
-  private val mapper = ObjectMapper()
-  private val shared0_9Factory: JsonSchemaFactory by lazy {
-    JsonSchemaFactory.builder(JsonSchemaFactory.getInstance(SpecVersion.VersionFlag.V202012))
-      .schemaMappers { schemaMappers ->
-        schemaMappings.forEach { (prefix, target) -> schemaMappers.mapPrefix(prefix, target) }
+  private val shared0_9Registry: SchemaRegistry by lazy {
+    SchemaRegistry.withDialect(Dialects.getDraft202012()) { builder ->
+      builder.schemaIdResolvers { schemaIdResolvers ->
+        schemaMappings.forEach { (prefix, target) -> schemaIdResolvers.mapPrefix(prefix, target) }
       }
-      .build()
-  }
-  private val sharedConfig: SchemaValidatorsConfig by lazy {
-    SchemaValidatorsConfig.builder().build()
+    }
   }
-  private val subValidators = mutableMapOf<String, JsonSchema>()
+  private val sharedConfig: SchemaRegistryConfig by lazy { SchemaRegistryConfig.builder().build() }
+  private val subValidators = mutableMapOf<String, Schema>()
+  private val validator: Schema = buildValidator()
 
-  private fun buildValidator(): JsonSchema =
+  private fun buildValidator(): Schema =
     if (catalog.version == A2uiVersion.VERSION_0_8) build0_8Validator() else build0_9Validator()
 
   private fun injectAdditionalProperties(
@@ -121,7 +118,7 @@ constructor(
     return bundled as JsonObject
   }
 
-  private fun build0_8Validator(): JsonSchema {
+  private fun build0_8Validator(): Schema {
     val bundledSchema = bundle0_8Schemas()
     val fullSchema = SchemaResourceLoader.wrapAsJsonArray(bundledSchema).toMutableMap()
     fullSchema[KEY_DOLLAR_SCHEMA] = JsonPrimitive(SCHEMA_DRAFT_2020_12)
@@ -132,38 +129,29 @@ constructor(
     val baseDirUri = baseUri.substringBeforeLast("/")
     val commonTypesUri = "$baseDirUri/$FILE_COMMON_TYPES"
 
-    val config = SchemaValidatorsConfig.builder().build()
-
     val jsonFmt = Json { prettyPrint = false }
 
-    val factory =
-      JsonSchemaFactory.builder(JsonSchemaFactory.getInstance(SpecVersion.VersionFlag.V202012))
-        .schemaMappers { schemaMappers ->
-          schemaMappings.forEach { (prefix, target) -> schemaMappers.mapPrefix(prefix, target) }
-          schemaMappers.mapPrefix(FILE_COMMON_TYPES, commonTypesUri)
+    val registry =
+      SchemaRegistry.withDialect(Dialects.getDraft202012()) { builder ->
+        builder.schemaIdResolvers { schemaIdResolvers ->
+          schemaMappings.forEach { (prefix, target) -> schemaIdResolvers.mapPrefix(prefix, target) }
+          schemaIdResolvers.mapPrefix(FILE_COMMON_TYPES, commonTypesUri)
         }
-        .build()
+        builder.schemaRegistryConfig(sharedConfig)
+      }
 
     val schemaString = jsonFmt.encodeToString(JsonElement.serializer(), JsonObject(fullSchema))
-    return factory.getSchema(schemaString, config)
+    return registry.getSchema(schemaString, InputFormat.JSON)
   }
 
-  private fun build0_9Validator(): JsonSchema {
+  private fun build0_9Validator(): Schema {
     val fullSchema =
       SchemaResourceLoader.wrapAsJsonArray(catalog.serverToClientSchema).toMutableMap()
     fullSchema[KEY_DOLLAR_SCHEMA] = JsonPrimitive(SCHEMA_DRAFT_2020_12)
 
-    val config = SchemaValidatorsConfig.builder().build()
-    val factory =
-      JsonSchemaFactory.builder(JsonSchemaFactory.getInstance(SpecVersion.VersionFlag.V202012))
-        .schemaMappers { schemaMappers ->
-          schemaMappings.forEach { (prefix, target) -> schemaMappers.mapPrefix(prefix, target) }
-        }
-        .build()
-
     val jsonFmt = Json { prettyPrint = false }
     val schemaString = jsonFmt.encodeToString(JsonElement.serializer(), JsonObject(fullSchema))
-    return factory.getSchema(schemaString, config)
+    return shared0_9Registry.getSchema(schemaString, InputFormat.JSON)
   }
 
   /**
@@ -185,9 +173,8 @@ constructor(
       // Basic schema validation
       val jsonFmt = Json { prettyPrint = false }
       val messagesString = jsonFmt.encodeToString(JsonElement.serializer(), messages)
-      val jsonNode = mapper.readTree(messagesString)
 
-      val errors = validator.validate(jsonNode)
+      val errors = validator.validate(messagesString, InputFormat.JSON)
       if (errors.isNotEmpty()) {
         val msg = buildString {
           append("Validation failed:")
@@ -290,7 +277,7 @@ constructor(
     }
   }
 
-  private fun getSubValidator(defName: String): JsonSchema {
+  private fun getSubValidator(defName: String): Schema {
     return subValidators.getOrPut(defName) {
       val defs =
         catalog.serverToClientSchema["\$defs"] as? JsonObject
@@ -310,22 +297,21 @@ constructor(
 
       val jsonFmt = Json { prettyPrint = false }
       val schemaString = jsonFmt.encodeToString(JsonElement.serializer(), tempSchema)
-      shared0_9Factory.getSchema(schemaString, sharedConfig)
+      shared0_9Registry.getSchema(schemaString, InputFormat.JSON)
     }
   }
 
   private fun getFormattedErrors(
-    validator: JsonSchema,
+    validator: Schema,
     instance: JsonElement,
     basePath: String,
   ): List<String> {
     val jsonFmt = Json { prettyPrint = false }
     val instanceStr = jsonFmt.encodeToString(JsonElement.serializer(), instance)
-    val jsonNode = mapper.readTree(instanceStr)
-    val errors = validator.validate(jsonNode)
+    val errors = validator.validate(instanceStr, InputFormat.JSON)
 
     return errors.map { err ->
-      val msg = err.message ?: ""
+      val msg = err.toString()
       val unexpectedRegex =
         Regex(
           "property '(.*?)' is not defined in the schema and the schema does not allow additional properties"
@@ -419,7 +405,7 @@ constructor(
           )
         val jsonFmt = Json { prettyPrint = false }
         val schemaString = jsonFmt.encodeToString(JsonElement.serializer(), tempSchema)
-        shared0_9Factory.getSchema(schemaString, sharedConfig)
+        shared0_9Registry.getSchema(schemaString, InputFormat.JSON)
       }
 
     return getFormattedErrors(validator, comp, path)

agent_sdks/python/README.md

@@ -77,21 +77,7 @@ uv run pyink .
 
 ## Releasing the SDK
 
-To release a new version of the SDK, follow these steps:
-
-1. Update the version in `src/a2ui/version.py`.
-2. Run the release script from the `agent_sdks/python` directory:
-
-   ```bash
-   ./release.sh
-   ```
-
-The script will build the package, upload it to the Artifact Registry, and trigger the release pipeline.
-
-## Tracking the release
-
-1. Check the release status through Sponge: go/spng2?q=PROJECT%3Aa2ui%2Fpypi
-2. You will see logs for the jobs involved in your release. A successful release is confirmed when the final job, named "publisher", completes successfully.
+See internal guidance at go/a2ui-release-pipy.
 
 ## Disclaimer
 

agent_sdks/python/src/a2ui/adk/a2a/part_converter.py

@@ -55,17 +55,25 @@ class A2uiPartConverter:
   This converter handles both tool-based A2UI (via `send_a2ui_json_to_client`)
   and text-based A2UI (via A2UI delimiter tags). It uses the provided
   catalog to validate and fix JSON payloads.
+
+  Args:
+      a2ui_catalog: The A2UI catalog.
+      bypass_tool_check: If True, bypass tool validation.
+      fallback_text: Optional text to fall back on if parsing fails.
+      version: The A2UI version to use. Defaults to "0.8".
   """
 
   def __init__(
       self,
       a2ui_catalog: A2uiCatalog,
       bypass_tool_check: bool = False,
       fallback_text: Optional[str] = None,
+      version: str = constants.VERSION_0_8,
   ):
     self._catalog = a2ui_catalog
     self._bypass_tool_check = bypass_tool_check
     self._fallback_text = fallback_text
+    self._version = version
 
   def convert(self, part: genai_types.Part) -> list[a2a_types.Part]:
     """Converts a GenAI part to A2A parts, with A2UI validation.
@@ -97,7 +105,10 @@ def convert(self, part: genai_types.Part) -> list[a2a_types.Part]:
         ):
           json_data = response_dict.get(constants.A2UI_VALIDATED_JSON_KEY)
           if json_data:
-            return [create_a2ui_part(message) for message in json_data]
+            return [
+                create_a2ui_part(message, version=self._version)
+                for message in json_data
+            ]
 
         if is_send_a2ui_json_to_client_response:
           logger.info("No result in A2UI tool response")
@@ -111,6 +122,7 @@ def convert(self, part: genai_types.Part) -> list[a2a_types.Part]:
               result,
               validator=self._catalog.validator,
               fallback_text=self._fallback_text,
+              version=self._version,
           )
 
     # 2. Handle Tool Calls (FunctionCall) - Skip sending to client
@@ -126,6 +138,7 @@ def convert(self, part: genai_types.Part) -> list[a2a_types.Part]:
             text,
             validator=self._catalog.validator,
             fallback_text=self._fallback_text,
+            version=self._version,
         )
 
     # 4. Default conversion for other parts

agent_sdks/python/src/a2ui/version.py

@@ -21,4 +21,4 @@
 #   Switch to VCS-based versioning which derives version
 #   from Git tags automatically with the hatch-vcs plugin
 #   when we start using Git tags.
-__version__ = "0.2.2"
+__version__ = "0.2.4"