Update types

Author: holtskinner

src/a2a/types.py

@@ -1,12 +1,13 @@
 # generated by datamodel-codegen:
-#   filename:  https://raw.githubusercontent.com/a2aproject/A2A/refs/heads/main/specification/json/a2a.json
+#   filename:  https://raw.githubusercontent.com/a2aproject/A2A/refs/heads/jsdoc-strings/specification/json/a2a.json
 
 from __future__ import annotations
 
+from datetime import datetime
 from enum import Enum
 from typing import Any, Literal
 
-from pydantic import Field, RootModel
+from pydantic import AnyUrl, Field, RootModel
 
 from a2a._base import A2ABaseModel
 
@@ -87,7 +88,7 @@ class AgentExtension(A2ABaseModel):
     If true, the client must understand and comply with the extension's requirements
     to interact with the agent.
     """
-    uri: str
+    uri: AnyUrl
     """
     The unique URI identifying the extension.
     """
@@ -103,7 +104,7 @@ class AgentInterface(A2ABaseModel):
     """
     The transport protocol supported at this URL.
     """
-    url: str = Field(
+    url: AnyUrl = Field(
         ...,
         examples=[
             'https://api.example.com/a2a/v1',
@@ -125,7 +126,7 @@ class AgentProvider(A2ABaseModel):
     """
     The name of the agent provider's organization.
     """
-    url: str
+    url: AnyUrl
     """
     A URL for the agent provider's website or relevant documentation.
     """
@@ -165,7 +166,7 @@ class AgentSkill(A2ABaseModel):
     The set of supported output MIME types for this skill, overriding the agent's defaults.
     """
     tags: list[str] = Field(
-        ..., examples=[['cooking', 'customer support', 'billing']]
+        ..., examples=[['cooking', 'customer support', 'billing']], min_length=1
     )
     """
     A set of keywords describing the skill's capabilities.
@@ -177,12 +178,12 @@ class AuthorizationCodeOAuthFlow(A2ABaseModel):
     Defines configuration details for the OAuth 2.0 Authorization Code flow.
     """
 
-    authorization_url: str
+    authorization_url: AnyUrl
     """
     The authorization URL to be used for this flow.
     This MUST be a URL and use TLS.
     """
-    refresh_url: str | None = None
+    refresh_url: AnyUrl | None = None
     """
     The URL to be used for obtaining refresh tokens.
     This MUST be a URL and use TLS.
@@ -192,7 +193,7 @@ class AuthorizationCodeOAuthFlow(A2ABaseModel):
     The available scopes for the OAuth2 security scheme. A map between the scope
     name and a short description for it.
     """
-    token_url: str
+    token_url: AnyUrl
     """
     The token URL to be used for this flow.
     This MUST be a URL and use TLS.
@@ -204,7 +205,7 @@ class ClientCredentialsOAuthFlow(A2ABaseModel):
     Defines configuration details for the OAuth 2.0 Client Credentials flow.
     """
 
-    refresh_url: str | None = None
+    refresh_url: AnyUrl | None = None
     """
     The URL to be used for obtaining refresh tokens. This MUST be a URL.
     """
@@ -213,7 +214,7 @@ class ClientCredentialsOAuthFlow(A2ABaseModel):
     The available scopes for the OAuth2 security scheme. A map between the scope
     name and a short description for it.
     """
-    token_url: str
+    token_url: AnyUrl
     """
     The token URL to be used for this flow. This MUST be a URL.
     """
@@ -369,7 +370,7 @@ class FileWithUri(A2ABaseModel):
     """
     An optional name for the file (e.g., "document.pdf").
     """
-    uri: str
+    uri: AnyUrl
     """
     A URL pointing to the file's content.
     """
@@ -425,11 +426,11 @@ class ImplicitOAuthFlow(A2ABaseModel):
     Defines configuration details for the OAuth 2.0 Implicit flow.
     """
 
-    authorization_url: str
+    authorization_url: AnyUrl
     """
     The authorization URL to be used for this flow. This MUST be a URL.
     """
-    refresh_url: str | None = None
+    refresh_url: AnyUrl | None = None
     """
     The URL to be used for obtaining refresh tokens. This MUST be a URL.
     """
@@ -698,7 +699,7 @@ class OpenIdConnectSecurityScheme(A2ABaseModel):
     """
     An optional description for the security scheme.
     """
-    open_id_connect_url: str
+    open_id_connect_url: AnyUrl
     """
     The OpenID Connect Discovery URL for the OIDC provider's metadata.
     """
@@ -724,7 +725,7 @@ class PasswordOAuthFlow(A2ABaseModel):
     Defines configuration details for the OAuth 2.0 Resource Owner Password flow.
     """
 
-    refresh_url: str | None = None
+    refresh_url: AnyUrl | None = None
     """
     The URL to be used for obtaining refresh tokens. This MUST be a URL.
     """
@@ -733,7 +734,7 @@ class PasswordOAuthFlow(A2ABaseModel):
     The available scopes for the OAuth2 security scheme. A map between the scope
     name and a short description for it.
     """
-    token_url: str
+    token_url: AnyUrl
     """
     The token URL to be used for this flow. This MUST be a URL.
     """
@@ -748,7 +749,7 @@ class PushNotificationAuthenticationInfo(A2ABaseModel):
     """
     Optional credentials required by the push notification endpoint.
     """
-    schemes: list[str]
+    schemes: list[str] = Field(..., min_length=1)
     """
     A list of supported authentication schemes (e.g., 'Basic', 'Bearer').
     """
@@ -772,7 +773,7 @@ class PushNotificationConfig(A2ABaseModel):
     """
     A unique token for this task or session to validate incoming push notifications.
     """
-    url: str
+    url: AnyUrl
     """
     The callback URL where the agent should send push notifications.
     """
@@ -1326,7 +1327,7 @@ class Artifact(A2ABaseModel):
     """
     An optional, human-readable name for the artifact.
     """
-    parts: list[Part]
+    parts: list[Part] = Field(..., min_length=1)
     """
     An array of content parts that make up the artifact.
     """
@@ -1390,7 +1391,7 @@ class Message(A2ABaseModel):
     """
     Optional metadata for extensions. The key is an extension-specific identifier.
     """
-    parts: list[Part]
+    parts: list[Part] = Field(..., min_length=1)
     """
     An array of content parts that form the message body. A message can be
     composed of multiple parts of different types (e.g., text and files).
@@ -1574,7 +1575,7 @@ class TaskStatus(A2ABaseModel):
     """
     The current state of the task's lifecycle.
     """
-    timestamp: str | None = Field(
+    timestamp: datetime | None = Field(
         default=None, examples=['2023-10-27T10:00:00Z']
     )
     """
@@ -1668,12 +1669,12 @@ class AgentCard(A2ABaseModel):
     """
     A declaration of optional capabilities supported by the agent.
     """
-    default_input_modes: list[str]
+    default_input_modes: list[str] = Field(..., min_length=1)
     """
     Default set of supported input MIME types for all skills, which can be
     overridden on a per-skill basis.
     """
-    default_output_modes: list[str]
+    default_output_modes: list[str] = Field(..., min_length=1)
     """
     Default set of supported output MIME types for all skills, which can be
     overridden on a per-skill basis.
@@ -1685,11 +1686,11 @@ class AgentCard(A2ABaseModel):
     A human-readable description of the agent, assisting users and other agents
     in understanding its purpose.
     """
-    documentation_url: str | None = None
+    documentation_url: AnyUrl | None = None
     """
     An optional URL to the agent's documentation.
     """
-    icon_url: str | None = None
+    icon_url: AnyUrl | None = None
     """
     An optional URL to an icon for the agent.
     """
@@ -1730,7 +1731,7 @@ class AgentCard(A2ABaseModel):
     """
     JSON Web Signatures computed for this AgentCard.
     """
-    skills: list[AgentSkill]
+    skills: list[AgentSkill] = Field(..., min_length=1)
     """
     The set of skills, or distinct capabilities, that the agent can perform.
     """
@@ -1739,14 +1740,18 @@ class AgentCard(A2ABaseModel):
     If true, the agent can provide an extended agent card with additional details
     to authenticated users. Defaults to false.
     """
-    url: str = Field(..., examples=['https://api.example.com/a2a/v1'])
+    url: AnyUrl = Field(..., examples=['https://api.example.com/a2a/v1'])
     """
     The preferred endpoint URL for interacting with the agent.
     This URL MUST support the transport specified by 'preferredTransport'.
     """
-    version: str = Field(..., examples=['1.0.0'])
+    version: str = Field(
+        ...,
+        examples=['1.0.0'],
+        pattern='^(0|[1-9]\\d*)\\.(0|[1-9]\\d*)\\.(0|[1-9]\\d*)(?:-((?:0|[1-9]\\d*|\\d*[a-zA-Z-][0-9a-zA-Z-]*)(?:\\.(?:0|[1-9]\\d*|\\d*[a-zA-Z-][0-9a-zA-Z-]*))*))?(?:\\+([0-9a-zA-Z-]+(?:\\.[0-9a-zA-Z-]+)*))?$',
+    )
     """
-    The agent's own version number. The format is defined by the provider.
+    The agent's own version number. Semantic Versioning MUST be used.
     """