Skip to content
3 changes: 2 additions & 1 deletion qwen_agent/llm/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from .azure import TextChatAtAzure
from .base import LLM_REGISTRY, BaseChatModel, ModelServiceError
from .oai import TextChatAtOAI
from .openvino import OpenVINO
from .openvino import OpenVINO, OpenVINOGenAI
from .transformers_llm import Transformers
from .qwen_dashscope import QwenChatAtDS
from .qwenaudio_dashscope import QwenAudioChatAtDS
Expand Down Expand Up @@ -109,6 +109,7 @@ def get_chat_model(cfg: Union[dict, str] = 'qwen-plus') -> BaseChatModel:
'QwenAudioChatAtDS',
'QwenOmniChatAtOAI',
'OpenVINO',
'OpenVINOGenAI',
'Transformers',
'get_chat_model',
'ModelServiceError',
Expand Down
Loading