fix

agentpress/agents/simple_web_dev/tools/files_tool.py

@@ -1,7 +1,7 @@
 import os
 import asyncio
 from pathlib import Path
-from agentpress.tools.tool import Tool, ToolResult, openapi_schema, xml_schema
+from agentpress.tool import Tool, ToolResult, openapi_schema, xml_schema
 from agentpress.state_manager import StateManager
 from typing import Optional
 

agentpress/cli.py

@@ -46,9 +46,14 @@ MODULES = {
         "description": "Conversation Management System - Handles message threading, conversation history, and provides a UI for viewing conversation threads. Manages the flow of messages between the user, LLM, and tools."
     },
     "state_management": {
-        "required": False,
+        "required": True,
         "files": ["state_manager.py"],
         "description": "State Persistence System - Provides thread-safe storage and retrieval of conversation state, tool data, and other persistent information. Enables maintaining context across sessions and managing shared state between components."
+    },
+    "db_connection": {
+        "required": True,
+        "files": ["db_connection.py"],
+        "description": "Database Connection - Provides a connection to a SQLite database for storing and retrieving conversation state, tool data, and other persistent information."
     }
 }
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agentpress"
-version = "0.1.9"
+version = "0.1.10"
 description = "Building blocks for AI Agents"
 authors = ["marko-kraemer <mail@markokraemer.com>"]
 readme = "README.md"