fix: ruff format compliance

Author: Patel230

src/hawk/context_window.py

@@ -1,4 +1,3 @@
-
 from .types import Message
 
 
@@ -28,7 +27,7 @@ def compact_messages(self, messages: list[Message], system_prompt: str = "") ->
         # Traverse backwards (keep newest context)
         for msg in reversed(messages):
             # Extract content from the message model
-            raw = msg.content if isinstance(msg, Message) else str(msg.get('content', ''))
+            raw = msg.content if isinstance(msg, Message) else str(msg.get("content", ""))
             content = raw or ""
             msg_tokens = self.estimate_tokens(content)
 

src/hawk/graph_dsl.py

@@ -12,6 +12,7 @@
 
 class NodeType(str, Enum):
     """Types of nodes in orchestration graphs."""
+
     AGENT = "agent"
     TOOL = "tool"
     FUNCTION = "function"
@@ -23,6 +24,7 @@ class NodeType(str, Enum):
 @dataclass
 class GraphNode:
     """Represents a node in the graph."""
+
     id: str
     type: NodeType
     name: str
@@ -33,13 +35,14 @@ def to_dict(self) -> dict[str, Any]:
             "id": self.id,
             "type": self.type.value,
             "name": self.name,
-            "properties": self.properties
+            "properties": self.properties,
         }
 
 
 @dataclass
 class GraphEdge:
     """Represents an edge in the graph."""
+
     source: str
     target: str
     condition: Optional[str] = None
@@ -50,7 +53,7 @@ def to_dict(self) -> dict[str, Any]:
             "source": self.source,
             "target": self.target,
             "condition": self.condition,
-            "weight": self.weight
+            "weight": self.weight,
         }
 
 
@@ -77,7 +80,9 @@ def __init__(self) -> None:
         self._edges: list[GraphEdge] = []
         self._adj: dict[str, list[str]] = {}
 
-    def node(self, node_id: str, node_type: NodeType = NodeType.AGENT, name: str = "", **properties: Any) -> "GraphQuery":
+    def node(
+        self, node_id: str, node_type: NodeType = NodeType.AGENT, name: str = "", **properties: Any
+    ) -> "GraphQuery":
         """Add a node to the graph."""
         if node_id in self._nodes:
             return self
@@ -86,7 +91,9 @@ def node(self, node_id: str, node_type: NodeType = NodeType.AGENT, name: str = "
         self._adj[node_id] = []
         return self
 
-    def edge(self, source: str, target: str, condition: Optional[str] = None, weight: float = 1.0) -> "GraphQuery":
+    def edge(
+        self, source: str, target: str, condition: Optional[str] = None, weight: float = 1.0
+    ) -> "GraphQuery":
         """Add an edge between two nodes."""
         if source not in self._nodes or target not in self._nodes:
             raise ValueError(f"Node not found: {source if source not in self._nodes else target}")
@@ -221,7 +228,7 @@ def to_dict(self) -> dict[str, Any]:
         """Export graph as a dictionary."""
         return {
             "nodes": [n.to_dict() for n in self._nodes.values()],
-            "edges": [e.to_dict() for e in self._edges]
+            "edges": [e.to_dict() for e in self._edges],
         }
 
     def __len__(self) -> int:

src/hawk/harness.py

@@ -4,6 +4,7 @@
 
 logger = logging.getLogger(__name__)
 
+
 class TestHarness:
     """Harness Engineering: Wraps agent executions for testing and validation."""
 
@@ -22,6 +23,7 @@ def add_mock(self, tool_name: str, return_value: Any) -> "TestHarness":
 
     def wrap_async(self, func: Callable[..., Any]) -> Callable[..., Any]:
         """Wraps an async function with harness validations."""
+
         @functools.wraps(func)
         async def wrapper(*args: Any, **kwargs: Any) -> Any:
             logger.info(f"[Harness {self.name}] Starting execution")

tests/test_context_window.py

@@ -3,13 +3,13 @@
 
 
 def test_context_optimizer_compaction():
-    optimizer = ContextOptimizer(max_tokens=20) # Approx 80 chars
+    optimizer = ContextOptimizer(max_tokens=20)  # Approx 80 chars
 
     # 3 messages, ~30 chars each
     messages = [
         Message(role="user", content="This is message number one."),
         Message(role="user", content="This is message number two."),
-        Message(role="user", content="This is message number three.")
+        Message(role="user", content="This is message number three."),
     ]
 
     # "Sys" = 3 chars = 0 tokens (3//4). Available = 20

tests/test_harness.py

@@ -15,6 +15,7 @@ async def my_agent():
     result = await my_agent()
     assert result == 42
 
+
 @pytest.mark.asyncio
 async def test_harness_validator_failure():
     harness = TestHarness("test")

tests/test_tools.py

@@ -175,9 +175,7 @@ def test_executes_tool_and_loops(self) -> None:
         assert second_call_kwargs["tool_results"][0].content == "hi"
 
     def test_unknown_tool_returns_error_result(self) -> None:
-        call_resp = _chat_response(
-            tool_calls=[{"id": "tc-1", "name": "missing", "arguments": {}}]
-        )
+        call_resp = _chat_response(tool_calls=[{"id": "tc-1", "name": "missing", "arguments": {}}])
         final_resp = _chat_response(tool_calls=None)
 
         client = MagicMock()
@@ -193,9 +191,7 @@ def test_max_rounds_exceeded_raises(self) -> None:
         """If the daemon keeps returning tool_calls, the loop must not spin
         forever — it raises HawkError after max_rounds (matches TS SDK)."""
         tool = Tool(name="spin", description="never stops", fn=lambda: "again")
-        call_resp = _chat_response(
-            tool_calls=[{"id": "tc-1", "name": "spin", "arguments": {}}]
-        )
+        call_resp = _chat_response(tool_calls=[{"id": "tc-1", "name": "spin", "arguments": {}}])
 
         client = MagicMock()
         client.chat.return_value = call_resp
@@ -224,9 +220,7 @@ async def test_async_loops_and_returns_final(self) -> None:
     @pytest.mark.asyncio
     async def test_async_max_rounds_exceeded_raises(self) -> None:
         tool = Tool(name="spin", description="never stops", fn=lambda: "again")
-        call_resp = _chat_response(
-            tool_calls=[{"id": "tc-1", "name": "spin", "arguments": {}}]
-        )
+        call_resp = _chat_response(tool_calls=[{"id": "tc-1", "name": "spin", "arguments": {}}])
 
         client = MagicMock()
         client.chat = AsyncMock(return_value=call_resp)