Custom Plugins

# .env
# LLM API key for agent
OPENAI_API_KEY=sk-your-openai-api-key-here

# Weather API key for the custom plugin (example)
WEATHER_API_KEY=your-weather-api-key

# Database for agent persistence
DB_URL=sqlite://./astreus.db

import { Agent, ToolDefinition, Plugin } from '@astreus-ai/astreus';

// Define a custom tool
const weatherTool: ToolDefinition = {
  name: 'get_weather',
  description: 'Get current weather for a location',
  parameters: {
    location: {
      name: 'location',
      type: 'string',
      description: 'City name',
      required: true
    },
    units: {
      name: 'units',
      type: 'string',
      description: 'Temperature units',
      required: false,
      enum: ['celsius', 'fahrenheit']
    }
  },
  handler: async (params) => {
    try {
      // Simulate weather API call
      const weather = {
        temperature: 22,
        conditions: 'sunny',
        location: params.location
      };
      
      return {
        success: true,
        data: weather
      };
    } catch (error) {
      return {
        success: false,
        error: error.message
      };
    }
  }
};

// Create plugin
const weatherPlugin: Plugin = {
  name: 'weather-plugin',
  version: '1.0.0',
  description: 'Weather information tools',
  tools: [weatherTool]
};

// Create agent and register plugin
const agent = await Agent.create({
  name: 'WeatherAgent',
  model: 'gpt-4o'
});

await agent.registerPlugin(weatherPlugin, {
  name: 'weather-plugin'
});

// Use the plugin in conversation
const response = await agent.ask("What's the weather like in Tokyo?");
console.log(response); // Agent automatically uses the weather tool