34 lines
658 B
Python
34 lines
658 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,
|
|
)
|
|
# response = qwen_plus_llm.invoke("你好")
|
|
# print(response)
|