diff --git a/examples/models/openai_compatibility_model_example.py b/examples/models/openai_compatibility_model_examples/nemotron.py similarity index 100% rename from examples/models/openai_compatibility_model_example.py rename to examples/models/openai_compatibility_model_examples/nemotron.py diff --git a/examples/models/openai_compatibility_model_examples/qwen.py b/examples/models/openai_compatibility_model_examples/qwen.py new file mode 100644 index 0000000000..87f0b15f5a --- /dev/null +++ b/examples/models/openai_compatibility_model_examples/qwen.py @@ -0,0 +1,51 @@ +# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. =========== +# Licensed under the Apache License, Version 2.0 (the “License”); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an “AS IS” BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. =========== + +from camel.agents import ChatAgent +from camel.messages import BaseMessage +from camel.models import ModelFactory +from camel.types import ModelPlatformType + +# Take calling model from DashScope as an example +# Refer: https://dashscope.console.aliyun.com/overview +model = ModelFactory.create( + model_platform=ModelPlatformType.OPENAI_COMPATIBILITY_MODEL, + model_type="qwen-plus", + api_key="sk-xxxx", + url="https://dashscope.aliyuncs.com/compatible-mode/v1", + model_config_dict={"temperature": 0.4}, +) + +assistant_sys_msg = BaseMessage.make_assistant_message( + role_name="Assistant", + content="You are a helpful assistant.", +) + +agent = ChatAgent(assistant_sys_msg, model=model, token_limit=4096) + +user_msg = BaseMessage.make_user_message( + role_name="User", + content="""Say hi to CAMEL AI, one open-source community + dedicated to the study of autonomous and communicative agents.""", +) +assistant_response = agent.step(user_msg) +print(assistant_response.msg.content) + +""" +=============================================================================== +Hi to the CAMEL AI community! It's great to connect with an open-source +community focused on the study of autonomous and communicative agents. How can +I assist you or your projects today? +=============================================================================== +"""