初始化项目:langgraph 学习笔记与 lg_common 通用组件包

This commit is contained in:
gqt
2026-08-16 00:25:21 +08:00
commit b3f3b72f07
29 changed files with 25395 additions and 0 deletions
+6
View File
@@ -0,0 +1,6 @@
from lg_common.env import load_env
from lg_common.llm import get_llm
load_env()
__all__ = ["get_llm"]
+14
View File
@@ -0,0 +1,14 @@
"""环境变量加载。"""
from pathlib import Path
from dotenv import load_dotenv
# lg_common 位于项目根目录下,所以根目录是它的上一级
_PROJECT_ROOT = Path(__file__).resolve().parent.parent
_DEFAULT_ENV_FILE = _PROJECT_ROOT / ".env"
def load_env() -> None:
"""加载项目根目录下的 .env,覆盖已有同名环境变量。"""
load_dotenv(dotenv_path=_DEFAULT_ENV_FILE, override=True)
+17
View File
@@ -0,0 +1,17 @@
import os
from langchain.chat_models import init_chat_model
def get_llm(
model: str = "openai:qwen3.6-flash",
base_url: str | None = None,
api_key: str | None = None,
**kwargs,
):
"""初始化 ChatModel,默认从环境变量读取 BASE_URL / API_KEY。"""
if base_url is None:
base_url = os.getenv("BASE_URL")
if api_key is None:
api_key = os.getenv("API_KEY")
return init_chat_model(model=model, base_url=base_url, api_key=api_key, **kwargs)