Files
AiDA_Python/app/service/fashion_agent/init_llm.py

32 lines
596 B
Python

import os
from dotenv import load_dotenv
from langchain_qwq import ChatQwen
load_dotenv()
QWEN_API_KEY_INTL = os.environ.get("QWEN_API_KEY_INTL", "")
def build_llm(enable_thinking: bool = False):
llm = ChatQwen(
model="qwen3.6-plus",
timeout=None,
max_retries=2,
enable_thinking=enable_thinking,
streaming=True,
api_key=QWEN_API_KEY_INTL,
)
return llm
qwen_plus_llm = ChatQwen(
model="qwen-plus",
timeout=None,
max_retries=2,
streaming=False,
temperature=0.25,
top_p=0.8,
api_key=QWEN_API_KEY_INTL,
)