0.1.4 version

Author: XGenerationLab

Diff for: README.md

@@ -44,6 +44,8 @@
     - [Text-to-SQL SOTA model](#text-to-sql-sota-model)
     - [Local Model](#local-model)
   - [Database Configuration](#database-configuration)
+    - [MySQL](#mysql)
+    - [PostgreSQL](#postgresql)
 - [Launch](#launch)
   - [Claude Desktop](#claude-desktop)
   - [Cline](#cline)
@@ -57,14 +59,15 @@
 - 🌐 Fetch data by natural language through [XiYanSQL](https://github.com/XGenerationLab/XiYan-SQL)
 - 🤖 Support general LLMs (GPT,qwenmax), Text-to-SQL SOTA model
 - 💻 Support pure local mode (high security!)
-- 🖱️ List available MySQL tables as resources
+- 📝 Support MySQL and PostgreSQL.
+- 🖱️ List available tables as resources
 - 🔧 Read table contents
 
 ## Tool Preview
  - The tool ``get_data`` provides a natural language interface for retrieving data from a database. This server will convert the input natural language into SQL using a built-in model and call the database to return the query results.
 
- - The ``mysql://{table_name}`` resource allows obtaining a portion of sample data from the database for model reference when a specific table_name is specified.
-- The ``mysql://`` resource will list the names of the current databases
+ - The ``{dialect}://{table_name}`` resource allows obtaining a portion of sample data from the database for model reference when a specific table_name is specified. 
+- The ``{dialect}://`` resource will list the names of the current databases
 
 ## Installation
 ### Installing from pip
@@ -224,10 +227,11 @@ model:
 Til now the local mode is ready.
 
 ### Database Configuration
-``host``, ``port``, ``user``, ``password``, ``database`` are the connection information of the MySQL database.
+``host``, ``port``, ``user``, ``password``, ``database`` are the connection information of the database.
 
-You can use local or any remote databases. Now we support MySQL (more dialects soon).
+You can use local or any remote databases. Now we support MySQL and PostgreSQL(more dialects soon).
 
+#### MySQL
 
 ```yaml
 database:
@@ -237,7 +241,23 @@ database:
   password: ""
   database: ""
 ```
+#### PostgreSQL
+step1: Install python packages
+```bash
+pip install psycopg2
+```
+step2: prepare the config.yml like this:
+```yaml
+database:
+  dialect: "postgresql"
+  host: "localhost"
+  port: 5432
+  user: ""
+  password: ""
+  database: ""
+```
 
+Note that ``dialect`` should be ``postgresql`` for postgresql.
 ## Launch
 ### Claude desktop
 Add this in your claude desktop config file, ref <a href="https://github.com/XGenerationLab/xiyan_mcp_server/blob/main/imgs/claude_desktop.jpg">claude desktop config example</a>

Diff for: pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "xiyan_mcp_server"
-version = "0.1.3"
+version = "0.1.4"
 description = "A Model Context Protocol (MCP) server that utilizes XiyanSQL with databases. This server enables AI assistants to list tables, read data, and execute natural language queries"
 readme = "README.md"
 requires-python = ">=3.11"

Diff for: src/xiyan_mcp_server/server.py

@@ -38,7 +38,8 @@ def get_yml_config():
 
 
 def get_xiyan_config(db_config):
-    xiyan_db_config = DBConfig(dialect='mysql',db_name=db_config['database'], user_name=db_config['user'], db_pwd=db_config['password'], db_host=db_config['host'], port=db_config['port'])
+    dialect = db_config.get('dialect','mysql')
+    xiyan_db_config = DBConfig(dialect=dialect,db_name=db_config['database'], user_name=db_config['user'], db_pwd=db_config['password'], db_host=db_config['host'], port=db_config['port'])
     return xiyan_db_config
 
 
@@ -47,17 +48,17 @@ def get_xiyan_config(db_config):
 model_config = global_config['model']
 global_db_config = global_config['database']
 global_xiyan_db_config = get_xiyan_config(global_db_config)
+dialect = global_db_config.get('dialect','mysql')
+#print("dialect is !!!!"+dialect)
 
-
-
-@mcp.resource('mysql://'+global_db_config['database'])
+@mcp.resource(dialect+'://'+global_db_config['database'])
 async def read_resource() -> str:
 
     db_engine = init_db_conn(global_xiyan_db_config)
     db_source = HITLSQLDatabase(db_engine)
     return db_source.mschema.to_mschema()
 
-@mcp.resource("mysql://{table_name}")
+@mcp.resource(dialect+"://{table_name}")
 async def read_resource(table_name) -> str:
     """Read table contents."""
     config = global_db_config

Diff for: src/xiyan_mcp_server/utils/db_config.py

@@ -17,7 +17,7 @@ def __post_init__(self):
             self.db_path = self.db_path or 'book_1.sqlite'
         elif self.dialect in ['mysql', 'postgresql']:
             self.db_name = self.db_name or 'default_db'
-            self.user_name = self.user_name or 'default_user'
+            self.user_name = quote_plus(self.user_name) or 'default_user'
             self.db_pwd = quote_plus(self.db_pwd) or 'default_password'
             self.db_host = self.db_host or 'localhost'
             self.port = self.port or (3306 if self.dialect == 'mysql' else 5432)