当前位置: 首页 > news >正文

如何统一从一个共享的配置文件(比如 config.yaml 和 config.py)加载配置信息,避免硬编码。

环境:

Ubuntu20.04

问题描述:

如何统一从一个共享的配置文件(比如 config.yaml 和 config.py)加载配置信息,避免硬编码?

在这里插入图片描述

解决方案:

希望本地模型和线上模型两块代码完全分开,但它们都统一从一个**共享的配置文件(比如 config.yamlconfig.py)**加载配置信息,避免硬编码。

提供三份文件示范:

  • config/config.yaml:统一配置文件
  • config/config.py:统一读取配置并做环境变量覆盖
  • local_model.py:独立文件,调用本地模型,配置从config读取
  • online_model.py:独立文件,调用线上模型,配置从config读取

1. core/server/config/config.yaml

# 线上模型配置
api_key: "d3112f3e479"
base_url: "https://ark.cn-beijing.volces.com/api/v3"
model_name: "ep-20250602174215-fw5hx"# 本地模型配置
local_model_api: "http://192.168.1.9:11434/v1/chat/completions"
local_model_name: "qwen2.5-3bnsfwny"
local_model_max_tokens: 512
local_model_temperature: 0.25
local_model_timeout: 10

2. core/server/config/config.py

import os
import yamlCONFIG_PATH = os.path.join(os.path.dirname(__file__), "config.yaml")class Config:def __init__(self):with open(CONFIG_PATH, "r", encoding="utf-8") as f:data = yaml.safe_load(f)for k, v in data.items():env_val = os.getenv(k.upper())setattr(self, k, env_val if env_val is not None else v)config = Config()

3. local_model.py

import requests
from loguru import logger
from typing import Optional
import timefrom core.server.config.config import config  # 根据你的实际包结构调整导入路径def summarize_with_model(text: str) -> Optional[str]:headers = {"Content-Type": "application/json"}payload = {"model": config.local_model_name,"messages": [{"role": "user", "content": text}],"max_tokens": int(config.local_model_max_tokens),"temperature": float(config.local_model_temperature),"stream": False}try:logger.info("调用本地大模型接口,准备发送请求")start_time = time.time()r = requests.post(config.local_model_api, headers=headers, json=payload, timeout=int(config.local_model_timeout))r.raise_for_status()logger.info(f"接口响应状态码:{r.status_code},耗时:{time.time() - start_time:.2f}秒")resp_json = r.json()content = resp_json.get("choices", [{}])[0].get("message", {}).get("content")logger.info(f"模型返回内容长度:{len(content) if content else 0}")logger.info(f"模型返回具体内容:{content}")return contentexcept requests.exceptions.Timeout:logger.error("调用本地大模型失败: 请求超时")return Noneexcept Exception as e:logger.error(f"调用本地大模型失败: {e}")return Noneif __name__ == "__main__":test_text = "请帮我总结一下这段文字的内容。"logger.info("=== 测试本地模型 ===")result = summarize_with_model(test_text)logger.info(f"本地模型返回结果:{result}")

4. online_model.py

from openai import OpenAI
from loguru import logger
from typing import Optionalfrom core.server.config.config import config  # 根据你的实际包结构调整导入路径client = OpenAI(api_key=config.api_key,base_url=config.base_url,
)def summarize_with_model(text: str) -> Optional[str]:try:logger.info("调用线上DeepSeek模型接口,准备发送请求")messages = [{"role": "system", "content": ""},{"role": "user", "content": text},]completion = client.chat.completions.create(model=config.model_name,messages=messages,stream=False)content = completion.choices[0].message.contentlogger.info(f"线上模型返回内容长度:{len(content) if content else 0}")logger.info(f"线上模型返回内容:{content}")return contentexcept Exception as e:logger.error(f"调用线上模型失败: {e}")return Noneif __name__ == "__main__":test_text = "请帮我总结一下这段文字的内容。"logger.info("=== 测试线上模型 ===")result = summarize_with_model(test_text)logger.info(f"线上模型返回结果:{result}")

5. 包结构示例及运行说明

core\server\config\__init__.pyconfig.pyconfig.yamlmodules\llm\__init__.pylocal_model.pyonline_model.py
  • configmodulesllm目录都加空的 __init__.py,使其成为包。
  • 在项目根目录执行:
python -m core.server.modules.llm.local_model
python -m core.server.modules.llm.online_model

确保Python可以正确解析包路径。


6. 说明

  • 配置集中统一,方便维护,避免魔法数字和字符串散落代码。
  • 代码完全分开,互不影响,且都共享同一配置,便于管理。
  • 方便后续增加环境变量覆盖或动态配置读取。

相关文章:

  • 14:00开始面试,14:06就出来了,问的问题有点变态。。。
  • leetcode0765. 情侣牵手-hard
  • 【多线程初阶】详解线程池(上)
  • 什么是旋转开关?
  • 芯片测试之 trim修调测试详解
  • 掌握C#枚举:从交通灯看懂状态管理
  • 【0.3 漫画数据结构与算法】
  • Sngine 4.0.4海外社交平台PHP源码 – 多语言支持短视频和博客订阅(源码下载)
  • pdb/sdf转pdbqt错误、pdbqt小分子文件对接蛋白质dock分数为0
  • [Linux入门] Linux磁盘管理与文件系统
  • Heterogeneous Graph Transformer (WWW 2020)译文
  • Spring学习笔记
  • Java并发编程实战 Day 22:高性能无锁编程技术
  • RT-Thread Studio SDK管理器安装资源包失败
  • uniapp打包安卓和ios
  • 初学STM32编程技巧全功能按键非阻塞式实现
  • Amazon Q in QuickSight 实战:自然语言秒级生成数据报表与深度洞察
  • HTML+CSS 登陆框动态切换
  • 学习Oracle------认识VARCHAR2
  • 量化面试绿皮书:13. 贴错标签的袋子
  • 定西市网站建设咨询/杭州网站建设方案优化
  • 重庆移动网站建设/seo文章是什么
  • wordpress主题 手机/seo技术培训沈阳
  • 小程序开发教程文档/seo网站关键词优化报价
  • 福州高端品牌网站建设/网站开发费用
  • 合川网站建设公司/宁波网站关键词优化代码