mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-03 02:22:24 +00:00
Address review feedback for Parasail provider
- Add provider_endpoints_support.json entry (fixes the code-quality provider-documentation check). - Expand supported OpenAI params (user, seed, parallel_tool_calls, stream_options, max_completion_tokens). - Make the provider-info test hermetic by patching get_secret_str instead of os.environ.
This commit is contained in:
parent
3cc2bf5a15
commit
7cb49268df
3 changed files with 30 additions and 1 deletions
|
|
@ -40,10 +40,13 @@ class ParasailConfig(OpenAIGPTConfig):
|
||||||
def get_supported_openai_params(self, model: str) -> list:
|
def get_supported_openai_params(self, model: str) -> list:
|
||||||
return [
|
return [
|
||||||
"max_tokens",
|
"max_tokens",
|
||||||
|
"max_completion_tokens",
|
||||||
"n",
|
"n",
|
||||||
"temperature",
|
"temperature",
|
||||||
"top_p",
|
"top_p",
|
||||||
|
"seed",
|
||||||
"stream",
|
"stream",
|
||||||
|
"stream_options",
|
||||||
"logprobs",
|
"logprobs",
|
||||||
"top_logprobs",
|
"top_logprobs",
|
||||||
"frequency_penalty",
|
"frequency_penalty",
|
||||||
|
|
@ -53,4 +56,6 @@ class ParasailConfig(OpenAIGPTConfig):
|
||||||
"logit_bias",
|
"logit_bias",
|
||||||
"tools",
|
"tools",
|
||||||
"tool_choice",
|
"tool_choice",
|
||||||
|
"parallel_tool_calls",
|
||||||
|
"user",
|
||||||
]
|
]
|
||||||
|
|
|
||||||
|
|
@ -32,6 +32,24 @@
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"providers": {
|
"providers": {
|
||||||
|
"parasail": {
|
||||||
|
"display_name": "Parasail (`parasail`)",
|
||||||
|
"url": "https://docs.litellm.ai/docs/providers/parasail",
|
||||||
|
"endpoints": {
|
||||||
|
"chat_completions": true,
|
||||||
|
"messages": true,
|
||||||
|
"responses": true,
|
||||||
|
"embeddings": false,
|
||||||
|
"image_generations": false,
|
||||||
|
"audio_transcriptions": false,
|
||||||
|
"audio_speech": false,
|
||||||
|
"moderations": false,
|
||||||
|
"batches": false,
|
||||||
|
"rerank": false,
|
||||||
|
"a2a": true,
|
||||||
|
"interactions": true
|
||||||
|
}
|
||||||
|
},
|
||||||
"a2a": {
|
"a2a": {
|
||||||
"display_name": "A2A (Agent-to-Agent) (`a2a`)",
|
"display_name": "A2A (Agent-to-Agent) (`a2a`)",
|
||||||
"url": "https://docs.litellm.ai/docs/providers/a2a",
|
"url": "https://docs.litellm.ai/docs/providers/a2a",
|
||||||
|
|
|
||||||
|
|
@ -51,7 +51,13 @@ class TestParasailConfig:
|
||||||
assert ParasailConfig.API_BASE_URL == "https://api.parasail.io/v1"
|
assert ParasailConfig.API_BASE_URL == "https://api.parasail.io/v1"
|
||||||
|
|
||||||
def test_get_openai_compatible_provider_info(self):
|
def test_get_openai_compatible_provider_info(self):
|
||||||
with patch.dict(os.environ, {"PARASAIL_API_KEY": "sk-secret"}, clear=False):
|
def fake_secret(name, *args, **kwargs):
|
||||||
|
return "sk-secret" if name == "PARASAIL_API_KEY" else None
|
||||||
|
|
||||||
|
with patch(
|
||||||
|
"litellm.llms.parasail.chat.transformation.get_secret_str",
|
||||||
|
side_effect=fake_secret,
|
||||||
|
):
|
||||||
api_base, api_key = self.config._get_openai_compatible_provider_info(
|
api_base, api_key = self.config._get_openai_compatible_provider_info(
|
||||||
api_base=None, api_key=None
|
api_base=None, api_key=None
|
||||||
)
|
)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue