Assistant Demos

from qwen_agent.agents import Assistant
from qwen_agent.gui import WebUI

def test():
    bot = Assistant(llm={'model': 'qwen-plus-latest'})
    messages = [{
        'role': 'user',
        'content': [
            {'text': '介绍图一'},
            {'file': 'https://arxiv.org/pdf/1706.03762.pdf'}
        ]
    }]
    for rsp in bot.run(messages):
        print(rsp)

def app_gui():
    # Define the agent
    bot = Assistant(
        llm={'model': 'qwen-plus-latest'},
        name='Assistant',
        description='使用RAG检索并回答，支持文件类型：PDF/Word/PPT/TXT/HTML。'
    )
    chatbot_config = {
        'prompt.suggestions': [
            {'text': '介绍图一'},
            {'text': '第二章第一句话是什么？'},
        ]
    }
    WebUI(bot, chatbot_config=chatbot_config).run()

if __name__ == '__main__':
    app_gui()

from qwen_agent.agents import Assistant
from qwen_agent.gui import WebUI
import os

ROOT_RESOURCE = os.path.join(os.path.dirname(__file__), 'resource')

def init_agent_service():
    llm_cfg = {'model': 'qwen-max'}
    system = (
        '你扮演一个天气预报助手，你具有查询天气和画图能力。'
        '你需要查询相应地区的天气，然后调用给你的画图工具绘制一张城市的图，'
        '并从给定的诗词文档中选一首相关的诗词来描述天气，不要说文档以外的诗词。'
    )

    tools = ['image_gen', 'amap_weather']
    bot = Assistant(
        llm=llm_cfg,
        name='天气预报助手',
        description='查询天气和画图',
        system_message=system,
        function_list=tools,
    )
    return bot

def app_gui():
    bot = init_agent_service()
    chatbot_config = {
        'prompt.suggestions': [
            '查询北京的天气',
            '画一张北京的图片',
            '画一张北京的图片，然后配上一首诗',
        ]
    }
    WebUI(bot, chatbot_config=chatbot_config).run()

if __name__ == '__main__':
    app_gui()

import json
import urllib.parse
import json5
from qwen_agent.agents import Assistant
from qwen_agent.gui import WebUI
from qwen_agent.tools.base import BaseTool, register_tool

# Add a custom tool named my_image_gen
@register_tool('my_image_gen')
class MyImageGen(BaseTool):
    description = (
        'AI painting (image generation) service, input text description, '
        'and return the image URL drawn based on text information.'
    )
    parameters = [{
        'name': 'prompt',
        'type': 'string',
        'description': 'Detailed description of the desired image content, in English',
        'required': True,
    }]

    def call(self, params: str, **kwargs) -> str:
        prompt = json5.loads(params)['prompt']
        prompt = urllib.parse.quote(prompt)
        return json.dumps(
            {'image_url': f'https://image.pollinations.ai/prompt/{prompt}'},
            ensure_ascii=False,
        )

def init_agent_service():
    llm_cfg = {'model': 'qwen-max'}
    system = (
        "According to the user's request, you first draw a picture and then "
        "automatically run code to download the picture and select an image "
        "operation from the given document to process the image"
    )

    tools = ['my_image_gen', 'code_interpreter']
    bot = Assistant(
        llm=llm_cfg,
        name='AI painting',
        description='AI painting service',
        system_message=system,
        function_list=tools,
        files=[os.path.join(ROOT_RESOURCE, 'doc.pdf')],
    )
    return bot

def app_gui():
    bot = init_agent_service()
    chatbot_config = {
        'prompt.suggestions': [
            '画一只猫的图片',
            '画一只可爱的小腊肠狗',
            '画一幅风景画，有湖有山有树',
        ]
    }
    WebUI(bot, chatbot_config=chatbot_config).run()

if __name__ == '__main__':
    app_gui()