mirror of
https://github.com/BerriAI/litellm.git
synced 2026-11-01 07:03:02 +00:00
fix(websearch): type the chat patch builder kwargs instead of working around them
This commit is contained in:
parent
b1a14ff0de
commit
cca1943b6e
1 changed files with 2 additions and 2 deletions
|
|
@ -1826,7 +1826,7 @@ class WebSearchInterceptionLogger(CustomLogger):
|
|||
messages: list[dict],
|
||||
tool_calls: list[dict],
|
||||
optional_params: dict,
|
||||
kwargs: dict,
|
||||
kwargs: dict[str, object],
|
||||
response_format: str = "openai",
|
||||
) -> AgenticLoopRequestPatch:
|
||||
"""Execute litellm.search() and build chat-completion rerun patch."""
|
||||
|
|
@ -1912,7 +1912,7 @@ class WebSearchInterceptionLogger(CustomLogger):
|
|||
k: v for k, v in kwargs.items() if not k.startswith("_websearch_interception") and k not in internal_params
|
||||
}
|
||||
|
||||
custom_llm_provider: Final = kwargs["custom_llm_provider"] if "custom_llm_provider" in kwargs else ""
|
||||
custom_llm_provider: Final = kwargs.get("custom_llm_provider")
|
||||
full_model_name: Final = qualify_provider_stripped_model(
|
||||
model, custom_llm_provider if isinstance(custom_llm_provider, str) else ""
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue