fix fireworks

This commit is contained in:
Xi Yan 2024-11-12 12:09:23 -05:00
parent ec4fcad5ca
commit dc0fa5afa9
2 changed files with 7 additions and 0 deletions

View file

@ -4,9 +4,15 @@
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from pydantic import BaseModel
from .config import FireworksImplConfig
class FireworksProviderDataValidator(BaseModel):
fireworks_api_key: str
async def get_adapter_impl(config: FireworksImplConfig, _deps):
from .fireworks import FireworksInferenceAdapter