codemie-sdk-python 0.1.622__tar.gz → 0.1.624__tar.gz

{codemie_sdk_python-0.1.622 → codemie_sdk_python-0.1.624}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: codemie-sdk-python
-Version: 0.1.622
+Version: 0.1.624
 Summary: CodeMie SDK for Python
 Author: Vadym Vlasenko
 Author-email: vadym_vlasenko@epam.com

{codemie_sdk_python-0.1.622 → codemie_sdk_python-0.1.624}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "codemie-sdk-python"
-version = "0.1.622"
+version = "0.1.624"
 description = "CodeMie SDK for Python"
 authors = [
     "Vadym Vlasenko <vadym_vlasenko@epam.com>",

{codemie_sdk_python-0.1.622 → codemie_sdk_python-0.1.624}/src/codemie_sdk/__init__.py

@@ -157,6 +157,14 @@ from .models.integration import (
     TransferItem,
     TransferSettingsResponse,
 )
+from .models.a2a import (
+    AgentCard,
+    AgentProvider,
+    AgentCapabilities,
+    AgentAuthentication,
+    AgentSkill,
+)
+from .services.a2a import A2AService
 
 __version__ = "0.2.12"
 __all__ = [
@@ -276,4 +284,10 @@ __all__ = [
     "TransferMode",
     "TransferItem",
     "TransferSettingsResponse",
+    "AgentCard",
+    "AgentProvider",
+    "AgentCapabilities",
+    "AgentAuthentication",
+    "AgentSkill",
+    "A2AService",
 ]

{codemie_sdk_python-0.1.622 → codemie_sdk_python-0.1.624}/src/codemie_sdk/client/client.py

@@ -3,6 +3,7 @@
 from typing import Callable
 
 from ..auth.credentials import KeycloakCredentials, LocalAuthCredentials
+from ..services.a2a import A2AService
 from ..services.admin import AdminService
 from ..services.analytics import AnalyticsService
 from ..services.assistant import AssistantService
@@ -181,6 +182,9 @@ class CodeMieClient:
         self.project_budget_groups = ProjectBudgetGroupsService(
             self._api_domain, _token_source, verify_ssl=self._verify_ssl
         )
+        self.a2a = A2AService(
+            self._api_domain, _token_source, verify_ssl=self._verify_ssl
+        )
 
     @property
     def token(self) -> str:

codemie_sdk_python-0.1.624/src/codemie_sdk/models/a2a.py

@@ -0,0 +1,63 @@
+"""A2A (agent-to-agent) models."""
+
+from pydantic import BaseModel, ConfigDict
+
+
+class AgentProvider(BaseModel):
+    """Provider information for an A2A agent card."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    organization: str
+    url: str | None = None
+
+
+class AgentCapabilities(BaseModel):
+    """Capabilities advertised by an A2A agent card."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    streaming: bool = False
+    pushNotifications: bool = False
+    stateTransitionHistory: bool = False
+
+
+class AgentAuthentication(BaseModel):
+    """Authentication schemes advertised by an A2A agent card."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    schemes: list[str]
+    credentials: str | None = None
+
+
+class AgentSkill(BaseModel):
+    """A single skill exposed by an A2A agent card."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    id: str
+    name: str
+    description: str | None = None
+    tags: list[str] | None = None
+    examples: list[str] | None = None
+    inputModes: list[str] | None = None
+    outputModes: list[str] | None = None
+
+
+class AgentCard(BaseModel):
+    """A2A agent card describing an assistant for agent discovery."""
+
+    model_config = ConfigDict(extra="ignore")
+
+    name: str
+    description: str | None = None
+    url: str
+    provider: AgentProvider | None = None
+    version: str
+    documentationUrl: str | None = None
+    capabilities: AgentCapabilities
+    authentication: AgentAuthentication | None = None
+    defaultInputModes: list[str] = ["text"]
+    defaultOutputModes: list[str] = ["text"]
+    skills: list[AgentSkill]

codemie_sdk_python-0.1.624/src/codemie_sdk/services/a2a.py

@@ -0,0 +1,33 @@
+"""A2A (agent-to-agent) service implementation."""
+
+from ..models.a2a import AgentCard
+from ..utils import ApiRequestHandler, TokenSource
+
+
+class A2AService:
+    """Service for A2A agent discovery endpoints."""
+
+    def __init__(self, api_domain: str, token: TokenSource, verify_ssl: bool = True):
+        """Initialize the A2A service.
+
+        Args:
+            api_domain: Base URL for the CodeMie API
+            token: Authentication token
+            verify_ssl: Whether to verify SSL certificates
+        """
+        self._api = ApiRequestHandler(api_domain, token, verify_ssl)
+
+    def get_agent_card(self, assistant_id: str) -> AgentCard:
+        """Get the A2A agent card for an assistant.
+
+        Args:
+            assistant_id: ID of the assistant
+
+        Returns:
+            AgentCard: agent discovery metadata for the assistant
+        """
+        return self._api.get(
+            f"/v1/a2a/assistants/{assistant_id}/.well-known/agent.json",
+            AgentCard,
+            wrap_response=False,
+        )