From afcd362e819a53122f6e830f6d6ec0ec8816bcde Mon Sep 17 00:00:00 2001
From: marko-kraemer <markokraemer.mail@gmail.com>
Date: Tue, 19 Nov 2024 04:02:41 +0100
Subject: [PATCH] fix

---
 agentpress/agents/simple_web_dev/tools/files_tool.py | 2 +-
 agentpress/cli.py                                    | 7 ++++++-
 pyproject.toml                                       | 2 +-
 3 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/agentpress/agents/simple_web_dev/tools/files_tool.py b/agentpress/agents/simple_web_dev/tools/files_tool.py
index d87ae9b..5dfed56 100644
--- a/agentpress/agents/simple_web_dev/tools/files_tool.py
+++ b/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
 
diff --git a/agentpress/cli.py b/agentpress/cli.py
index 24ee991..9f8a21d 100644
--- a/agentpress/cli.py
+++ b/agentpress/cli.py
@@ -46,9 +46,14 @@
         "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."
     }
 }
 
diff --git a/pyproject.toml b/pyproject.toml
index 9e95ae7..9f0285f 100644
--- a/pyproject.toml
+++ b/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"