lint

src/lilith_agent/app.py

@@ -15,6 +15,11 @@ from langgraph.graph import END, StateGraph
 from langgraph.graph.message import add_messages
 from typing import Annotated, TypedDict
 
+from lilith_agent.config import Config
+from lilith_agent.checkpointing import build_checkpointer
+from lilith_agent.models import get_cheap_model, get_extra_strong_model
+
+
 class AgentState(TypedDict):
     messages: Annotated[list, add_messages]
     iterations: int
@@ -26,10 +31,6 @@ class AgentState(TypedDict):
     supervisor_review_count: int
 
 
-from lilith_agent.config import Config
-from lilith_agent.checkpointing import build_checkpointer
-from lilith_agent.models import get_cheap_model, get_extra_strong_model
-
 log = logging.getLogger(__name__)
 # Per-node child loggers so the logger-name column reads `lilith_agent.nodes.X`
 # and traces read like the gaia_agent reference output (`[model] invoking ...`,

src/lilith_agent/memory.py

@@ -305,13 +305,6 @@ def summarize_episode(messages: List[BaseMessage], model) -> None:
             if "ERROR" in content.upper():
                 outcome = "failed/struggled"
                 
-        conv_parts = []
-        for m in messages:
-            content = _content_to_text(m.content)
-            if content:
-                conv_parts.append(f"{m.type}: {content[:200]}...")
-        conv_str = "\n".join(conv_parts)
-
         prompt = f"""
 Summarize this task trajectory for Lilith's 'Episodic Memory'.
 Initial Question: {initial_question}

src/lilith_agent/tui.py

@@ -1,14 +1,11 @@
 import sys
 import uuid
 from pathlib import Path
-from dotenv import load_dotenv
-
-env_path = Path(__file__).resolve().parent.parent.parent / ".env"
-load_dotenv(dotenv_path=env_path, override=True)
 
+from dotenv import load_dotenv
 from rich.console import Console
-from rich.panel import Panel
 from rich.markdown import Markdown
+from rich.panel import Panel
 from rich.live import Live
 from rich.spinner import Spinner
 from rich.theme import Theme
@@ -19,6 +16,9 @@ from lilith_agent.config import Config
 from lilith_agent.app import build_react_agent
 from lilith_agent.observability import setup_logging, setup_arize, JsonlTraceCallback
 
+env_path = Path(__file__).resolve().parent.parent.parent / ".env"
+load_dotenv(dotenv_path=env_path, override=True)
+
 custom_theme = Theme({
     "info": "dim cyan",
     "warning": "magenta",

tests/test_memory_pick_up.py

@@ -1,11 +1,12 @@
+import sqlite3
+from datetime import datetime
+
 from lilith_agent.memory import _store, retrieve_relevant_context
 
 # Initialize DB via _store
 _store._init_db()
 
 # Manually inject
-from datetime import datetime
-import sqlite3
 conn = sqlite3.connect(str(_store.db_path))
 with conn:
     conn.execute("DELETE FROM memories")

tests/test_observability.py

@@ -62,7 +62,7 @@ def test_jsonl_callback_captures_chat_model_payload(tmp_path):
     messages = [[SystemMessage(content="sys"), HumanMessage(content="hi")]]
     cb.on_chat_model_start({"name": "ChatAnthropic"}, messages, run_id="r0")
 
-    events = [json.loads(l) for l in (tmp_path / "trace.jsonl").read_text().splitlines()]
+    events = [json.loads(line) for line in (tmp_path / "trace.jsonl").read_text().splitlines()]
     assert events[0]["event"] == "chat_model_start"
     assert events[0]["model"] == "ChatAnthropic"
     assert any(m["type"] == "system" and m["content"] == "sys" for m in events[0]["messages"][0])
@@ -76,7 +76,7 @@ def test_jsonl_callback_captures_chain_boundaries(tmp_path):
     cb.on_chain_start({"name": "agent"}, {"messages": []}, run_id="c1")
     cb.on_chain_end({"messages": ["ok"]}, run_id="c1")
 
-    events = [json.loads(l) for l in (tmp_path / "trace.jsonl").read_text().splitlines()]
+    events = [json.loads(line) for line in (tmp_path / "trace.jsonl").read_text().splitlines()]
     assert events[0]["event"] == "chain_start"
     assert events[0]["name"] == "agent"
     assert events[1]["event"] == "chain_end"
@@ -130,7 +130,7 @@ def test_jsonl_callback_writes_tool_and_llm_events(tmp_path):
 
     lines = path.read_text().strip().splitlines()
     assert len(lines) == 3
-    events = [json.loads(l) for l in lines]
+    events = [json.loads(line) for line in lines]
     assert events[0]["event"] == "tool_start"
     assert events[0]["name"] == "web_search"
     assert events[0]["input"] == "moon perigee"

tests/test_vision.py

@@ -1,12 +1,13 @@
 import os
-from dotenv import load_dotenv
 
-# Load .env before evaluating anything
-load_dotenv(override=True)
+from dotenv import load_dotenv
 
 from lilith_agent.config import Config
 from lilith_agent.tools.vision import inspect_visual_content
 
+# Load .env before evaluating test configuration.
+load_dotenv(override=True)
+
 def test_fal_vision_integration(tmp_path):
     """
     Test the FAL vision integration using GAIA task 8f80e01c-1296-4371-9486-bb3d68651a60.