'commit'

AI/Config.py

@@ -1,12 +1,12 @@
 from pathlib import Path
 
 # 固定配置项
-# 阿里云中用来调用deepseek r1的密钥
 MODEL_API_KEY = "sk-01d13a39e09844038322108ecdbd1bbc"
 MODEL_API_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1'
 MODEL_NAME = "deepseek-v3"
-QWEN_MODEL_NAME='qwen-plus'
 
+MODEL_GENERATION_TEXT_URL = "https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation" # 阿里云文本生成服务
+QWEN_MODEL_NAME = 'qwen-plus'
 
 # 华为云
 HW_API_KEY = '2R7vrjfIZO1chcfk4V3zDscx2Ms4I4kdoY7XSFnxa8Z2RNfAztg8qrXXKkwJGZWvMtS4qnN782WmaY4xZfMhyg'

AI/Text2Sql/Util/VannaUtil.py

@@ -4,14 +4,13 @@ import requests
 from vanna.base import VannaBase
 from Config import *
 
-class DeepSeekVanna(VannaBase):
+
+class VannaUtil(VannaBase):
     def __init__(self):
         super().__init__()
         self.api_key = MODEL_API_KEY
-        self.base_url = "https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation"  # 阿里云专用API地址
+        self.base_url = MODEL_GENERATION_TEXT_URL  # 阿里云专用API地址
         self.model = QWEN_MODEL_NAME  # 根据实际模型名称调整
-        #self.base_url= MODEL_API_URL,
-        #self.model = MODEL_NAME
         self.training_data = []
         self.chat_history = []
 
@@ -57,8 +56,6 @@ class DeepSeekVanna(VannaBase):
     def get_similar_question_sql(self, question: str, **kwargs) -> List[Dict[str, Any]]:
         return []
 
-
-
     def _clean_sql_output(self, raw_sql: str) -> str:
         """增强版清洗逻辑"""
         # 移除所有非SQL内容

AI/Text2Sql/YunXiao.py

@@ -5,7 +5,7 @@ from openai import OpenAI
 from Text2Sql.Util.MarkdownToDocxUtil import markdown_to_docx
 from Text2Sql.Util.PostgreSQLUtil import PostgreSQLUtil
 from Text2Sql.Util.SaveToExcel import save_to_excel
-from Text2Sql.Util.Text2SqlUtil import *
+from Text2Sql.Util.VannaUtil import *
 
 from Util.EchartsUtil import *
 
@@ -16,7 +16,7 @@ from Util.EchartsUtil import *
 3、应该有类似于 保存为用例，查询历史等功能，让用户方便利旧。
 '''
 if __name__ == "__main__":
-    vn = DeepSeekVanna()
+    vn = VannaUtil()
 
     # 开始训练
     print("开始训练...")