63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
|
"""
|
||
|
|
Base configuration for A2A protocol providers.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from abc import ABC, abstractmethod
|
||
|
|
from typing import Any, AsyncIterator, Dict
|
||
|
|
|
||
|
|
|
||
|
|
class BaseA2AProviderConfig(ABC):
|
||
|
|
"""
|
||
|
|
Base configuration class for A2A protocol providers.
|
||
|
|
|
||
|
|
Each provider should implement this interface to define how to handle
|
||
|
|
A2A requests for their specific agent type.
|
||
|
|
"""
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
async def handle_non_streaming(
|
||
|
|
self,
|
||
|
|
request_id: str,
|
||
|
|
params: Dict[str, Any],
|
||
|
|
api_base: str,
|
||
|
|
**kwargs,
|
||
|
|
) -> Dict[str, Any]:
|
||
|
|
"""
|
||
|
|
Handle non-streaming A2A request.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
request_id: A2A JSON-RPC request ID
|
||
|
|
params: A2A MessageSendParams containing the message
|
||
|
|
api_base: Base URL of the agent
|
||
|
|
**kwargs: Additional provider-specific parameters
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
A2A SendMessageResponse dict
|
||
|
|
"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
@abstractmethod
|
||
|
|
async def handle_streaming(
|
||
|
|
self,
|
||
|
|
request_id: str,
|
||
|
|
params: Dict[str, Any],
|
||
|
|
api_base: str,
|
||
|
|
**kwargs,
|
||
|
|
) -> AsyncIterator[Dict[str, Any]]:
|
||
|
|
"""
|
||
|
|
Handle streaming A2A request.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
request_id: A2A JSON-RPC request ID
|
||
|
|
params: A2A MessageSendParams containing the message
|
||
|
|
api_base: Base URL of the agent
|
||
|
|
**kwargs: Additional provider-specific parameters
|
||
|
|
|
||
|
|
Yields:
|
||
|
|
A2A streaming response events
|
||
|
|
"""
|
||
|
|
# This is an abstract method - subclasses must implement
|
||
|
|
# The yield is here to make this a generator function
|
||
|
|
if False: # pragma: no cover
|
||
|
|
yield {}
|