mirror of
https://github.com/encounter/adk-python.git
synced 2026-07-09 18:19:28 -07:00
feat: expose credential service in readonly context
This commit is contained in:
@@ -23,6 +23,7 @@ if TYPE_CHECKING:
|
|||||||
from google.genai import types
|
from google.genai import types
|
||||||
|
|
||||||
from .invocation_context import InvocationContext
|
from .invocation_context import InvocationContext
|
||||||
|
from ..auth.credential_service.base_credential_service import BaseCredentialService
|
||||||
|
|
||||||
|
|
||||||
class ReadonlyContext:
|
class ReadonlyContext:
|
||||||
@@ -52,3 +53,8 @@ class ReadonlyContext:
|
|||||||
def state(self) -> MappingProxyType[str, Any]:
|
def state(self) -> MappingProxyType[str, Any]:
|
||||||
"""The state of the current session. READONLY field."""
|
"""The state of the current session. READONLY field."""
|
||||||
return MappingProxyType(self._invocation_context.session.state)
|
return MappingProxyType(self._invocation_context.session.state)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def credential_service(self) -> Optional[BaseCredentialService]:
|
||||||
|
"""The credential service for the current invocation."""
|
||||||
|
return self._invocation_context.credential_service
|
||||||
|
|||||||
@@ -116,6 +116,7 @@ class AgentTool(BaseTool):
|
|||||||
artifact_service=ForwardingArtifactService(tool_context),
|
artifact_service=ForwardingArtifactService(tool_context),
|
||||||
session_service=InMemorySessionService(),
|
session_service=InMemorySessionService(),
|
||||||
memory_service=InMemoryMemoryService(),
|
memory_service=InMemoryMemoryService(),
|
||||||
|
credential_service=tool_context.credential_service,
|
||||||
)
|
)
|
||||||
session = await runner.session_service.create_session(
|
session = await runner.session_service.create_session(
|
||||||
app_name=self.agent.name,
|
app_name=self.agent.name,
|
||||||
|
|||||||
@@ -2,15 +2,25 @@ from types import MappingProxyType
|
|||||||
from unittest.mock import MagicMock
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
from google.adk.agents.readonly_context import ReadonlyContext
|
from google.adk.agents.readonly_context import ReadonlyContext
|
||||||
|
from google.adk.auth.credential_service.base_credential_service import BaseCredentialService
|
||||||
import pytest
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
|
class DummyCredentialService(BaseCredentialService):
|
||||||
|
async def load_credential(self,auth_config, tool_context):
|
||||||
|
pass
|
||||||
|
|
||||||
|
async def save_credential(self,auth_config, tool_context):
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def mock_invocation_context():
|
def mock_invocation_context():
|
||||||
mock_context = MagicMock()
|
mock_context = MagicMock()
|
||||||
mock_context.invocation_id = "test-invocation-id"
|
mock_context.invocation_id = "test-invocation-id"
|
||||||
mock_context.agent.name = "test-agent-name"
|
mock_context.agent.name = "test-agent-name"
|
||||||
mock_context.session.state = {"key1": "value1", "key2": "value2"}
|
mock_context.session.state = {"key1": "value1", "key2": "value2"}
|
||||||
|
mock_context.credential_service = DummyCredentialService()
|
||||||
|
|
||||||
return mock_context
|
return mock_context
|
||||||
|
|
||||||
@@ -32,3 +42,9 @@ def test_state_content(mock_invocation_context):
|
|||||||
assert isinstance(state, MappingProxyType)
|
assert isinstance(state, MappingProxyType)
|
||||||
assert state["key1"] == "value1"
|
assert state["key1"] == "value1"
|
||||||
assert state["key2"] == "value2"
|
assert state["key2"] == "value2"
|
||||||
|
|
||||||
|
|
||||||
|
def test_credential_service(mock_invocation_context):
|
||||||
|
readonly_context = ReadonlyContext(mock_invocation_context)
|
||||||
|
assert readonly_context.credential_service is not None
|
||||||
|
assert isinstance(readonly_context.credential_service, BaseCredentialService)
|
||||||
|
|||||||
Reference in New Issue
Block a user