diff --git a/lang_agent/utils.py b/lang_agent/utils.py new file mode 100644 index 0000000..9adf7ec --- /dev/null +++ b/lang_agent/utils.py @@ -0,0 +1,21 @@ +from langchain.chat_models import init_chat_model +from langchain_core.language_models import BaseChatModel + +import os +from dotenv import load_dotenv +load_dotenv() + +def make_llm(model="qwen-plus", + model_provider="openai", + api_key=None, + base_url="https://dashscope.aliyuncs.com/compatible-mode/v1", + **kwargs)->BaseChatModel: + api_key = os.environ.get("ALI_API_KEY") if api_key is None else api_key + + llm = init_chat_model(model=model, + model_provider=model_provider, + api_key=api_key, + base_url=base_url, + **kwargs) + + return llm \ No newline at end of file