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