tx_flow_analysis/config.py

# SQL血缘解析数据库配置
sql_parse_db_config = {
    "db_file": "/Applications/work/宇信科技/WMXT/wmxt.duckdb",
    # "db_file": "/Applications/work/宇信科技/XMGJ/sql_parse.duckdb",
    "parse_task_concurrency": 10, # 解析任务并发度
}

model_name="qwen-max-latest"
# coder_model_name="qwen2.5-coder-32b-instruct"
#coder_model_name="qwen2.5-coder-7b-instruct"

coder_model_name="qwen3-coder-plus"
# model_name="qwen3-32b"
# model_name="qwq-32b"
# model_name="qwen2.5-72b-instruct"

# model_name = "deepseek-r1-distill-qwen-32b"

# LLM config
llm_config = {
    "temperature": 0,
    "model": model_name,
    "base_url": "http://103.154.31.78:20001/compatible-mode/v1",
    "api_key": "IfWZqv1F55hsRWlWB2fqCca91VK9IMET",
    "coder_model": coder_model_name
}

# 210.12.198.141:20010


# 词根 rag config
word_root_config = {
    "file_path": "/Users/jiaqiang/Downloads/词根V1.0.xls",
    "index_name": "words_root_idx",
    "jieba_dict": "/Applications/work/宇信科技/智能数据平台/word_root_dict.txt"
}

term_mapping_config = {
    "index_name": "term_mapping_idx",
    "file_path": "/Applications/work/宇信科技/智能数据平台/业务术语映射.xls"
}

# schema rag config
schema_handler_config = {
    "index_name": "schema"
}

# check agent config
check_agent_config = {
    "dialect": "hive"
}

# comment agent config
comment_agent_config = {
    "dialect": "hive",
    "top_k": 10,
    "similarity_threshold": 0.6
}

explain_agent_config = {
    "top_k": 10,
    "similarity_threshold": 0.6
}

linage_agent_config = {
    "dialect": "hive",
    "concurrency": 5,
    "parse_log_file": "/Applications/work/宇信科技/WMXT/SP/parse_time.txt"
}

script_agent_config = {
    "dialect": "hive",
    "rag_schema_topk": 20,
    "rag_schema_similarity": 0.3
}

# 逻辑模型生成config
model_agent_config = {
    "db_dialect": "hive",
    "session_expire_time": 1800,
    "clean_interval": 600,
    "similarity_threshold": 0.4,
    "jieba_dict": "/Applications/work/宇信科技/智能数据平台/word_root_dict.txt"
}

# 业务问题转 sql config
biz_query_agent_config = {
    "db_dialect": "hive",
    "similarity_threshold": 0.4
}

# 数据库方言SQL转换 config
transformer_agent_config = {

}


gen_biz_mapping_agent_config = {

}


qwen32_app_key = "eyJ0eXAiOiJKV1QiLCJhbGciOiJSUzI1NiJ9.eyJzdWIiOiJhZGU0YmRhODc4YzA0ZGVmYTk0NWY5ODFiNTQ5MjUxMyIsImlzcyI6ImFwaS1hdXRoLWtleSIsImV4cCI6NDkwNzU0Njk0Nn0.jCa-XVPspH3GzXuU-sm1SZJuWALgSrFI2QvQbunP50SIiBvlhjIdy7BwZ2oPsqBMoidJx0ujUlo-TGBc6Ea86yjn2J54QQb4pyKZ-r2r6II4qbk9PktqWDO6D9m2M2uide2eiYyDqIkHlYcsl6vmjw0w9I-KZntQrwrJSHgGJF5x_cqDB2iJkH41lDjMxOhNjwRIfj317U5PCoUCDK3GGRxTHwm2Eg1PWDJT1IP7AUnBa6o404ilHBR3o-bCOW2jWY8dAF67ogr8ApnT91TzCuQWrUdQ6IP3PuGiLSjedC-85bDtvgAML_APCguUf6J7K3RpwcnEIcM9u9fznOCJTw"
qwen32_url="https://ai.sinochem.com/kunlun/ingress/api-safe/5351ea/e4079068806f428b960c0537164cc5a2/ai-fd44da9a49ea4cfc8696e20948160fb8/service-180c8450621e444e81638c9faa545875/v1"

llm_qwen32 = {
    "name": "qwencoder7",
    "app-key": qwen32_app_key,
    "url": qwen32_url
}

coder7_app_key="eyJ0eXAiOiJKV1QiLCJhbGciOiJSUzI1NiJ9.eyJzdWIiOiI4MzY1ZmE3YzEyOTM0OGM5OTA0YTY5Mzk4NGY1ZTFiOCIsImlzcyI6ImFwaS1hdXRoLWtleSIsImV4cCI6NDkwNDUxOTc4MX0.hpO5psDAqlxnzCFSKynghiuxfiair0rkCqulhNTXOd0ZyW1acjqEWq_AyjNv6Zgce2cGenNxx9tkVYPEuGydaiCnQBOGgPjXSoKx5VNa98gxGio1ohQ8O2Rqv7-uarIGow206ROLKhmTLVs48mlk8UpQQUBwBmn9iFliREKhvVBbUNNCMyNPlDJHoh8_pQ38vQmF6MI4xu6loJh-3a6gO_pIWF025KANmoth-IgvlXjjR-1QDpuTnxt2oi_AjLy4rlWd8q9NLCWddBcHN08JlU3Yl87Vb-rHhexPMdOdkOzAZa-7hnAorrxomktLXpNH0j0VgbBBb-Vnved90F5MSg"
coder7_url="https://ai.sinochem.com/kunlun/ingress/api-safe/5351ea/e4079068806f428b960c0537164cc5a2/ai-6d7b1fff09994a0198093a43980583ea/service-7cb26a8f786140009073a608535f0c95/v1"

llm_coder7 = {
    "name": "qwencoder7",
    "app-key": coder7_app_key,
    "url": coder7_url
}


linage_agent_config = {
    "dialect": "hive",
    "concurrency": 12,
    "parse_log_file": "/home/appuser/parse_time.txt"
}

# 解析客户端配置
sql_parse_client_config = {
    "parse_url": "http://127.0.0.1:3699/api/sqllineage/parse",
    "export_table_url": "http://127.0.0.1:3699/api/sqllineage/table/export",
    "export_col_url": "http://127.0.0.1:3699/api/sqllineage/table/col/export"
}


# 交易流水分析配置
tx_flow_analysis_config = {
    "llm": model_name
}