Skip to content

Commit b64813f

Browse files
committed
feat: add triggers and variables to TaxonomyTerm model
1 parent 1411e5b commit b64813f

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,8 @@ class TaxonomyTerm(BaseModel):
3131
name: (str)
3232
definition: (Optional[str])
3333
alt_labels: (list[str])
34+
variables: (dict[str, str])
35+
triggers: (list[str])
3436
"""
3537

3638
model_config = ConfigDict(populate_by_name=True)
@@ -40,6 +42,8 @@ class TaxonomyTerm(BaseModel):
4042
name: str
4143
definition: Optional[str] = None
4244
alt_labels: list[str] = Field(default_factory=list, alias="altLabels")
45+
variables: dict[str, str] = Field(default_factory=dict)
46+
triggers: list[str] = Field(default_factory=list)
4347

4448

4549
class TaxonomyRegistry(BaseModel):
@@ -105,12 +109,16 @@ def from_json_ld(cls, data: list[dict]) -> TaxonomyRegistry:
105109
]
106110

107111
broader = item.get("broader")
112+
variables = item.get("variables", {})
113+
triggers = item.get("triggers", [])
108114
term = TaxonomyTerm(
109115
id=term_id,
110116
parent_id=broader,
111117
name=pref_label,
112118
definition=definition,
113119
alt_labels=alt_labels,
120+
variables=variables,
121+
triggers=triggers,
114122
)
115123
terms[term_id] = term
116124
return cls(terms=terms)

0 commit comments

Comments
 (0)