Skip to main content

A package for building intelligent agents through OpenAI's API.

Project description

Personal Project: AI Agent Framework

A modular and extensible framework for building AI agents capable of interacting with users, managing tools, and delegating tasks to helper agents. The project is designed for flexibility, maintainability, and ease of integration with tools.

Key Features

  • Agent Class: A self-sufficient AI agent with chat capabilities, history management, and tool integration.
  • ToolBox: A utility for registering and managing tools with strict JSON schemas.
  • Multi-Agent Systems: Agents can delegate tasks to other helper agents, enabling complex workflows and collaborative problem-solving.
  • Extensibility: Easily add new tools, including to integrate with external APIs.

Repository Structure

  1. agent.py

    • Implements the Agent class, which serves as the core of the framework.
    • Features:
      • Chat interface with history management using deque.
      • Integration with tools and helper agents.
      • Graceful handling of user interruptions (KeyboardInterrupt).
      • Modularized helper-agent tool registration.
  2. tool_box.py

    • Provides the ToolBox class for managing tools.
    • Features:
      • Tool registration with JSON schema validation.
      • Logging of tool calls and results.
      • Support for merging multiple ToolBox instances.
  3. tools.py

    • A compilation of example tools for integration with the framework.
    • Includes tools for tasks like managing Google Calendar events and sending emails.

Getting Started

Prerequisites

  • Python 3.10 or higher
  • Required Python packages (install via pip):
    • openai
    • google-auth
    • google-auth-oauthlib
    • google-api-python-client
    • python-dotenv

Installation

  1. Clone the repository:

    git clone https://github.com/lundmj/aiAgents.git
    cd aiAgents
    
  2. Install dependencies:

    pip install -r requirements.txt
    

main Usage

  1. Run the agent with a system prompt:

    python main.py system_prompts/calendar_assistant.md -t calendar_tool_box
    
  2. Use the -h flag for help:

    python main.py -h
    

Defining Your Own

  1. Agent
    • Create an instance of the Agent class in agent.py.
  2. Tool
    • Import ToolBox into the file in which you want to define tools (see tools.py). Create an instance of the tool box.
    • Write a function that you want your agent to be able to call.
      • The parameters need to be strings or a numeric type.
      • Annotate the types of the parameters, give the parameters clear names, and define a return type (typically str). The ToolBox will enable the agent to read and understand this function signature.
      • Write a simple docstring explaining the functionality of the tool, using """ notation. ToolBox will also help the agent read this as an explanation of the tool.
    • Decorate the function with @<tool_box>.tool to store the tool in the tool box.
      • Replace <tool_box> with the variable name of your tool box.
    • Any number of tools can go in a tool box, and by stacking decorations, you can put a tool in any number of tool boxes.

Example Files

System Prompts

The system_prompts folder contins some example prompts to give an AI agent. These are what you provide a Path to when instantiating an agent.

Delegator Example

The file delegator.py is an example of three agents:

  • delegator_agent uses a reasoning model and has the other two agents provided to it as tools. It is instructed to simply carry out tasks, with knowledge that it may need to delegate. Notice that its system prompt contains no direction as to which agents it can delegate to; it deduces that from the agents it gets.
  • calendar_agent and email_agent are non-reasoning models (using gpt-4.1) that are instructed to handle their various tasks. They are provided their own set of tools in their tool boxes, which are independent of each other and the delegator agent above them.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

agentics_lundmj-1.0.0.tar.gz (7.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

agentics_lundmj-1.0.0-py3-none-any.whl (15.9 kB view details)

Uploaded Python 3

File details

Details for the file agentics_lundmj-1.0.0.tar.gz.

File metadata

  • Download URL: agentics_lundmj-1.0.0.tar.gz
  • Upload date:
  • Size: 7.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.4

File hashes

Hashes for agentics_lundmj-1.0.0.tar.gz
Algorithm Hash digest
SHA256 e7ede1e7c654b58a43ff8cb5c7396279c9678f29bcf05572f49f0081d44ac3b0
MD5 1305e33d605e90b3d396224085258f07
BLAKE2b-256 7b33f2a224ef083d3d1b539492e9fdaa8e5bf37ca3919f0d4fb501836bdc3eee

See more details on using hashes here.

File details

Details for the file agentics_lundmj-1.0.0-py3-none-any.whl.

File metadata

File hashes

Hashes for agentics_lundmj-1.0.0-py3-none-any.whl
Algorithm Hash digest
SHA256 9d23faa72eef5eea1aed33a0790c696bb9706f8ac38998bf5dc7754c592bc940
MD5 e6b4a69cd518dccec9f5aaf47a3b2635
BLAKE2b-256 7d1b7ded5bc9c1f6e8ae7e28556ef62df9d1e2ff94edd60c1652b87aa4b79257

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page