From cbadcd4a776aa09b47a6f924a7f1325ed8b6dbdf Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Sat, 25 Oct 2025 17:06:01 -0700 Subject: [PATCH] TestPerplexityIntegration --- litellm/llms/__init__.py | 4 ++++ .../llms/perplexity/test_perplexity_integration.py | 10 +++++----- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/litellm/llms/__init__.py b/litellm/llms/__init__.py index 1676e7bb1e..44e1fb4eb9 100644 --- a/litellm/llms/__init__.py +++ b/litellm/llms/__init__.py @@ -39,6 +39,10 @@ def get_cost_for_web_search_request( ) return cost_per_web_search_request_vertex_ai(usage=usage, model_info=model_info) + elif custom_llm_provider == "perplexity": + # Perplexity handles search costs internally in its own cost calculator + # Return 0.0 to indicate costs are already accounted for + return 0.0 else: return None diff --git a/tests/test_litellm/llms/perplexity/test_perplexity_integration.py b/tests/test_litellm/llms/perplexity/test_perplexity_integration.py index ae72b8a962..a702e9ebc4 100644 --- a/tests/test_litellm/llms/perplexity/test_perplexity_integration.py +++ b/tests/test_litellm/llms/perplexity/test_perplexity_integration.py @@ -20,7 +20,7 @@ import litellm from litellm import ModelResponse from litellm.cost_calculator import completion_cost, cost_per_token from litellm.llms.perplexity.chat.transformation import PerplexityChatConfig -from litellm.types.utils import Usage, PromptTokensDetailsWrapper +from litellm.types.utils import PromptTokensDetailsWrapper, Usage from litellm.utils import get_model_info @@ -48,10 +48,10 @@ class TestPerplexityIntegration: "output_cost_per_token": 8e-06, "output_cost_per_reasoning_token": 3e-06, "citation_cost_per_token": 2e-06, - "search_queries_cost_per_query": { - "search_queries_size_low": 0.005, - "search_queries_size_medium": 0.005, - "search_queries_size_high": 0.005 + "search_context_cost_per_query": { + "search_context_size_low": 0.005, + "search_context_size_medium": 0.005, + "search_context_size_high": 0.005 }, "litellm_provider": "perplexity", "mode": "chat",