first commit

2026-04-10 16:52:07 +08:00
commit 84fe545640
87 changed files with 20842 additions and 0 deletions
@@ -0,0 +1,59 @@
+# Text2SQL Agent 环境配置示例
+# 复制为 .env 并填入实际值
+
+# ========== DeepSeek API 配置 ==========
+DEEPSEEK_API_KEY=sk-2e3cc9f79a2c40e9a6b7c960d8c090d9
+DEEPSEEK_BASE_URL=https://api.deepseek.com
+
+# 模型配置
+# 主生成模型
+MODEL_PRIMARY=deepseek-reasoner
+TEMPERATURE=0  # 确定性模式：每次生成相同结果
+MAX_TOKENS=4096
+# 生成与校验默认 SQL Server（T-SQL）；可改为 mysql / postgresql 等
+TEXT2SQL_DIALECT=sqlserver
+
+# ========== Embedding 配置 ==========
+# false = 使用兼容 OpenAI 的远程 /v1/embeddings（优先 ModelScope，未配置 MODELSCOPE_* 时用 DashScope）
+USE_LOCAL_EMBEDDING=false
+# ModelScope 推理 API：https://www.modelscope.cn/docs/inference-api
+# MODELSCOPE_API_KEY=ms-b95f3233-4797-4efd-90e6-b3728743571d
+# MODELSCOPE_BASE_URL=https://api-inference.modelscope.cn/v1
+# MODELSCOPE_EMBEDDING_MODEL=Qwen/Qwen3-Embedding-8B
+# # 单次请求最大条数，按服务端限制调整
+# MODELSCOPE_EMBEDDING_MAX_BATCH=32
+
+# 兼容 OpenAI 的网关：地址含 /v1（若网关根路径即 OpenAI API 则按实际填写）
+OPENAI_BASE_URL=http://113.192.49.54:9080/v1
+OPENAI_API_KEY=sk-proj-FEb2ChHZK5Llm0tBkmNIT3BlbkFJbu0b0zQpTW8762yD6HDv
+OPENAI_EMBEDDING_MODEL=text-embedding-ada-002
+
+ 
+# ========== Schema 配置 ==========
+SCHEMA_DIR=./data/schemas
+SCHEMA_FILE=${SCHEMA_DIR}/G3SB_MCDataDictionary_table_structure.json
+
+# ========== 向量数据库配置 ==========
+VECTOR_DB_PATH=./data/embeddings/chroma
+
+# ========== 验证配置 ==========
+ENABLE_SQL_VALIDATION=true
+BLOCK_DANGEROUS_SQL=true
+
+# ========== 重试配置 ==========
+MAX_RETRY=1  # 确定性模式：只尝试一次，不重试
+RETRY_DELAY=1.0
+
+# ========== Few-shot 配置 ==========
+# 是否启用经验数据集few-shot增强
+FEWSHOT_ENABLED=true
+# 每次使用的示例数量
+FEWSHOT_TOP_K=3
+# 示例最低评分（1-10）
+FEWSHOT_MIN_RATING=7
+# 经验数据集路径
+FEWSHOT_DATA_PATH=./data/experiences/all_samples.jsonl
+
+# ========== 日志配置 ==========
+LOG_LEVEL=INFO
+LOG_FILE=./logs/text2sql.log