PlanOpticon
| 0981a08… | noreply | 1 | """Fireworks AI 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 FireworksProvider(OpenAICompatibleProvider): |
| 0981a08… | noreply | 7 | """Fireworks AI API provider (OpenAI-compatible).""" |
| 0981a08… | noreply | 8 | |
| 0981a08… | noreply | 9 | provider_name = "fireworks" |
| 0981a08… | noreply | 10 | base_url = "https://api.fireworks.ai/inference/v1" |
| 0981a08… | noreply | 11 | env_var = "FIREWORKS_API_KEY" |
| 0981a08… | noreply | 12 | |
| 0981a08… | noreply | 13 | |
| 0981a08… | noreply | 14 | ProviderRegistry.register( |
| 0981a08… | noreply | 15 | name="fireworks", |
| 0981a08… | noreply | 16 | provider_class=FireworksProvider, |
| 0981a08… | noreply | 17 | env_var="FIREWORKS_API_KEY", |
| 0981a08… | noreply | 18 | model_prefixes=["accounts/fireworks/"], |
| 0981a08… | noreply | 19 | default_models={ |
| 0981a08… | noreply | 20 | "chat": "accounts/fireworks/models/llama-v3p1-70b-instruct", |
| 0981a08… | noreply | 21 | "vision": "", |
| 0981a08… | noreply | 22 | "audio": "", |
| 0981a08… | noreply | 23 | }, |
| 0981a08… | noreply | 24 | ) |