diff --git a/aixplain/modules/team_agent/__init__.py b/aixplain/modules/team_agent/__init__.py index 5215bc8b..08a1fa91 100644 --- a/aixplain/modules/team_agent/__init__.py +++ b/aixplain/modules/team_agent/__init__.py @@ -560,5 +560,9 @@ def update(self) -> None: error_msg = f"Team Agent Update Error (HTTP {r.status_code}): {resp}" raise Exception(error_msg) + def save(self) -> None: + """Save the Agent.""" + self.update() + def __repr__(self): return f"TeamAgent: {self.name} (id={self.id})"