ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

TestPerplexityIntegration

Browse Source
This commit is contained in:
Ishaan Jaffer 2025-10-25 17:06:01 -07:00
parent 04ff660276
commit cbadcd4a77
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
litellm/llms/__init__.py
View File

@ -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

10
tests/test_litellm/llms/perplexity/test_perplexity_integration.py
View File

@ -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",