wehub-resource-sync
ec2b666284
Continuous Integration / Pre-commit Linter (push) Waiting to run
Continuous Integration / Mypy Check (Python 3.10) (push) Waiting to run
Continuous Integration / Mypy Check (Python 3.11) (push) Waiting to run
Continuous Integration / Mypy Check (Python 3.12) (push) Waiting to run
Continuous Integration / Mypy Check (Python 3.13) (push) Waiting to run
Continuous Integration / Unit Tests (Python 3.10) (push) Waiting to run
Continuous Integration / Unit Tests (Python 3.11) (push) Waiting to run
Continuous Integration / Unit Tests (Python 3.12) (push) Waiting to run
Continuous Integration / Unit Tests (Python 3.13) (push) Waiting to run
Continuous Integration / Unit Tests (Python 3.14) (push) Waiting to run
Continuous Integration / A2A v0.3 Tests (Python 3.10) (push) Waiting to run
Continuous Integration / A2A v0.3 Tests (Python 3.11) (push) Waiting to run
Continuous Integration / A2A v0.3 Tests (Python 3.12) (push) Waiting to run
Continuous Integration / A2A v0.3 Tests (Python 3.13) (push) Waiting to run
Continuous Integration / A2A v0.3 Tests (Python 3.14) (push) Waiting to run
Copybara PR Handler / close-imported-pr (push) Waiting to run
47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
# Copyright 2026 Google LLC
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from google.adk.features import FeatureName
|
|
from google.adk.features._feature_registry import temporary_feature_override
|
|
from google.adk.tools.load_memory_tool import load_memory_tool
|
|
from google.genai import types
|
|
|
|
|
|
def test_get_declaration_with_json_schema_feature_disabled():
|
|
"""Test that _get_declaration uses parameters when feature is disabled."""
|
|
with temporary_feature_override(FeatureName.JSON_SCHEMA_FOR_FUNC_DECL, False):
|
|
declaration = load_memory_tool._get_declaration()
|
|
|
|
assert declaration.name == 'load_memory'
|
|
assert declaration.parameters_json_schema is None
|
|
assert isinstance(declaration.parameters, types.Schema)
|
|
assert declaration.parameters.type == types.Type.OBJECT
|
|
assert 'query' in declaration.parameters.properties
|
|
|
|
|
|
def test_get_declaration_with_json_schema_feature_enabled():
|
|
"""Test that _get_declaration uses parameters_json_schema when feature is enabled."""
|
|
with temporary_feature_override(FeatureName.JSON_SCHEMA_FOR_FUNC_DECL, True):
|
|
declaration = load_memory_tool._get_declaration()
|
|
|
|
assert declaration.name == 'load_memory'
|
|
assert declaration.parameters is None
|
|
assert declaration.parameters_json_schema == {
|
|
'type': 'object',
|
|
'properties': {
|
|
'query': {'type': 'string'},
|
|
},
|
|
'required': ['query'],
|
|
}
|