Update Types from spec restrictions

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.
@@ -187,7 +188,7 @@ class AuthenticatedExtendedCardNotConfiguredError(A2ABaseModel):
     A primitive or structured value containing additional information about the error.
     This may be omitted.
     """
-    message: str | None = 'Authenticated Extended Card not configured'
+    message: str | None = 'Authenticated Extended Card is not configured'
     """
     The error message.
     """
@@ -198,12 +199,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.
@@ -213,7 +214,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.
@@ -225,7 +226,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.
     """
@@ -234,7 +235,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.
     """
@@ -390,7 +391,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.
     """
@@ -467,11 +468,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.
     """
@@ -740,7 +741,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.
     """
@@ -766,7 +767,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.
     """
@@ -775,7 +776,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.
     """
@@ -790,7 +791,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').
     """
@@ -814,7 +815,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.
     """
@@ -1371,7 +1372,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.
     """
@@ -1435,7 +1436,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).
@@ -1619,7 +1620,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']
     )
     """
@@ -1715,12 +1716,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.
@@ -1732,11 +1733,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.
     """
@@ -1777,7 +1778,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.
     """
@@ -1786,14 +1787,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.
     """