[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

llm/climate_negotiation/app.py

@@ -4,12 +4,11 @@
 import matplotlib.pyplot as plt
 import pandas as pd
 import solara
+from climate_negotiation.agents import CountryAgent
+from climate_negotiation.model import ClimateNegotiationModel
 from dotenv import load_dotenv
 from mesa.visualization import SolaraViz, make_plot_component
 from mesa.visualization.utils import update_counter
-
-from climate_negotiation.agents import CountryAgent
-from climate_negotiation.model import ClimateNegotiationModel
 from mesa_llm.reasoning.react import ReActReasoning
 
 warnings.filterwarnings("ignore", category=UserWarning, module="pydantic.main")
@@ -78,7 +77,10 @@ def PledgeBarChart(model):
                 bar.get_x() + bar.get_width() / 2,
                 bar.get_height() + 1.2,
                 f"{pledge:.0f}%",
-                ha="center", va="bottom", fontsize=9, fontweight="bold",
+                ha="center",
+                va="bottom",
+                fontsize=9,
+                fontweight="bold",
             )
 
     plt.tight_layout()
@@ -105,13 +107,15 @@ def CoalitionStatusPanel(model):
     rows = []
     for a in sorted(countries, key=lambda x: x.country_name):
         coalition = [id_to_name.get(i, str(i)) for i in a.coalition_members]
-        rows.append({
-            "Country": a.country_name,
-            "Pledge": f"{a.current_pledge:.1f}%",
-            "Accepted": "✓" if a.accepted_treaty else "—",
-            "Coalition": ", ".join(coalition) or "—",
-            "Proposals": a.proposals_made,
-        })
+        rows.append(
+            {
+                "Country": a.country_name,
+                "Pledge": f"{a.current_pledge:.1f}%",
+                "Accepted": "✓" if a.accepted_treaty else "—",
+                "Coalition": ", ".join(coalition) or "—",
+                "Proposals": a.proposals_made,
+            }
+        )
 
     solara.DataFrame(pd.DataFrame(rows))
 
@@ -133,9 +137,7 @@ def PledgeTrajectoriesChart(model):
         return solara.FigureMatplotlib(fig)
 
     id_to_name = {
-        a.unique_id: a.country_name
-        for a in model.agents
-        if isinstance(a, CountryAgent)
+        a.unique_id: a.country_name for a in model.agents if isinstance(a, CountryAgent)
     }
 
     if isinstance(df.index, pd.MultiIndex):
@@ -146,7 +148,9 @@ def PledgeTrajectoriesChart(model):
         return solara.FigureMatplotlib(fig)
 
     for country in pledge_df.columns:
-        ax.plot(pledge_df.index, pledge_df[country], marker="o", label=country, linewidth=2)
+        ax.plot(
+            pledge_df.index, pledge_df[country], marker="o", label=country, linewidth=2
+        )
 
     ax.set_xlabel("Round", fontsize=11)
     ax.set_ylabel("Reduction Pledge (%)", fontsize=11)

llm/climate_negotiation/climate_negotiation/agents.py

@@ -39,7 +39,9 @@ def get_negotiation_history(agent, max_messages: int = 6) -> str:
                     sender_agent = next(
                         a for a in agent.model.agents if a.unique_id == sender_id
                     )
-                    sender_name = getattr(sender_agent, "country_name", f"Agent {sender_id}")
+                    sender_name = getattr(
+                        sender_agent, "country_name", f"Agent {sender_id}"
+                    )
                 except StopIteration:
                     sender_name = f"Agent {sender_id}"
                 messages.append(f"  {sender_name}: {msg}")

llm/climate_negotiation/climate_negotiation/model.py

@@ -1,16 +1,19 @@
 import logging
 import os
+
 from mesa.datacollection import DataCollector
 from mesa.model import Model
+from mesa_llm.reasoning.react import ReActReasoning
+from mesa_llm.reasoning.reasoning import Reasoning
 from rich import print as rprint
 
 from .agents import CountryAgent
-from mesa_llm.reasoning.react import ReActReasoning
-from mesa_llm.reasoning.reasoning import Reasoning
 
 _log_path = os.environ.get("CLIMATE_LOG_FILE", "climate_negotiation.log")
 _file_handler = logging.FileHandler(_log_path, mode="w", encoding="utf-8")
-_file_handler.setFormatter(logging.Formatter("%(asctime)s | %(levelname)s | %(message)s"))
+_file_handler.setFormatter(
+    logging.Formatter("%(asctime)s | %(levelname)s | %(message)s")
+)
 
 sim_logger = logging.getLogger("climate_negotiation")
 sim_logger.setLevel(logging.DEBUG)
@@ -162,7 +165,6 @@ def __init__(
             },
         )
 
-
     def _treaty_reached(self) -> bool:
         """Return True when at least 2/3 of countries have accepted."""
         agents = list(self.agents)
@@ -182,18 +184,12 @@ def _largest_coalition(self) -> int:
         """Size (including self) of the largest active coalition."""
         if not self.agents:
             return 0
-        return max(
-            len(getattr(a, "coalition_members", [])) + 1 for a in self.agents
-        )
-
+        return max(len(getattr(a, "coalition_members", [])) + 1 for a in self.agents)
 
     def step(self):
         self.datacollector.collect(self)
         round_num = self.steps
-        rprint(
-            f"\n[bold cyan]- Climate Summit  Round {round_num} "
-            f"[/bold cyan]"
-        )
+        rprint(f"\n[bold cyan]- Climate Summit  Round {round_num} [/bold cyan]")
         sim_logger.info("=" * 60)
         sim_logger.info(f"ROUND {round_num} START")
         sim_logger.info("=" * 60)

llm/climate_negotiation/climate_negotiation/tools.py

@@ -11,9 +11,10 @@
 import logging
 from typing import TYPE_CHECKING
 
-from .agents import country_tool_manager
 from mesa_llm.tools.tool_decorator import tool
 
+from .agents import country_tool_manager
+
 if TYPE_CHECKING:
     from mesa_llm.llm_agent import LLMAgent
 
@@ -138,10 +139,13 @@ def form_coalition(
     """
     if isinstance(partner_ids, str):
         import json
+
         try:
             partner_ids = json.loads(partner_ids)
         except (json.JSONDecodeError, ValueError):
-            partner_ids = [int(x.strip()) for x in partner_ids.strip("[]").split(",") if x.strip()]
+            partner_ids = [
+                int(x.strip()) for x in partner_ids.strip("[]").split(",") if x.strip()
+            ]
     partner_ids = [int(pid) for pid in (partner_ids or [])]
 
     # Filter out hallucinated IDs - only keep IDs that map to real agents.
@@ -168,10 +172,9 @@ def form_coalition(
         ):
             partner.coalition_members.append(agent.unique_id)
 
-    member_names = (
-        [getattr(a, "country_name", str(a.unique_id)) for a in partner_agents]
-        + [agent.country_name]
-    )
+    member_names = [
+        getattr(a, "country_name", str(a.unique_id)) for a in partner_agents
+    ] + [agent.country_name]
 
     agent.send_message(
         f"[COALITION] {agent.country_name} proposes the '{coalition_name}'. "
@@ -207,11 +210,11 @@ def reject_and_counter(
     """
     counter_reduction_percent = float(counter_reduction_percent or 20.0)
     proposer_id = int(proposer_id or 0)
-    proposer = next(
-        (a for a in agent.model.agents if a.unique_id == proposer_id), None
-    )
+    proposer = next((a for a in agent.model.agents if a.unique_id == proposer_id), None)
     proposer_name = (
-        getattr(proposer, "country_name", str(proposer_id)) if proposer else str(proposer_id)
+        getattr(proposer, "country_name", str(proposer_id))
+        if proposer
+        else str(proposer_id)
     )
 
     counter_msg = (