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