diff --git a/.github/workflows/test-unit.yml b/.github/workflows/test-unit.yml index 6dee2bb605f..1aea375af47 100644 --- a/.github/workflows/test-unit.yml +++ b/.github/workflows/test-unit.yml @@ -340,6 +340,7 @@ jobs: tests/unit/llms/searchapi tests/unit/llms/searxng tests/unit/llms/serper + tests/unit/llms/serpapi tests/unit/llms/snowflake tests/unit/llms/soniox tests/unit/llms/stability diff --git a/litellm/llms/serpapi/search/transformation.py b/litellm/llms/serpapi/search/transformation.py index ed40caba956..63bd409cadc 100644 --- a/litellm/llms/serpapi/search/transformation.py +++ b/litellm/llms/serpapi/search/transformation.py @@ -111,7 +111,7 @@ class SerpApiSearchConfig(BaseSearchConfig): domains: Final = optional_params.get("search_domain_filter") q_final: Final = ( - f"({q_str}) ({" OR ".join(f"site:{d}" for d in domains)})" # pyright: ignore[reportUnknownVariableType] # items are strings + f"({q_str}) ({' OR '.join(f'site:{d}' for d in domains)})" # pyright: ignore[reportUnknownVariableType] # items are strings if isinstance(domains, list) and len(domains) > 0 # pyright: ignore[reportUnknownArgumentType] # domains is a list else q_str ) @@ -135,7 +135,11 @@ class SerpApiSearchConfig(BaseSearchConfig): and param not in optional_mappings } - request_data: Final[dict[str, object]] = {**base_params, **optional_mappings, **extra_params} # mutable-ok: mapping type + request_data: Final[dict[str, object]] = { # mutable-ok: mapping type + **base_params, + **optional_mappings, + **extra_params, + } return {"_serpapi_params": request_data} diff --git a/litellm/utils.py b/litellm/utils.py index 7d215f27017..20dcf41e09a 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -9949,8 +9949,8 @@ class ProviderConfigManager: from litellm.llms.perplexity.search.transformation import PerplexitySearchConfig from litellm.llms.searchapi.search.transformation import SearchAPIConfig from litellm.llms.searxng.search.transformation import SearXNGSearchConfig - from litellm.llms.serper.search.transformation import SerperSearchConfig from litellm.llms.serpapi.search.transformation import SerpApiSearchConfig + from litellm.llms.serper.search.transformation import SerperSearchConfig from litellm.llms.tavily.search.transformation import TavilySearchConfig from litellm.llms.tinyfish.search.transformation import TinyfishSearchConfig from litellm.llms.you_com.search.transformation import YouComSearchConfig