fix: correct type annotation

Overridden `supported_models` should be a `classmethod` rather than a `staticmethod`.

PiperOrigin-RevId: 790989895
This commit is contained in:
Google Team Member
2025-08-04 19:25:52 -07:00
committed by Copybara-Service
parent 283303032a
commit 97318bcd19
4 changed files with 9 additions and 8 deletions
+3 -2
View File
@@ -282,8 +282,9 @@ class MockModel(BaseLlm):
return cls(responses=responses)
@staticmethod
def supported_models() -> list[str]:
@classmethod
@override
def supported_models(cls) -> list[str]:
return ['mock']
def generate_content(