CrewAI集成国内大语言模型参考范例

397 阅读2分钟

本文的方法支持openai三方框架的模型皆可使用 注:这里的openai指的并非公司名字 而是框架库名字 基本上国内的模型都支持了这个库 可以去对应官网找到模型名称 前缀加上open/字眼即可进行调用

通用部分

进入crewai官网Quickstart - CrewAI按照官方指示进行顺序操作

本文作者文章风格实用简洁 直接把安装代码放到下面

pip install uv
uv tool install crewai
crewai create crew <此处去掉尖括号替换成实际项目名称>

提示选择模型提供商随机选择即可 不影响后续代码执行 执行完上面代码会创建对应名称的文件夹 文件结构如下:

my_project/
├── .gitignore
├── knowledge/
├── pyproject.toml
├── README.md
├── .env
└── src/
 └── my_project/
     ├── __init__.py
     ├── main.py
     ├── crew.py
     ├── tools/
     │   ├── custom_tool.py
     │   └── __init__.py
     └── config/
         ├── agents.yaml
         └── tasks.yaml

本文旨在快速入门测试 跑通用例 不探讨各个文件含义

非通用部分

注:差异部分仅仅在于不通模型官网的模型名称不同 使用方法一致

以智谱清言为例

首先去官方网站(www.bigmodel.cn)找到所需要的对应模型 并生成保存自己的api key 并在生成的crew.py文件头部加上

  from crewai import LLM

  glm_4v_flash = LLM(
    model="openai/拟选用的模型名称",  # Specify the model name
    api_key="对应的api key",
    base_url="对应的模型url"
)

剩下的代码只需要修改每个你定义的agent部分的返回值 操作如下:

  return Agent(
            config=self.agents_config['对应的智能体行为 无需修改'],
            llm=glm_4v_flash(这里是上面的创建的crew LLM),  # Use the GLM-4V-Flash model
            verbose=True
        )

全文查询return Agent字段 修改llm=(你的创建的LLM对象名称) 接下来crew run即可运行 控制台输出全绿结果即代表你已入门ai agent领域 迈出了成为agent大师的第一步 超越国内99%的人 让我们一起创建agent 引领人类未来吧