Commit 18113f6
Changed files (2)
src
src/llm/models.py
@@ -65,15 +65,13 @@ def get_model_config_with_contexts(model_type: str, contexts: list[dict]) -> dic
completions |= tools
completions |= openrouter_hook(client["base_url"])
- config = {
+ return {
"friendly_name": model_names[model_type],
"bot_msg_prefix": f"🤖**{model_names[model_type]}**: ({BOT_TIPS})",
"client": client,
"completions": completions,
}
- return config
-
def openrouter_hook(base_url: str) -> dict:
"""Add special parameters for OpenRouter."""
src/llm/tool_scheme.py
@@ -45,10 +45,7 @@ def remove_tool(params: dict, tool_name: str) -> dict:
Returns: list[dict]
"""
- keep_tools = []
- for tool in params.get("tools", []):
- if tool.get("function", {}).get("name") != tool_name:
- keep_tools.append(tool)
+ keep_tools = [tool for tool in params.get("tools", []) if tool.get("function", {}).get("name") != tool_name]
if keep_tools:
params["tools"] = keep_tools