Commit 18113f6

benny-dou <60535774+benny-dou@users.noreply.github.com>
2025-02-10 14:05:07
fix: lint error
1 parent 5f19ecd
Changed files (2)
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