Skip to content

Commit 2386ca0

Browse files
refactor(xiyan_mcp_server): 优化配置文件加载路径并修复 SQL 代码块格式问题
- 设置配置文件路径为当前目录下的 config_demo.yml,解决无配置文件问题 - 修复了 SQL代码块的 Markdown 格式问题,保证正确显示 - 新增 __main__.py 文件,实现包的直接运行功能
1 parent 8650b66 commit 2386ca0

File tree

3 files changed

+6
-2
lines changed

3 files changed

+6
-2
lines changed

src/xiyan_mcp_server/__main__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
from . import main
2+
3+
if __name__ == "__main__":
4+
main()
File renamed without changes.

src/xiyan_mcp_server/server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
logger = logging.getLogger("xiyan_mcp_server")
2424

2525
def get_yml_config():
26-
config_path = os.getenv("YML","")
26+
config_path = os.getenv("YML", os.path.join(os.path.dirname(__file__), "config_demo.yml"))
2727
try:
2828
with open(config_path, 'r') as file:
2929
config = yaml.safe_load(file)
@@ -82,7 +82,7 @@ def sql_gen_and_execute(db_env, query: str):
8282
"""
8383

8484
#db_env = context_variables.get('db_env', None)
85-
prompt = f"""你现在是一名{db_env.dialect}数据分析专家,你的任务是根据参考的数据库schema和用户的问题,编写正确的SQL来回答用户的问题,生成的SQL用```sql 和```包围起来。
85+
prompt = f"""你现在是一名{db_env.dialect}数据分析专家,你的任务是根据参考的数据库schema和用户的问题,编写正确的SQL来回答用户的问题,生成的SQL用``sql 和```包围起来。
8686
【数据库schema】
8787
{db_env.mschema_str}
8888

0 commit comments

Comments
 (0)