mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-29 01:42:19 +00:00
b604e2b20c
About 35,000 fixes ruff marks safe across 32 rules (UP006/UP045/UP007 modern annotations, UP032 f-strings, SIM114/SIM118, RET501, and friends), removal of the 1,296 typing imports the rewrite orphaned, and hand fixes for what the fixers could not see: five star-import freeloaders of typing names, two F823 late-import annotations, the /get/config/list introspection crash on types.UnionType, redundant function-local RoleMappings imports in ui_sso.py that shadowed the module-level name once the annotation lost its quotes, and one FURB168 tautology. B009/B010/PIE804/RUF019 are excluded on purpose: their safe fixes rewrite getattr/setattr/**-splat/key-in-dict escape hatches into forms basedpyright then rejects (283 new errors measured), so their budgets stay at base values. ruff-strict-budget.json drops by 39,579 this commit (39,968 across the branch) with 28 rules at an actual 0 and 9 more sharply down. type-discipline-budget.json ratchets LIT002/LIT006/LIT009 down; LIT001 moves to the now-honest total: the checker matches the spelling `set` but not the alias `Set`, so the 160 typing.Set annotations rewritten to set[...] were always mutable-set annotations and only now count.
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
"""
|
|
Build the litellm_content_retrieve tool definition for the LLM.
|
|
"""
|
|
|
|
|
|
def build_retrieval_tool(available_keys: list[str]) -> dict:
|
|
"""
|
|
Return an OpenAI-format tool definition that lets the model
|
|
retrieve the full content of a compressed message.
|
|
"""
|
|
return {
|
|
"type": "function",
|
|
"function": {
|
|
"name": "litellm_content_retrieve",
|
|
"description": (
|
|
"Retrieve the full content of a file or message that was "
|
|
"compressed to save tokens. Use this when you need the complete "
|
|
"content to answer accurately. Available keys: " + ", ".join(available_keys)
|
|
),
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"key": {
|
|
"type": "string",
|
|
"description": "The identifier of the content to retrieve",
|
|
"enum": available_keys,
|
|
}
|
|
},
|
|
"required": ["key"],
|
|
},
|
|
},
|
|
}
|