Refactored and simplified

Author: Knucklessg1

.pre-commit-config.yaml

@@ -27,7 +27,7 @@ repos:
     hooks:
     - id: black
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.15.1
+    rev: v0.15.2
     hooks:
       - id: ruff
         types_or: [ python, pyi, jupyter ]

documentdb_mcp/__init__.py

@@ -8,12 +8,12 @@
 __all__: List[str] = []
 
 CORE_MODULES = [
-    "documentdb_mcp.documentdb_mcp",
+    "documentdb_mcp.mcp",
 ]
 
 OPTIONAL_MODULES = {
-    "documentdb_mcp.documentdb_agent": "a2a",
-    "documentdb_mcp.documentdb_mcp": "mcp",
+    "documentdb_mcp.agent": "agent",
+    "documentdb_mcp.mcp": "mcp",
 }
 
 
@@ -47,12 +47,12 @@ def _expose_members(module):
     else:
         globals()[f"_{extra_name.upper()}_AVAILABLE"] = False
 
-_MCP_AVAILABLE = OPTIONAL_MODULES.get("documentdb_mcp.documentdb_mcp") in [
+_MCP_AVAILABLE = OPTIONAL_MODULES.get("documentdb_mcp.mcp") in [
     m.__name__ for m in globals().values() if hasattr(m, "__name__")
 ]
-_A2A_AVAILABLE = "documentdb_mcp.documentdb_agent" in globals()
+_AGENT_AVAILABLE = "documentdb_mcp.agent" in globals()
 
-__all__.extend(["_MCP_AVAILABLE", "_A2A_AVAILABLE"])
+__all__.extend(["_MCP_AVAILABLE", "_AGENT_AVAILABLE"])
 
 
 """

documentdb_mcp/__main__.py

@@ -1,6 +1,6 @@
 #!/usr/bin/python
 # coding: utf-8
-from documentdb_mcp.documentdb_mcp import documentdb_mcp
+from documentdb_mcp.mcp import documentdb_mcp
 
 if __name__ == "__main__":
     documentdb_mcp()

documentdb_mcp/agent/HEARTBEAT.md

@@ -8,7 +8,10 @@ You are running a scheduled heartbeat. Perform these checks and report results c
 2. **Memory Review** — Read `MEMORY.md` and check for any pending follow-up tasks or action items. List any that are overdue.
 3. **Cron Log** — Read `CRON_LOG.md` and check for recent errors (❌). Summarize any failures from the last 24 hours.
 4. **Peer Agents** — Read `AGENTS.md` and note if any registered peers need attention.
-5. **Self-Diagnostics** — Report your current model, available tool count, and any anomalies.
+5. **Domain-Specific Checks**:
+   - **Database Connectivity**: Verify connection to all registered databases.
+   - **Performance**: Identify any active slow queries or high connection counts.
+6. **Self-Diagnostics** — Report your current model, available tool count, and any anomalies.
 
 ## Response Format
 

documentdb_mcp/agent/IDENTITY.md

@@ -1,18 +1,18 @@
 # IDENTITY.md - Who I Am, Core Personality, & Boundaries
 
- * **Name:** Documentdb Mcp Agent
- * **Role:** A research specialist agent for web search and information gathering using Documentdb Mcp.
- * **Emoji:** 🔍
- * **Vibe:** Precise, objective, concise
+## [default]
+ * **Name:** DocumentDB Agent
+ * **Role:** Management and analysis of DocumentDB/MongoDB databases.
+ * **Emoji:** 🗃️
+ * **Vibe:** Data-driven, technical, methodical
 
- ## System Prompt
- You are a Research Specialist Agent for Documentdb Mcp.
- You have access to a powerful metasearch engine to find information on the web.
- Your responsibilities:
- 1. Analyze the user's research topic or query.
- 2. Use the 'web_search' tool (or related skills) to find relevant information.
- 3. Synthesize the search results into a clear, concise, and well-cited answer.
- 4. Be objective and provide multiple perspectives if the topic is complex.
- 5. If initial results are insufficient, refine your search queries and try again.
- 6. Always include the URLs of the sources you used.
- 7. MEMORY: You have long-term memory in MEMORY.md. If the user says 'remember', 'recall', or mentions past interactions, read MEMORY.md to retrieve context. Save important decisions, outcomes, and user preferences to MEMORY.md using append_note_to_file.
+ ### System Prompt
+ You are the DocumentDB Agent.
+ Your goal is to assist the user with database operations on DocumentDB or MongoDB.
+ You handle collection management, CRUD operations, database analysis, and user management.
+ Help the user interact with their data efficiently and securely.
+ You can:
+ - List and manage databases and collections.
+ - Perform CRUD operations on documents.
+ - Analyze data structures and runs database commands.
+ - Manage database users and permissions.

documentdb_mcp/agent/__init__.py

@@ -2,7 +2,7 @@
 name: test-skill
 description: A test skill.
 version: 0.1.0
-tags: [custom]
+tags: [test-skill]
 input_modes: [text]
 output_modes: [text]
 ---