OpenAI Agents Integration

import { Agent, setDefaultOpenAIClient } from "@openai/agents";
import OpenAI from "openai";
import dotenv from "dotenv";

dotenv.config();

const client = new OpenAI({
  baseURL: "https://ai-gateway.helicone.ai/v1",
  apiKey: process.env.HELICONE_API_KEY
});

// Set the client globally for all agents
setDefaultOpenAIClient(client);

import os
from agents import set_default_openai_client
from openai import OpenAI

client = OpenAI(
    base_url="https://ai-gateway.helicone.ai/v1",
    api_key=os.getenv("HELICONE_API_KEY")
)

# Set the client globally for all agents
set_default_openai_client(client)

import { Agent, run, tool } from "@openai/agents";
import { z } from "zod";

// Define tools
const calculator = tool({
  name: "calculator",
  description: "Perform basic arithmetic operations",
  parameters: z.object({
    operation: z.enum(["add", "subtract", "multiply", "divide"]),
    a: z.number(),
    b: z.number()
  }),
  async execute({ operation, a, b }) {
    switch (operation) {
      case "add":
        return a + b;
      case "subtract":
        return a - b;
      case "multiply":
        return a * b;
      case "divide":
        if (b === 0) return "Error: Division by zero";
        return a / b;
    }
  }
});

// Create an agent with tools
const agent = new Agent({
  name: "Assistant",
  instructions: "You are a helpful assistant.",
  tools: [calculator],
  model: "gpt-4o-mini",
});

// Run the agent
const result = await run(agent, "Multiply 2 by 2");
console.log(result.finalOutput);

from agents import Agent, Runner, tool
from typing import Literal

# Define tools
@tool
def calculator(operation: Literal["add", "subtract", "multiply", "divide"], a: float, b: float) -> float | str:
    """Perform basic arithmetic operations."""
    if operation == "add":
        return a + b
    elif operation == "subtract":
        return a - b
    elif operation == "multiply":
        return a * b
    elif operation == "divide":
        if b == 0:
            return "Error: Division by zero"
        return a / b

# Create an agent with tools
agent = Agent(
    name="Assistant",
    instructions="You are a helpful assistant.",
    tools=[calculator],
    model="gpt-4o-mini"
)

# Run the agent
result = Runner.run_sync(agent, "Multiply 2 by 2")
print(result.final_output)