github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-03 10:12:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

lint + ignore

Browse Source
This commit is contained in:
Mason Daugherty 2025-07-16 12:07:53 -04:00
parent 73eab00c8b
commit 31b6392651
No known key found for this signature in database
2 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
libs/partners/perplexity/langchain_perplexity/chat_models.py
View File

@ -51,38 +51,38 @@ def _is_pydantic_class(obj: Any) -> bool:
def _create_usage_metadata(token_usage: dict) -> tuple[UsageMetadata, dict[str, Any]]: def _create_usage_metadata(token_usage: dict) -> tuple[UsageMetadata, dict[str, Any]]:
"""Create usage metadata and non-token metadata from token usage data. """Create usage metadata and non-token metadata from token usage data.
Returns a tuple of (UsageMetadata, response_metadata_dict). Returns a tuple of (UsageMetadata, response_metadata_dict).
""" """
input_tokens = token_usage.get("prompt_tokens", 0) input_tokens = token_usage.get("prompt_tokens", 0)
output_tokens = token_usage.get("completion_tokens", 0) output_tokens = token_usage.get("completion_tokens", 0)
total_tokens = token_usage.get("total_tokens", input_tokens + output_tokens) total_tokens = token_usage.get("total_tokens", input_tokens + output_tokens)
output_token_details: OutputTokenDetails = {} output_token_details: OutputTokenDetails = {}
if "reasoning_tokens" in token_usage: if "reasoning_tokens" in token_usage:
output_token_details["reasoning"] = token_usage["reasoning_tokens"] output_token_details["reasoning"] = token_usage["reasoning_tokens"]
if "citation_tokens" in token_usage: if "citation_tokens" in token_usage:
output_token_details["citation"] = token_usage["citation_tokens"] output_token_details["citation"] = token_usage["citation_tokens"] # type: ignore[typeddict-unknown-key]
usage_metadata_dict = { usage_metadata_dict = {
"input_tokens": input_tokens, "input_tokens": input_tokens,
"output_tokens": output_tokens, "output_tokens": output_tokens,
"total_tokens": total_tokens, "total_tokens": total_tokens,
} }
if output_token_details: if output_token_details:
usage_metadata_dict["output_token_details"] = output_token_details usage_metadata_dict["output_token_details"] = output_token_details
usage_metadata = UsageMetadata(**usage_metadata_dict) usage_metadata = UsageMetadata(**usage_metadata_dict) # type: ignore[typeddict-item]
response_metadata = {} response_metadata = {}
if "num_search_queries" in token_usage: if "num_search_queries" in token_usage:
response_metadata["num_search_queries"] = token_usage["num_search_queries"] response_metadata["num_search_queries"] = token_usage["num_search_queries"]
if "search_context_size" in token_usage: if "search_context_size" in token_usage:
response_metadata["search_context_size"] = token_usage["search_context_size"] response_metadata["search_context_size"] = token_usage["search_context_size"]
return usage_metadata, response_metadata return usage_metadata, response_metadata
@ -362,7 +362,7 @@ class ChatPerplexity(BaseChatModel):
if (model_name := chunk.get("model")) and not added_model_name: if (model_name := chunk.get("model")) and not added_model_name:
generation_info["model_name"] = model_name generation_info["model_name"] = model_name
added_model_name = True added_model_name = True
# Add usage response metadata to generation info # Add usage response metadata to generation info
if usage_response_metadata: if usage_response_metadata:
generation_info.update(usage_response_metadata) generation_info.update(usage_response_metadata)

2
libs/partners/perplexity/pyproject.toml
View File

@ -60,7 +60,7 @@ target-version = "py39"
[tool.ruff.lint] [tool.ruff.lint]
select = ["E", "F", "I", "T201", "UP", "S"] select = ["E", "F", "I", "T201", "UP", "S"]
ignore = [ "UP007", ] ignore = [ "UP007", "UP045"]
[tool.ruff.format] [tool.ruff.format]
docstring-code-format = true docstring-code-format = true