lei_y601/sqlbot_agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

去掉错误引用

Browse Source
This commit is contained in:
yujj128
2025-10-15 10:28:34 +08:00
parent d869507c37
commit cfc694b821
3 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
.env
View File

@@ -1,8 +1,17 @@
IS_FIRST_LOAD=True IS_FIRST_LOAD=True
#CHAT_MODEL_BASE_URL=https://api.siliconflow.cn
#CHAT_MODEL_API_KEY=sk-iyhiltycmrfnhrnbljsgqjrinhbztwdplyvuhfihcdlepole
#CHAT_MODEL_NAME=zai-org/GLM-4.5
#CHAT_MODEL_BASE_URL=http://10.226.128.221:8000/v1
#CHAT_MODEL_API_KEY=sk-iyhiltycmrfnhrnbljsgqjrinhbztwdplyvuhfihcdlepole
#CHAT_MODEL_NAME=deepseek-70b
#CHAT_MODEL_BASE_URL=https://dashscope.aliyuncs.com/compatible-mode/v1
#CHAT_MODEL_API_KEY=sk-72575159d3ec43a68c6e222a15719bed
#CHAT_MODEL_NAME=qwen-plus
CHAT_MODEL_BASE_URL=https://api.siliconflow.cn CHAT_MODEL_BASE_URL=https://api.siliconflow.cn
CHAT_MODEL_API_KEY=sk-iyhiltycmrfnhrnbljsgqjrinhbztwdplyvuhfihcdlepole CHAT_MODEL_API_KEY=sk-tnmkzvzbipohjfbqxhictewzdgrrxoghbmicrfjgxbgdkjfq
CHAT_MODEL_NAME=zai-org/GLM-4.5 CHAT_MODEL_NAME=Qwen/Qwen3-32B
#使用ai中台的模型 #使用ai中台的模型
EMBEDDING_MODEL_BASE_URL=http://10.225.128.2:13206/member1/small-model/bge/encode EMBEDDING_MODEL_BASE_URL=http://10.225.128.2:13206/member1/small-model/bge/encode
EMBEDDING_MODEL_API_KEY=sk-iyhiltycmrfnhrnbljsgqjrinhbztwdplyvuhfihcdlepole EMBEDDING_MODEL_API_KEY=sk-iyhiltycmrfnhrnbljsgqjrinhbztwdplyvuhfihcdlepole
@@ -32,5 +41,11 @@ MYSQL_DATABASE_DBNAME=test
#达梦数据库 #达梦数据库
DAMENG_DATABASE_HOST=10.254.193.63 DAMENG_DATABASE_HOST=10.254.193.63
DAMENG_DATABASE_PORT=5236 DAMENG_DATABASE_PORT=5236
DAMENG_DATABASE_PASSWORD= DAMENG_DATABASE_PASSWORD=Ai@0923!!
DAMENG_DATABASE_USER=ai_view DAMENG_DATABASE_USER=ai_view
TTL_CACHE=43200
SESSION_LENGTH=2
ALLOWED_USERS=ea17e939-10f1-492d-adf1-e6ac42f89d81,5fba57a1-0d5f-4988-bcc9-a2b340f8adf1,25bb2ddb-d26d-486f-9aab-5497a9f1e10e,793b49e9-98e6-4f15-8b56-148a691d3022,c76720b5-15f7-4d23-a213-19da38b64d89,103bdc74-a88c-4e1e-a379-794937443c77,c4b2e802-58ba-4927-82fe-f938deb41cf0,1cf59aed-bbf0-4f1c-9246-1d96cc5e2719,6b28c87e-061d-4aca-9a12-71c52e677e73,4c105cfa-ce70-4eca-8d36-c518eda89005,cc93a83b-b145-43f4-80af-fe4ce8a55a04

9
main_service.py
View File

@@ -1,18 +1,11 @@
import copy import copy
from email.policy import default
import logging import logging
from functools import wraps from functools import wraps
from Demos.mmapfile_demo import page_size
import util.utils
from logging_config import LOGGING_CONFIG
from service.cus_vanna_srevice import CustomVanna, QdrantClient, TTLCacheWrapper from service.cus_vanna_srevice import CustomVanna, QdrantClient, TTLCacheWrapper
from decouple import config from decouple import config
import flask import flask
from util import load_ddl_doc from util import load_ddl_doc
from flask import Flask, Response, jsonify, request
from flask import Flask, Response, jsonify, request, send_from_directory
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)

2
util/train_ddl.py
View File

@@ -149,7 +149,7 @@ person_database_ddl = """
"type": "VARCHAR(108)", "type": "VARCHAR(108)",
"comment": "内部单位ID", "comment": "内部单位ID",
"role": "dimension", "role": "dimension",
"tags": ["组织信息", "内部架构"] "tags": ["组织信息", "内部架构ID"]
}, },
{ {
"name": "person_type", "name": "person_type",