Skip to main content

A comprehensive toolkit for AI agent development and workflow orchestration with Gemini AI integration

Project description

Agentium - AI Agent Development Toolkit

PyPI version Python Support License: MIT

A comprehensive Python library designed for AI agent development and workflow orchestration. Agentium provides a rich set of tools and utilities that seamlessly integrate with popular AI frameworks like LangChain, LangGraph, and now features built-in Google Gemini API integration.

New Features

  • Google Gemini Integration - Built-in support for Gemini Pro, Gemini 1.5 Pro, and Gemini 1.5 Flash
  • Model Selection - Easy switching between different AI models
  • Enhanced AI Processing - All core features now support AI-enhanced processing
  • Advanced Configuration - Comprehensive model and processing configuration

Features

  • Condense: Intelligent content condensation and compression with AI enhancement
  • Optimizer: Refine text, code, and workflows for better performance
  • Rearranger: Organize and restructure content logically
  • Extractor: Extract structured information from various data sources
  • Communicator: Send messages and notifications across platforms
  • Translator: Multi-language translation with AI-powered tone adaptation
  • Insight Generator: Generate actionable insights from data using AI
  • Workflow Helper: Orchestrate complex tasks and triggers
  • Template Manager: Standardize outputs with customizable templates
  • Memory Helper: Context storage and retrieval system
  • Custom Summarizer: Create AI-enhanced summaries tailored to specific needs
  • Logger Utils: Track and debug operations with detailed logging
  • Gemini Integration: Native Google Gemini API support with model selection

Installation

pip install agentium

With AI Enhancement (Recommended):

# For full AI capabilities including Gemini
pip install agentium google-generativeai

# For LangChain integration
pip install agentium[langchain]

# For LangGraph integration
pip install agentium[langgraph]

# For development
pip install agentium[dev]

API Key Setup

Getting Your Google Gemini API Key

To use the AI-enhanced features, you need a Google Gemini API key:

  1. Visit Google AI Studio

  2. Create API Key

    • Click "Create API Key"
    • Choose "Create API key in new project" or select an existing project
    • Copy the generated API key and save it securely
  3. Set Up Your API Key (Choose one method):

    Method 1: Environment Variable (Recommended)

    # Windows (Command Prompt)
    set GEMINI_API_KEY=your_api_key_here
    
    # Windows (PowerShell)
    $env:GEMINI_API_KEY="your_api_key_here"
    
    # Linux/Mac
    export GEMINI_API_KEY=your_api_key_here
    

    Method 2: In Your Python Code

    from agentium.integrations.gemini import GeminiIntegration
    
    # Initialize with API key directly
    gemini = GeminiIntegration(api_key="your_api_key_here")
    

    Method 3: Using .env File

    # Create a .env file in your project root
    echo "GEMINI_API_KEY=your_api_key_here" > .env
    
    import os
    from dotenv import load_dotenv
    
    load_dotenv()
    api_key = os.getenv('GEMINI_API_KEY')
    

Important Security Notes

  • Never commit API keys to version control
  • Use environment variables in production
  • Keep your API keys secure and private
  • Monitor your API usage in Google Cloud Console

Quick Start

from agentium import Agentium
from agentium.integrations.gemini import GeminiIntegration

# Initialize with Gemini AI enhancement
agentium = Agentium()
gemini = GeminiIntegration(api_key="your-gemini-api-key")

# Basic text processing
condensed_text = agentium.condenser.condense("Your long text here...")
optimized_text = agentium.optimizer.optimize(condensed_text)

# AI-enhanced processing
enhanced_summary = gemini.enhance_condenser(
    text="Complex document content",
    style="executive-summary"
)

# Workflow orchestration
workflow_result = agentium.workflow_helper.process_workflow({
    "input": "Data to process",
    "steps": ["condense", "optimize", "extract"],
    "ai_enhanced": True
})

print(f"Processed result: {workflow_result}")

Configuration with Gemini

from agentium.integrations.gemini import GeminiIntegration, GeminiConfig

# Configure Gemini integration
config = GeminiConfig(
    model_name="gemini-1.5-pro",
    temperature=0.7,
    max_tokens=1000
)

gemini = GeminiIntegration(
    api_key="your-api-key",
    config=config
)

# Use with any Agentium feature
enhanced_insights = gemini.enhance_insights(
    data="Your data here",
    context="Financial analysis",
    insight_type="trend_analysis"
)

Sample Projects

The sample_projects/ directory contains 5 comprehensive examples demonstrating all Agentium features:

1. Content Processing Pipeline (content_pipeline.py)

Advanced document processing system with AI-enhanced content analysis, multi-format support, and intelligent workflow orchestration.

Features Demonstrated:

  • Content condensation and optimization
  • Multi-language translation with tone adaptation
  • Template-based report generation
  • Memory management and context storage
  • Gemini AI enhancement for superior content quality
# Example usage
pipeline = ContentProcessingPipeline()
result = pipeline.process_document("document.pdf", output_format="executive-summary")

2. Multilingual News Analyzer (news_analyzer.py)

Real-time news analysis system with sentiment analysis, trend detection, and multi-language support.

Features Demonstrated:

  • Real-time content extraction and processing
  • Advanced insight generation with trend analysis
  • Multi-language translation and sentiment analysis
  • Workflow orchestration for news processing
  • AI-enhanced content understanding
# Example usage
analyzer = NewsAnalyzer()
analysis = analyzer.analyze_news_feed("https://news-feed-url", languages=["en", "es", "fr"])

3. Data Intelligence Dashboard (data_dashboard.py)

Comprehensive data analysis and visualization system with AI-powered insights.

Features Demonstrated:

  • Advanced data extraction and processing
  • AI-enhanced insight generation
  • Real-time data visualization
  • Memory-based context management
  • Intelligent report generation
# Example usage
dashboard = DataIntelligenceDashboard()
insights = dashboard.generate_intelligence_report("sales_data.csv")

4. Automated Report Generator (report_generator.py)

Professional report generation system with AI-enhanced content creation and multi-format output.

Features Demonstrated:

  • Template-based report generation
  • AI-enhanced content optimization
  • Multi-format output (PDF, HTML, Markdown)
  • Workflow automation for report creation
  • Memory management for report templates
# Example usage
generator = AutomatedReportGenerator()
report = generator.generate_comprehensive_report("project_data", "quarterly-report")

5. Smart Communication Hub (communication_hub.py)

Intelligent communication orchestration system with multi-channel messaging and workflow automation.

Features Demonstrated:

  • Multi-channel communication management
  • Workflow-based message orchestration
  • AI-enhanced content optimization for different channels
  • Memory management for communication history
  • Intelligent message routing and optimization
# Example usage
hub = SmartCommunicationHub()
result = hub.process_and_distribute_message(
    "Important announcement",
    channels=["email", "slack", "teams"],
    ai_enhanced=True
)

Streamlit Demo

Try the interactive demo to explore all Agentium features:

# Install Streamlit
pip install streamlit plotly pandas

# Run the demo
streamlit run sample_projects/agentium_streamlit_demo.py

The demo includes:

  • Interactive feature testing
  • Real-time processing examples
  • AI model selection interface
  • Processing history and export capabilities
  • Comprehensive feature demonstrations

Framework Integration

LangChain Integration

from agentium.integrations.langchain import AgentiumTool
from langchain.agents import initialize_agent

# Create Agentium tools for LangChain
tools = AgentiumTool.create_all_tools()
agent = initialize_agent(tools, llm, agent_type="zero-shot-react-description")

LangGraph Integration

from agentium.integrations.langgraph import AgentiumNode
from langgraph import Graph

# Add Agentium nodes to LangGraph
graph = Graph()
graph.add_node("condenser", AgentiumNode.condenser_node)
graph.add_node("optimizer", AgentiumNode.optimizer_node)

Documentation

For detailed documentation and examples, visit our documentation site.

Contributing

We welcome contributions! Please see our Contributing Guide for details.

License

This project is licensed under the MIT License - see the LICENSE file for details.

Support

If you encounter any issues or have questions, please open an issue on GitHub.

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

agentium-1.1.0.tar.gz (78.6 kB view details)

Uploaded Source

Built Distribution

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

agentium-1.1.0-py3-none-any.whl (85.9 kB view details)

Uploaded Python 3

File details

Details for the file agentium-1.1.0.tar.gz.

File metadata

  • Download URL: agentium-1.1.0.tar.gz
  • Upload date:
  • Size: 78.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.5

File hashes

Hashes for agentium-1.1.0.tar.gz
Algorithm Hash digest
SHA256 649d24b266c0cad5b0e3e713123bbe4cd475c8b036d037c453d1c2bdb632c59f
MD5 bb866bad3808ee0bed62a5af6ca05444
BLAKE2b-256 d1cd97e6e6b4fafa772cada1a6baa680b46ca31e7965e01de5a3692d51844747

See more details on using hashes here.

File details

Details for the file agentium-1.1.0-py3-none-any.whl.

File metadata

  • Download URL: agentium-1.1.0-py3-none-any.whl
  • Upload date:
  • Size: 85.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.5

File hashes

Hashes for agentium-1.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 4775df1eaf6a88ed342f503d2e7440c07b2e9273320b0d6875ab29be53c9a51a
MD5 d8a3ca150c4cc068f418d90bd229985e
BLAKE2b-256 21ab5f2034f49cd0367c09cffdb3b32cf72d13e98ea2c492f835f0d1b01245cf

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