PlanOpticon
| 0981a08… | noreply | 1 | """Cerebras provider implementation.""" |
| 0981a08… | noreply | 2 | |
| 0981a08… | noreply | 3 | from video_processor.providers.base import OpenAICompatibleProvider, ProviderRegistry |
| 0981a08… | noreply | 4 | |
| 0981a08… | noreply | 5 | |
| 0981a08… | noreply | 6 | class CerebrasProvider(OpenAICompatibleProvider): |
| 0981a08… | noreply | 7 | """Cerebras AI API provider (OpenAI-compatible).""" |
| 0981a08… | noreply | 8 | |
| 0981a08… | noreply | 9 | provider_name = "cerebras" |
| 0981a08… | noreply | 10 | base_url = "https://api.cerebras.ai/v1" |
| 0981a08… | noreply | 11 | env_var = "CEREBRAS_API_KEY" |
| 0981a08… | noreply | 12 | |
| 0981a08… | noreply | 13 | |
| 0981a08… | noreply | 14 | ProviderRegistry.register( |
| 0981a08… | noreply | 15 | name="cerebras", |
| 0981a08… | noreply | 16 | provider_class=CerebrasProvider, |
| 0981a08… | noreply | 17 | env_var="CEREBRAS_API_KEY", |
| 0981a08… | noreply | 18 | model_prefixes=["cerebras/"], |
| 0981a08… | noreply | 19 | default_models={"chat": "llama3.1-70b", "vision": "", "audio": ""}, |
| 0981a08… | noreply | 20 | ) |