openai-agents 中custom example agent

代码

pip show openai-agents

Name: openai-agents
Version: 0.0.4
Summary: OpenAI Agents SDK
Home-page: https://github.com/openai/openai-agents-python
Author: 
Author-email: OpenAI 
License-Expression: MIT
Location: d:\soft\anaconda\envs\oi_agents\Lib\site-packages
Requires: griffe, openai, pydantic, requests, types-requests, typing-extensions
Required-by: 
Note: you may need to restart the kernel to use updated packages.

import asyncio
import os

from openai import AsyncOpenAI

from agents import Agent, OpenAIChatCompletionsModel, Runner, function_tool, set_tracing_disabled

BASE_URL = "https://open.bigmodel.cn/api/paas/v4/"
API_KEY = "your api key"
MODEL_NAME = "glm-4-flash"

if not BASE_URL or not API_KEY or not MODEL_NAME:
    raise ValueError(
        "Please set EXAMPLE_BASE_URL, EXAMPLE_API_KEY, EXAMPLE_MODEL_NAME via env var or code."
    )

client = AsyncOpenAI(base_url=BASE_URL, api_key=API_KEY)
set_tracing_disabled(disabled=True)

@function_tool
def get_weather(city: str):
    print(f"[debug] getting weather for {city}")
    return f"The weather in {city} is sunny."


async def main():
    # This agent will use the custom LLM provider
    agent = Agent(
        name="Assistant",
        instructions="You only respond in haikus.",
        model=OpenAIChatCompletionsModel(model=MODEL_NAME, openai_client=client),
        tools=[get_weather],
    )

    result = await Runner.run(agent, "What's the weather in Tokyo?")
    print(result.final_output)

await main()

[debug] getting weather for Tokyo
Sunny skies in Tokyo.

代码解释

1. 环境信息展示：

pip show openai-agents

这个命令显示了当前安装的 openai-agents 包的信息，包括版本号(0.0.4)、依赖项等。
版本号(0.0.3)不可用

2. 主要代码结构：

• 导入必要模块：

  • asyncio: 用于异步编程
  • AsyncOpenAI: OpenAI 的异步客户端
  • agents 模块中导入了必要的组件

• 配置设置：

  BASE_URL = "https://open.bigmodel.cn/api/paas/v4/"
  API_KEY = "..."
  MODEL_NAME = "glm-4-flash"
  

  这里配置了智谱 AI 的接口地址和认证信息。

• 工具函数定义：

  @function_tool
  def get_weather(city: str):
  

  定义了一个获取天气的工具函数，这是一个模拟函数，总是返回晴天。

• 主要逻辑：

  async def main():
      agent = Agent(...)
      result = await Runner.run(agent, "What's the weather in Tokyo?")
  

  创建了一个智能助手，它：

  • 被设置为只用俳句形式回答
  • 使用智谱 AI 的 GLM-4-flash 模型
  • 具备查询天气的能力
  • 被要求查询东京的天气

3. 运行结果：

[debug] getting weather for Tokyo
Sunny skies in Tokyo.

显示了调用天气工具的调试信息和最终的响应结果。

这是一个展示如何使用自定义 LLM 提供商（这里是智谱 AI）来创建智能助手的示例，通过异步方式执行，并且集成了自定义工具函数。助手被特别设置为用俳句形式回答，这是一个有趣的约束条件。

参考链接：https://github.com/openai/openai-agents-python/tree/main/examples/model_providers