refactor(fireworks_ai/): inherit from openai like base config (#7146)

* refactor(fireworks_ai/): inherit from openai like base config

refactors fireworks ai to use a common config

* test: fix import in test

* refactor(watsonx/): refactor watsonx to use llm base config

refactors chat + completion routes to base config path

* fix: fix linting error

* test: fix test

* fix: fix test
This commit is contained in:
Krish Dholakia 2024-12-10 16:15:19 -08:00 committed by GitHub
parent 2fb2801eb4
commit 311432ca17
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 449 additions and 307 deletions

View file

@ -9,6 +9,7 @@ from typing import (
Any,
AsyncIterator,
Callable,
Dict,
Iterator,
List,
Optional,
@ -33,7 +34,7 @@ class BaseLLMException(Exception):
self,
status_code: int,
message: str,
headers: Optional[httpx.Headers] = None,
headers: Optional[Union[Dict, httpx.Headers]] = None,
request: Optional[httpx.Request] = None,
response: Optional[httpx.Response] = None,
):