distro codegen

This commit is contained in:
Xi Yan 2025-01-06 15:21:04 -08:00
parent 7a90fc5854
commit 60fca40917
6 changed files with 140 additions and 129 deletions

View file

@ -22,7 +22,7 @@ class FireworksImplConfig(BaseModel):
)
@classmethod
def sample_run_config(cls, __distro_dir__: str) -> Dict[str, Any]:
def sample_run_config(cls, **kwargs) -> Dict[str, Any]:
return {
"url": "https://api.fireworks.ai/inference/v1",
"api_key": "${env.FIREWORKS_API_KEY}",