Skip to content

Commit 361fb67

Browse files
committed
feat: wire Plugin interceptors for skill and prompt tailoring
1 parent 3ec1dbe commit 361fb67

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

src/google/adk_community/plugins/taxonomy/taxonomy_plugin.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,12 @@ def __init__(
5858
super().__init__(name)
5959
self.taxonomy_registry = taxonomy_registry or TaxonomyRegistry()
6060
self.resolver = resolver
61-
self.policy = policy
61+
self.policy = policy or DefaultSkillPolicy(self.taxonomy_registry)
62+
if self.policy and getattr(self.policy, "registry", None) is None:
63+
try:
64+
self.policy.registry = self.taxonomy_registry
65+
except Exception:
66+
pass
6267

6368
async def before_model_callback(
6469
self, *, callback_context: CallbackContext, llm_request: LlmRequest

0 commit comments

Comments
 (0)