File tree Expand file tree Collapse file tree 4 files changed +9
-8
lines changed Expand file tree Collapse file tree 4 files changed +9
-8
lines changed Original file line number Diff line number Diff line change @@ -249,9 +249,9 @@ class Claude(BaseLlm):
249
249
model : str = "claude-3-5-sonnet-v2@20241022"
250
250
max_tokens : int = 8192
251
251
252
- @staticmethod
252
+ @classmethod
253
253
@override
254
- def supported_models () -> list [str ]:
254
+ def supported_models (cls ) -> list [str ]:
255
255
return [r"claude-3-.*" , r"claude-.*-4.*" ]
256
256
257
257
@override
Original file line number Diff line number Diff line change @@ -75,9 +75,9 @@ class Gemini(BaseLlm):
75
75
```
76
76
"""
77
77
78
- @staticmethod
78
+ @classmethod
79
79
@override
80
- def supported_models () -> list [str ]:
80
+ def supported_models (cls ) -> list [str ]:
81
81
"""Provides the list of supported models.
82
82
83
83
Returns:
Original file line number Diff line number Diff line change @@ -848,9 +848,9 @@ async def generate_content_async(
848
848
response = await self .llm_client .acompletion (** completion_args )
849
849
yield _model_response_to_generate_content_response (response )
850
850
851
- @staticmethod
851
+ @classmethod
852
852
@override
853
- def supported_models () -> list [str ]:
853
+ def supported_models (cls ) -> list [str ]:
854
854
"""Provides the list of supported models.
855
855
856
856
LiteLlm supports all models supported by litellm. We do not keep track of
Original file line number Diff line number Diff line change @@ -282,8 +282,9 @@ def create(
282
282
283
283
return cls (responses = responses )
284
284
285
- @staticmethod
286
- def supported_models () -> list [str ]:
285
+ @classmethod
286
+ @override
287
+ def supported_models (cls ) -> list [str ]:
287
288
return ['mock' ]
288
289
289
290
def generate_content (
You can’t perform that action at this time.
0 commit comments