开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

想在modelscope-agent里面注册新的tool,比如添加文本分类的API,该怎么操作?

想在modelscope-agent里面注册新的tool,比如添加文本分类的API,该怎么操作?

展开
收起
Lucidly 2024-01-09 19:33:26 45 0
2 条回答
写回答
取消 提交回答
  • 面对过去,不要迷离;面对未来,不必彷徨;活在今天,你只要把自己完全展示给别人看。

    要在ModelScope Agent中注册新的tool,你需要遵循以下步骤:

    1. 创建一个新的Python文件,例如text_classification_tool.py。在这个文件中,定义你的文本分类API函数。确保这个函数接受一个字符串作为输入,并返回一个分类结果。
    def text_classification(text):
        # 在这里实现你的文本分类逻辑
        # 返回分类结果
        pass
    
    1. modelscope-agent的根目录下创建一个名为tools的文件夹。将你刚刚创建的text_classification_tool.py文件移动到这个文件夹中。

    2. 修改modelscope-agent的配置文件(例如config.yaml),以包含你的新tool。在tools部分添加一个新的条目,如下所示:

    tools:
      - name: text_classification
        entrypoint: text_classification_tool:text_classification
        description: A tool for text classification.
    
    1. 保存配置文件并重新启动modelscope-agent。现在,你应该可以在Agent中使用你的新文本分类API了。

    注意:这里的示例代码仅用于演示目的。你需要根据你的实际需求和模型来实现文本分类的逻辑。

    2024-01-13 19:34:45
    赞同 展开评论 打赏
  • 文末有agent大本营链接可以关注,也可以直接参考demo。
    https://developer.aliyun.com/article/1387748
    https://github.com/modelscope/modelscope-agent/blob/master/demo/demo_register_new_tool.ipynb
    此回答整理自钉群“魔搭ModelScope开发者联盟群 ①”

    2024-01-09 23:14:00
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

相关电子书

更多
API 网关实践 立即下载
ACE 区域技术发展峰会:Flink Python Table API入门及实践 立即下载
Java Spring Boot开发实战系列课程【第15讲】:Spring Boot 2.0 API与Spring REST Docs实战 立即下载