mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-25 09:08:46 +00:00
Ishaan Jaff
ef7f8cce93
* TestPerplexityWebSearch * use supports_web_search * Update tests/test_litellm/llms/perplexity/test_perplexity.py Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> --------- Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
26 lines
810 B
Python
26 lines
810 B
Python
import os
|
|
import sys
|
|
|
|
sys.path.insert(0, os.path.abspath("../../.."))
|
|
|
|
import pytest
|
|
|
|
|
|
class TestPerplexityWebSearch:
|
|
"""Test suite for Perplexity web search functionality."""
|
|
|
|
@pytest.mark.parametrize(
|
|
"model",
|
|
["perplexity/sonar", "perplexity/sonar-pro"]
|
|
)
|
|
def test_web_search_options_in_supported_params(self, model):
|
|
"""
|
|
Test that web_search_options is in the list of supported parameters for Perplexity sonar models
|
|
"""
|
|
from litellm.llms.perplexity.chat.transformation import PerplexityChatConfig
|
|
|
|
config = PerplexityChatConfig()
|
|
supported_params = config.get_supported_openai_params(model=model)
|
|
|
|
assert "web_search_options" in supported_params, f"web_search_options should be supported for {model}"
|