mirror of
https://github.com/tiennm99/litellm.git
synced 2026-07-12 21:04:10 +00:00
Ishaan Jaffer
25 lines
796 B
Python
25 lines
796 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}"
|