Merge branch 'main' into sdk_buf

Author: rajeshvelicheti

.github/workflows/linter.yaml

@@ -64,3 +64,4 @@ jobs:
           VALIDATE_GIT_COMMITLINT: false
           PYTHON_MYPY_CONFIG_FILE: .mypy.ini
           FILTER_REGEX_INCLUDE: "^src/**"
+          PYTHON_RUFF_CONFIG_FILE: .ruff.toml

.ruff.toml

@@ -18,11 +18,7 @@ ignore = [
     "FBT002",
     "D203",
     "D213",
-    "ANN001",
-    "ANN201",
-    "ANN204",
     "D100", # Ignore Missing docstring in public module (often desired at top level __init__.py)
-    "D102", # Ignore return type annotation in public method
     "D104", # Ignore Missing docstring in public package (often desired at top level __init__.py)
     "D107", # Ignore Missing docstring in __init__ (use class docstring)
     "TD002", # Ignore Missing author in TODOs (often not required)
@@ -116,7 +112,7 @@ docstring-quotes = "double"
 inline-quotes = "single"
 
 [lint.per-file-ignores]
-"__init__.py" = ["F401"]  # Ignore unused imports in __init__.py
+"__init__.py" = ["F401", "D", "ANN"]  # Ignore unused imports in __init__.py
 "*_test.py" = ["D", "ANN"]  # Ignore docstring and annotation issues in test files
 "test_*.py" = ["D", "ANN"]  # Ignore docstring and annotation issues in test files
 "types.py" = ["D", "E501", "N815"]  # Ignore docstring and annotation issues in types.py

src/a2a/types.py

@@ -43,22 +43,26 @@ class APIKeySecurityScheme(BaseModel):
     type: Literal['apiKey'] = 'apiKey'
 
 
-class AgentCapabilities(BaseModel):
+class AgentExtension(BaseModel):
     """
-    Defines optional capabilities supported by an agent.
+    A declaration of an extension supported by an Agent.
     """
 
-    pushNotifications: bool | None = None
+    description: str | None = None
     """
-    true if the agent can notify updates to client.
+    A description of how this agent uses this extension.
     """
-    stateTransitionHistory: bool | None = None
+    params: dict[str, Any] | None = None
     """
-    true if the agent exposes status change history for tasks.
+    Optional configuration for the extension.
     """
-    streaming: bool | None = None
+    required: bool | None = None
     """
-    true if the agent supports SSE.
+    Whether the client must follow specific requirements of the extension.
+    """
+    uri: str
+    """
+    The URI of the extension.
     """
 
 
@@ -820,6 +824,29 @@ class A2AError(
     )
 
 
+class AgentCapabilities(BaseModel):
+    """
+    Defines optional capabilities supported by an agent.
+    """
+
+    extensions: list[AgentExtension] | None = None
+    """
+    extensions supported by this agent.
+    """
+    pushNotifications: bool | None = None
+    """
+    true if the agent can notify updates to client.
+    """
+    stateTransitionHistory: bool | None = None
+    """
+    true if the agent exposes status change history for tasks.
+    """
+    streaming: bool | None = None
+    """
+    true if the agent supports SSE.
+    """
+
+
 class CancelTaskRequest(BaseModel):
     """
     JSON-RPC request model for the 'tasks/cancel' method.
@@ -1075,6 +1102,10 @@ class Artifact(BaseModel):
     """
     Optional description for the artifact.
     """
+    extensions: list[str] | None = None
+    """
+    The URIs of extensions that are present or contributed to this Artifact.
+    """
     metadata: dict[str, Any] | None = None
     """
     Extension metadata.
@@ -1107,6 +1138,10 @@ class Message(BaseModel):
     """
     The context the message is associated with
     """
+    extensions: list[str] | None = None
+    """
+    The URIs of extensions that are present or contributed to this Message.
+    """
     kind: Literal['message'] = 'message'
     """
     Event type