copywriting_production/utils/llm.py

import os
from dotenv import load_dotenv
load_dotenv()
from langchain.chat_models import init_chat_model
from langchain_community.cache import SQLiteCache
# from langchain.globals import set_llm_cache
# set_llm_cache(SQLiteCache(database_path="output/.langchain.db"))

def get_llm_model(model_name: str = 'GLM-4.5-AirX', model_provider="openai"):
    """初始化LangGraph支持的LLM模型"""
    try:
        model = init_chat_model(
            model=model_name,
            model_provider=model_provider,
            base_url=os.getenv("LITELLM_API_BASE"),
            api_key=os.getenv("LITELLM_API_KEY"),
            temperature=0.1,
            # response_format={"type": "json_object"}
        )
        return model
    except Exception as e:
        print(f"❌ 模型初始化失败: {e}")
        return None