feat: support for tree model

.gitignore

@@ -18,6 +18,8 @@ uv.lock
 # IDE settings (optional)
 .vscode/
 .idea/
+.specstory/
+.cursor/
 
 # Distribution directories
 *.dist-info/

README.md

@@ -14,17 +14,30 @@ The server doesn't expose any resources.
 The server doesn't provide any prompts.
 
 ### Tools
-The server offers three core tools:
+The server offers different tools for IoTDB Tree Model and Table Model. You can choose between them by setting the "IOTDB_SQL_DIALECT" configuration to either "tree" or "table".
 
-#### Query Tools
+#### Tree Model
+- `metadata_query`
+   - Execute SHOW/COUNT queries to read metadata from the database
+   - Input:
+     - `query_sql` (string): The SHOW/COUNT SQL query to execute 
+   - Returns: Query results as array of objects
+- `select_query`
+   - Execute SELECT queries to read data from the database
+   - Input:
+     - `query_sql` (string): The SELECT SQL query to execute
+   - Returns: Query results as array of objects
+
+#### Table Model
+
+##### Query Tools
 - `read_query`
    - Execute SELECT queries to read data from the database
    - Input:
      - `query` (string): The SELECT SQL query to execute
    - Returns: Query results as array of objects
 
-
-#### Schema Tools
+##### Schema Tools
 - `list_tables`
    - Get a list of all tables in the database
    - No input required
@@ -37,7 +50,6 @@ The server offers three core tools:
    - Returns: Array of column definitions with names and types
 
 
-
 ## Claude Desktop Integration
 
 ## Prerequisites
@@ -90,7 +102,8 @@ Location: `%APPDATA%/Claude/claude_desktop_config.json`
         "IOTDB_PORT": "6667",
         "IOTDB_USER": "root",
         "IOTDB_PASSWORD": "root",
-        "IOTDB_DATABASE": "test"
+        "IOTDB_DATABASE": "test",
+        "IOTDB_SQL_DIALECT": "table"
       }
     }
   }

src/iotdb_mcp_server/config.py

@@ -49,7 +49,12 @@ class Config:
 
     database: str
     """
-    IoTDB database name
+    IoTDB database
+    """
+
+    sql_dialect: str
+    """
+    SQL dialect: tree or table
     """
 
     @staticmethod
@@ -73,13 +78,6 @@ def from_env_arguments() -> "Config":
             default=os.getenv("IOTDB_PORT", 6667),
         )
 
-        parser.add_argument(
-            "--database",
-            type=str,
-            help="IoTDB connect database name",
-            default=os.getenv("IOTDB_DATABASE", "test"),
-        )
-
         parser.add_argument(
             "--user",
             type=str,
@@ -94,11 +92,26 @@ def from_env_arguments() -> "Config":
             default=os.getenv("IOTDB_PASSWORD", "root"),
         )
 
+        parser.add_argument(
+            "--database",
+            type=str,
+            help="IoTDB connect database name",
+            default=os.getenv("IOTDB_DATABASE", "test"),
+        )
+
+        parser.add_argument(
+            "--sql-dialect",
+            type=str,
+            help="SQL dialect: tree or table",
+            default=os.getenv("IOTDB_SQL_DIALECT", "table"),
+        )
+
         args = parser.parse_args()
         return Config(
             host=args.host,
             port=args.port,
-            database=args.database,
             user=args.user,
             password=args.password,
+            database=args.database,
+            sql_dialect=args.sql_dialect,
         )