Compare commits

...
278 Commits
Author SHA1 Message Date
Mervin Praison dbd6062edf chore: bump praisonai to v2.2.70 and praisonaiagents to v0.0.142 across all files 2025-07-17 01:31:56 +01:00
Mervin Praison 7fc615ff70 Merge pull request #974 from MervinPraison/claude/pr-965-20250716-2356
fix: make TelemetryCollector.stop() consistent with shutdown behavior
2025-07-17 01:24:26 +01:00
Mervin Praison fabac466f9 Merge pull request #966 from MervinPraison/claude/issue-964-20250716-2340
fix: improve Ollama sequential tool execution to prevent redundant calls
2025-07-17 01:22:11 +01:00
github-actions[bot]andMervin Praison 8d98f27f9c fix: correct Ollama early stopping logic placement for sequential tool execution
Fixes critical issues identified in PR review:
- Move early stopping logic from inside ''if tool_calls:'' block to ''else:'' block
- Fix logic placement in both sync and async execution paths
- Remove unreachable code in _generate_ollama_tool_summary()
- Ensure early stopping triggers correctly when Ollama returns empty responses after tool execution

This ensures the fix properly handles the Ollama edge case where it returns
empty responses after successful tool execution, preventing infinite loops
while allowing proper sequential tool calls.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-17 00:02:48 +00:00
github-actions[bot]andMervin Praison cf0af4999d fix: make TelemetryCollector.stop() consistent with shutdown behavior
Ensures TelemetryCollector.stop() calls shutdown() instead of just flush()
for consistency with the main atexit handler and proper PostHog session cleanup.

Co-authored-by: Mervin Praison <[email protected]>
2025-07-17 00:01:42 +00:00
github-actions[bot]andMervin Praison f6ecd3b4db fix: improve Ollama early stopping logic to allow proper sequential tool execution
- Only trigger early stopping when no new tool calls are present in current response
- Prevents premature termination during sequential tool calls like get_stock_price -> multiply
- Maintains backward compatibility while fixing issue #964
- Applied to both sync and async execution paths

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:56:48 +00:00
Mervin Praison a348d186c3 Merge pull request #965 from MervinPraison/claude/issue-963-20250716-2334
fix: properly shutdown telemetry aiohttp sessions to prevent ImportError during Python shutdown
2025-07-17 00:54:16 +01:00
Mervin Praison 30c0f2458f Merge pull request #961 from MervinPraison/claude/issue-958-20250716-2304
fix: ensure consistent Task/Response formatting across all LLM providers
2025-07-17 00:53:53 +01:00
Mervin Praison 668d61344f Merge branch 'main' into claude/issue-958-20250716-2304 2025-07-17 00:47:55 +01:00
github-actions[bot]andMervin Praison e8e9625f22 fix: improve Ollama sequential tool execution to prevent redundant calls
- Add early stopping logic for Ollama after first successful tool execution
- Enhance tool summary generation to create more natural, OpenAI-like responses
- Apply fix to both sync and async execution paths
- Prevent multiple tool calls when execution is complete
- Maintains full backward compatibility

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:47:43 +00:00
Mervin Praison cc721ba0db feat: upgrade praisonai to v2.2.69 and add realtime streaming support 2025-07-17 00:45:38 +01:00
Mervin Praison 07a30ab33d Merge pull request #962 from MervinPraison/claude/issue-956-20250716-2316
fix: enable real-time streaming regardless of verbose setting
2025-07-17 00:41:34 +01:00
github-actions[bot]andMervin Praison 92978fdb1c fix: properly shutdown telemetry aiohttp sessions to prevent ImportError during Python shutdown
- Change atexit handler from flush() to shutdown() in telemetry/__init__.py
- This ensures PostHog aiohttp sessions are properly closed before Python shutdown
- Prevents ''ImportError: sys.meta_path is None, Python is likely shutting down'' error
- Maintains backward compatibility with existing functionality

Fixes #963

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:40:40 +00:00
github-actions[bot]andMervin Praison 15a307dd4e fix: complete streaming fix for all LLM providers and improve console handling
- Fix custom LLM streaming (Gemini/Vertex AI) in llm.py by removing verbose condition
- Improve console parameter consistency when streaming is enabled
- Ensures streaming works for stream=True regardless of verbose setting
- Maintains full backward compatibility

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:38:27 +00:00
Mervin Praison 053ad043b8 Merge pull request #960 from MervinPraison/claude/issue-940-20250716-2252
fix: accumulate tool results across iterations in Ollama sequential execution
2025-07-17 00:35:49 +01:00
github-actions[bot]andMervin Praison 6e664995e8 fix: add missing flag reset in async achat method for consistent display behavior
- Added _final_display_shown = False reset in achat() method to match chat() method
- Ensures proper display flag management for both sync and async agent interactions
- Maintains consistency across all agent conversation entry points

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:32:12 +00:00
github-actions[bot]andMervin Praison ccb1bccf0f chore: remove unused custom_display_fn from agent.py
- Removed unused custom_display_fn function that was defined but never called
- Cleanup after streaming fix to avoid confusion
- No functional changes, streaming fix remains intact

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:31:38 +00:00
Mervin Praison 94c8fc21e6 Merge pull request #959 from MervinPraison/claude/issue-947-20250716-2244
fix: resolve task_name undefined error in async execution
2025-07-17 00:29:30 +01:00
github-actions[bot]andMervin Praison 48575b4ee0 fix: ensure callback execution in all achat() paths
- Enhanced _chat_completion method signature to accept task metadata parameters
- Updated all calls to _chat_completion to pass task context
- Fixed NameError where task_name, task_description, task_id were undefined
- Updated _apply_guardrail_with_retry to propagate task metadata
- Ensures complete task metadata propagation in sync chat execution path

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:28:05 +00:00
github-actions[bot]andMervin Praison 2ce7136d0c refactor: improve Ollama sequential fix with better test and deduplication
- Enhanced test file with comprehensive behavioral validation
- Added proper module import handling and mock-based testing
- Extracted duplicated logic into _handle_ollama_sequential_logic helper
- Eliminated 18-line code duplication between sync and async methods
- Improved maintainability while preserving backward compatibility

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:27:36 +00:00
github-actions[bot]andMervin Praison 3bea2ffb10 fix: enable real-time streaming regardless of verbose setting
- Remove dependency on verbose=True for streaming display function
- Allow stream=True to work with verbose=False for real-time output
- Maintain backward compatibility with existing verbose=True usage
- Fixes issue where Gemini/Google Vertex AI streaming was buffered

Resolves #956

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:22:35 +00:00
Mervin Praison fcbd915a0e docs: move star history section above feature roadmap for better visibility 2025-07-17 00:21:52 +01:00
github-actions[bot]andMervin Praison ae7eb29684 fix: ensure callback execution in all achat() paths
Adds missing callback execution to tool-based responses and JSON/Pydantic
output paths in the achat() method to ensure consistent task metadata
propagation across all execution flows.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:19:57 +00:00
Mervin Praison c5ec1ffea0 refactor: move test files to src/praisonai-agents directory for better organization 2025-07-17 00:19:56 +01:00
Mervin Praison aad5f286e2 chore: bump PraisonAI to v2.2.68 and PraisonAI Agents to v0.0.140 2025-07-17 00:19:32 +01:00
github-actions[bot]andMervin Praison 1aa20de337 fix: ensure consistent Task/Response formatting across all LLM providers
- Modified display condition to show formatted output for both OpenAI and custom LLM providers (like Gemini)
- Added _final_display_shown flag to prevent duplicate displays within same execution
- Reset flag at start of each chat() call for new conversations
- Maintains backward compatibility and existing verbose control behavior

Fixes #958

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:15:08 +00:00
Mervin Praison 3bc185d15d Merge pull request #957 from MervinPraison/claude/comprehensive-task-name-fix-20250716
fix: comprehensive task_name undefined error resolution
2025-07-17 00:14:20 +01:00
github-actions[bot]andMervin Praison 6b8a7110f2 refactor: simplify redundant hasattr checks and fix test formatting
- Remove redundant hasattr checks in agent.py as getattr handles missing attributes
- Fix unnecessary f-string prefixes in test file
- Remove unused variable assignment in test

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:08:04 +00:00
Mervin Praison 5ab0f245c5 Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-17 00:03:23 +01:00
Mervin Praison 5baa3f9b96 fix: update LLM model in Ollama sequential agent from qwen3 to llama3.2 for improved performance 2025-07-17 00:03:04 +01:00
github-actions[bot]andMervin Praison 28c6d026f7 fix: resolve task_name undefined error in async execution
Addresses issue #947 where task_name undefined errors occurred during async parallel task execution.

Key Changes:
- Enhanced Agent.achat() method to accept task context parameters (task_name, task_description, task_id)
- Updated aexecute_task() to pass task metadata when calling agent.achat()
- Added callback execution to both async execution paths in achat() method
- Updated callback system to properly propagate task context

Root Cause: Async execution path was missing task metadata propagation to LLM callbacks.

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 23:00:34 +00:00
github-actions[bot]andMervin Praison 802716e5ec fix: comprehensive task_name undefined error resolution
- Add task_name, task_description, task_id parameters to achat method signature
- Update all achat call sites in agents.py to pass task parameters where available
- Provide fallback None values for API handler contexts without task context
- Fix Agent.aexecute to extract task info when available
- Resolves NameError: name ''task_name'' is not defined in parallel agent execution

Fixes persistent errors in agentic parallelization workflows where callbacks
expect task context parameters but they were not being passed correctly through
the async execution chain.

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 22:58:16 +00:00
github-actions[bot]andMervin Praison 9eaa2731da fix: accumulate tool results across iterations in Ollama sequential execution
- After successful tool execution, give Ollama LLM explicit prompt for final answer
- Only generate tool summary if LLM fails to respond to explicit prompt
- Prevents premature tool summary generation after sequential execution
- Applied to both sync and async tool execution loops
- Preserves infinite loop prevention while enabling natural responses

Fixes #940

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 22:56:02 +00:00
Mervin Praison 0d3e1632a3 Merge pull request #949 from Dhivya-Bharathy/models-providers-reorganization
Reorganize models and providers structure with comprehensive examples
2025-07-16 23:45:08 +01:00
Mervin Praison da14b1d8e9 Merge pull request #955 from MervinPraison/claude/pr-953-20250716-1258
fix: resolve Ollama infinite loop issue with minimal changes
2025-07-16 16:02:48 +01:00
Mervin Praison c0679f4528 Merge pull request #954 from MervinPraison/claude/issue-940-20250716-1247
fix: prevent premature termination in Ollama sequential tool execution
2025-07-16 15:18:42 +01:00
github-actions[bot]andMervin Praison 930555d921 fix: accumulate tool results across iterations in Ollama sequential execution
- Replace hardcoded iteration threshold (3) with class constant OLLAMA_SUMMARY_ITERATION_THRESHOLD
- Add proper constant definition for better maintainability and readability
- Applied consistently to both sync and async methods
- Addresses reviewer feedback about magic numbers
- Ensures proper tool result accumulation across iterations

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 13:24:35 +00:00
Mervin Praison 859fb6a263 chore: bump praisonai to v2.2.67 and praisonaiagents to v0.0.138 across all files 2025-07-16 14:16:30 +01:00
github-actions[bot]andMervin Praison f154dba804 fix: accumulate tool results across iterations in Ollama sequential execution
- Add accumulated_tool_results list to store tool results across all iterations
- Prevents loss of tool execution history when generating summaries
- Applied to both sync (get_response) and async (get_response_async) methods
- Maintains backward compatibility with existing functionality
- Fixes issue where only current iteration''s tool results were available to summary generation

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 13:15:14 +00:00
Mervin Praison fb2a16126f Merge pull request #952 from MervinPraison/claude/issue-947-20250716-1205
fix: resolve task_name undefined error in LLM callback execution
2025-07-16 14:14:52 +01:00
Mervin Praison 77ab361d64 refactor: reorganize test files and update package versions to 2.2.66 2025-07-16 14:12:45 +01:00
Mervin Praison a276a610a0 Merge pull request #951 from MervinPraison/claude/issue-950-20250716-1207
feat: Enable imports from PraisonAI package for issue #950
2025-07-16 14:08:34 +01:00
github-actions[bot] ddb547f6a5 cleanup: remove debug test files 2025-07-16 13:08:03 +00:00
github-actions[bot]andMervin Praison 24437b40d9 fix: resolve Ollama infinite loop issue with minimal changes
- Remove redundant length check in _generate_ollama_tool_summary
- Simplify verbose conditional checks in both sync and async methods
- Add safety checks to prevent infinite loops after 5 iterations
- Maintain backward compatibility with other LLM providers
- Add comprehensive test suite to validate fixes

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 13:07:39 +00:00
Dhivya-Bharathy 582049f759 Update model examples to use new clean format with praisonaiagents.Agent 2025-07-16 18:32:35 +05:30
github-actions[bot]andMervin Praison e8f4eff173 refactor: Use dynamic approach for praisonaiagents imports
- Replace manual symbol list with dynamic import using
- Use  for automatic maintenance
- Remove manual tracking of imported symbols
- Reduce code from 49 to 17 lines
- Future-proof: automatically inherits new exports from praisonaiagents

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 12:56:29 +00:00
github-actions[bot]andMervin Praison e1b93000be fix: prevent premature termination in Ollama sequential tool execution
- Modified tool execution logic to allow at least 3 iterations before triggering summary generation
- This prevents Ollama from prematurely terminating when sequential tool calls are needed
- Applied fix to both conversation() and reasoning() methods in LLM class
- Resolves issue where Ollama would generate tool summaries too early, blocking proper tool chaining

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Mervin Praison <[email protected]>
2025-07-16 12:51:13 +00:00
Mervin Praison 464cc128db Merge pull request #953 from MervinPraison/revert-945-claude/issue-940-20250716-0711
Revert "fix: prevent premature termination in Ollama sequential tool execution"
2025-07-16 13:46:12 +01:00
Mervin Praison f44bb7fa39 Revert "fix: prevent premature termination in Ollama sequential tool execution" 2025-07-16 13:46:01 +01:00
Mervin Praison 5b70f4b34d Merge pull request #945 from MervinPraison/claude/issue-940-20250716-0711
fix: prevent premature termination in Ollama sequential tool execution
2025-07-16 13:45:16 +01:00
github-actions[bot]andMervin Praison 5514f17346 fix: Resolve unconditional __all__ list issue causing import errors
- Fixed critical bug where __all__ list included symbols from praisonaiagents
  even when import failed due to missing dependencies
- Implemented dynamic __all__ construction based on successfully imported symbols
- Maintained backward compatibility while fixing the cursor review issue
- Added comprehensive test files to verify fix and backward compatibility

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 12:38:33 +00:00
github-actions[bot]andMervin Praison 318ececfb4 fix: correct function name in test_fix_verification.py
- Fixed unused variable 'results' in test_fix_verification.py
- Maintained test functionality while removing static analysis warning

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 12:35:26 +00:00
github-actions[bot]andMervin Praison 83afc0891f fix: correct function name in test_fix_verification.py
- Changed register_callback to register_display_callback
- Ensures test runs correctly and validates the fix

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 12:29:42 +00:00
github-actions[bot]andMervin Praison 8e7323c484 refactor: replace hardcoded patterns with dynamic tool usage detection
- Replace hardcoded acknowledgment_patterns and status_patterns with intelligent context analysis
- Implement _analyze_tool_chain_context() for dynamic tool execution pattern detection
- Add _assess_response_quality() for content-based response evaluation
- Replace iteration_count >= 5 threshold with _should_generate_tool_summary() dynamic logic
- Improve tool chain completion detection with confidence scoring
- Increase safety fallback threshold from 5 to 7 iterations
- Maintain backward compatibility with existing providers

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 12:28:18 +00:00
Dhivya-Bharathy 8dd6494a59 Update all provider examples to use new clean format with praisonaiagents.Agent 2025-07-16 17:54:50 +05:30
github-actions[bot]andMervin Praison 857238a82a fix: resolve task_name undefined error in LLM callback execution
- Fixed 6 direct execute_sync_callback calls missing task-related parameters
- Added agent_name, agent_role, agent_tools, task_name, task_description, task_id
- Ensures consistent callback parameter passing between verbose and non-verbose modes
- Prevents infinite loop errors when callbacks expect task_name parameter
- Maintains backward compatibility with existing functionality

Fixes #947

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 12:21:27 +00:00
github-actions[bot]andMervin Praison 658a2397fc feat: Enable imports from PraisonAI package for issue #950
This change allows users to import from the main PraisonAI package instead of
just from praisonaiagents, while maintaining full backward compatibility.

Changes:
- Modified src/praisonai/praisonai/__init__.py to re-export all classes from praisonaiagents
- Added comprehensive __all__ list for proper exports
- Maintained backward compatibility - all existing imports continue to work

New usage patterns now supported:
- from praisonai import Agent, Task, PraisonAIAgents
- from praisonai import Tools, Memory, Knowledge, etc.

Backward compatibility maintained:
- from praisonaiagents import Agent, Task, PraisonAIAgents (still works)

Tests added:
- test_import_conversion.py: Comprehensive test suite
- test_issue_950.py: Specific test for the issue requirements

Resolves #950

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 12:14:45 +00:00
Dhivya-Bharathy 1ab09fa259 Reorganize models and providers structure - Move Kimi2 agents to models/kimi with kimi_ prefix - Move Grok4 agents to models/grok with grok_ prefix - Create 10 new Groq agents in models/groq with groq_ prefix - Add provider examples for OpenAI, Anthropic, Google, DeepSeek, Mistral, Cohere, xAI - Remove README files as requested 2025-07-16 16:35:00 +05:30
Dhivya-Bharathy f7f4a6c0bf Add 10 new agents to Kimi2_Intelligent_Agents (aerospace, bioinformatics, blockchain, energy, IoT, metaverse, nanotechnology, quantum AI, quantum computing, robotics) 2025-07-16 15:02:46 +05:30
github-actions[bot]andMervin Praison a743f4b68c fix: prevent premature termination in Ollama sequential tool execution
- Added iteration count threshold (>=5) before generating Ollama tool summary
- Prevents premature termination after first tool call in sequential execution
- Maintains infinite loop prevention for truly stuck scenarios
- Applied fix to both sync and async methods in llm.py
- Preserves backward compatibility with other LLM providers

Fixes issue where Ollama would terminate after first tool call instead of
continuing with sequential execution (e.g., get_stock_price -> multiply)

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-16 07:28:09 +00:00
Mervin Praison 77b7c3b1ce Merge pull request #944 from MervinPraison/claude/fix-ollama-response-overwrite-20250715-2344
fix: prevent Ollama tool summary from being overwritten by empty response
2025-07-16 01:08:51 +01:00
Mervin Praison 48c65a742c Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-16 00:56:06 +01:00
Mervin Praison 35295ad167 refactor: move test files to src/praisonai-agents directory 2025-07-16 00:56:05 +01:00
Mervin Praison 1a6078d7eb Delete test_ollama_response_fix.py 2025-07-16 00:50:14 +01:00
github-actions[bot]andMervin Praison fbd4c53796 fix: prevent Ollama tool summary from being overwritten by empty response
- Add condition to check if final_response_text is already set before overwriting
- Fix applies to both sync (get_response) and async (get_response_async) methods
- Prevents tool summary from being lost when Ollama returns empty response after tool execution
- Maintains backward compatibility for non-Ollama providers
- Resolves issue where agent.start() returns None despite successful tool execution

Fixes the remaining issue from PR #943 where tool summary was generated but overwritten

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 23:48:51 +00:00
Mervin Praison 55e70d521e Merge pull request #942 from MervinPraison/claude/issue-939-20250715-2248
feat: Add 7 comprehensive examples for advanced PraisonAI agent features
2025-07-16 00:47:52 +01:00
github-actions[bot]andMervin Praison d4023c0484 feat: Organize advanced examples into appropriate folders
- Move callback systems to general/
- Move memory examples to memory/
- Move state management to stateful/
- Move hierarchical agents to agents/
- Move guardrails to guardrails/
- Move self-reflection to concepts/

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 23:46:29 +00:00
Mervin Praison d88200e258 Merge pull request #943 from MervinPraison/claude/issue-940-20250715-2304
fix: Resolve Ollama infinite tool call loops by improving response handling
2025-07-16 00:35:29 +01:00
github-actions[bot]andMervin Praison beac0b47a2 refactor: eliminate code duplication and improve test quality in Ollama fix
- Extract duplicated Ollama tool summary logic into _generate_ollama_tool_summary() helper method
- Replace magic number 10 with OLLAMA_MIN_RESPONSE_LENGTH constant
- Fix test_tool_summary_generation() to call production code instead of reimplementing logic
- Add comprehensive test scenarios for different response types and provider types
- Maintain backward compatibility and fix functionality

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 23:26:15 +00:00
github-actions[bot]andClaude 6d6c44e7ca fix: Convert advanced examples to flat format with minimal code
Simplified all 7 advanced examples to follow flat format requirements:
- Removed 20+ custom classes across all examples
- Reduced total lines from 2,829 to 514 (82% reduction)
- Limited to 1-2 agents max per example
- Focused on core functionality demonstration
- Maintained advanced feature examples while making them accessible

Changes per file:
- advanced-callback-systems.py: 404→59 lines, removed 3 classes
- advanced-graph-memory-integration.py: 226→75 lines, reduced to 1 agent
- advanced-memory-search-optimization.py: 568→76 lines, removed 4 classes
- advanced-state-management-patterns.py: 437→80 lines, removed 3 classes
- autoagents-hierarchical-generation.py: 158→73 lines, removed custom class
- production-guardrails-patterns.py: 480→65 lines, removed 5 classes
- self-reflection-optimization-patterns.py: 556→46 lines, removed 4 classes

All examples now follow the established flat format pattern with simple,
direct agent/task creation and minimal code complexity.

🤖 Generated with Claude Code

Co-Authored-By: Claude <[email protected]>
2025-07-15 23:22:16 +00:00
Mervin Praison 1cd838f691 chore: bump praisonai version to 2.2.65 and add autogen-v4 dependencies 2025-07-16 00:12:42 +01:00
Mervin Praison 319944e012 Merge pull request #941 from MervinPraison/claude/issue-938-20250715-2246
fix: Ensure agent names are valid Python identifiers for AutoGen v0.4
2025-07-16 00:10:52 +01:00
github-actions[bot]andMervin Praison b53162aa9e fix: Improve sanitization to preserve double underscores
- Fixed aggressive underscore collapsing that broke dunder names like __init__
- Now only collapses 5+ consecutive underscores instead of any consecutive underscores
- Preserves trailing underscores in double underscore patterns (__name__)
- Handles edge case where purely invalid characters become single underscore
- All 15 test cases now pass including magic method names
- Maintains backward compatibility for AutoGen v0.2, CrewAI, PraisonAI

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 23:09:41 +00:00
github-actions[bot]andClaude 7d42e31e0a feat: Add comprehensive examples for advanced PraisonAI agent features
Added 7 new example files demonstrating advanced functionality:
- autoagents-hierarchical-generation.py: Hierarchical agent generation patterns
- advanced-graph-memory-integration.py: Graph-based memory systems
- advanced-callback-systems.py: Custom callback implementations
- production-guardrails-patterns.py: Production safety and guardrails
- advanced-state-management-patterns.py: Complex state handling
- advanced-memory-search-optimization.py: Memory search optimizations
- self-reflection-optimization-patterns.py: Self-improvement patterns

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>
2025-07-15 23:09:07 +00:00
github-actions[bot]andMervin Praison 9c96a838a8 fix: Resolve Ollama infinite tool call loops by improving response handling
- Add special handling for Ollama models in both sync and async tool execution loops
- When Ollama returns empty/minimal response after tool execution, generate summary from tool results
- Prevents infinite loops while maintaining functionality for other LLM providers
- Add comprehensive test suite to validate fix and ensure backward compatibility
- Fixes issue where Ollama models would repeat tool calls indefinitely

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 23:08:01 +00:00
github-actions[bot]andMervin Praison 52c249fde0 fix: Ensure agent names are valid Python identifiers for AutoGen v0.4
AutoGen v0.4 requires agent names to be valid Python identifiers,
but role names like ''Narrative Designer'' contain spaces which are invalid.

- Added sanitize_agent_name_for_autogen_v4() function to convert any
  string to a valid Python identifier
- Applied sanitization only in AutoGen v0.4 code path for backward compatibility
- AutoGen v0.2 and other frameworks remain unchanged
- Handles spaces, special characters, Python keywords, and edge cases

Resolves issue where ''Narrative Designer'' role causes error:
''The agent name must be a valid Python identifier''

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 22:51:54 +00:00
Mervin Praison 7607df753f Merge pull request #937 from MervinPraison/claude/issue-934-20250715_165720
fix: Replace deprecated Pydantic .dict() with .model_dump() in example files
2025-07-15 22:04:30 +01:00
Mervin Praison 8fa181bc55 Merge pull request #936 from MervinPraison/claude/issue-935-20250715_170032
feat: Add AutoGen v0.4 support with backward compatibility
2025-07-15 22:03:55 +01:00
github-actions[bot]andMervin Praison ccbf0ebebb fix: Ensure backward compatibility for Pydantic v1/v2 in hasattr checks
- Fix inconsistent hasattr checks in ZeroScript_AI_TestExecutor.ipynb
- Update fix_notebook.py to handle both method call and hasattr replacement
- Add comprehensive backward compatibility test suite
- Ensure Pydantic v1 models fallback to str() when model_dump() unavailable

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 17:19:41 +00:00
Mervin Praison d3f0c217ca feat: update default model from gpt-4o to gpt-4o-mini and bump version to 2.2.64 2025-07-15 18:09:57 +01:00
github-actions[bot]andMervin Praison 6995a446e1 feat: Add AutoGen v0.4 support with backward compatibility
- Add optional dependency ''autogen-v4'' with autogen-agentchat>=0.4.0, autogen-ext[openai]>=0.4.0, autogen-core>=0.4.0
- Implement async _run_autogen_v4() method with RoundRobinGroupChat for parallel multi-agent execution
- Add AUTOGEN_VERSION environment variable for version control (auto/v0.2/v0.4)
- Maintain full backward compatibility with existing v0.2 implementation
- Auto-detect available versions with intelligent fallback logic
- Simplify tool integration for v0.4 (single AssistantAgent vs dual-agent pattern)
- Add proper termination conditions and error handling for v0.4

Resolves #935

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 17:08:28 +00:00
github-actions[bot]andMervin Praison 72c70ce050 fix: Replace deprecated Pydantic .dict() with .model_dump() in example files
- Fixed examples/python/api/secondary-market-research-api.py:443
- Fixed examples/python/concepts/reasoning-extraction.py:34
- Fixed ZeroScript_AI_TestExecutor.ipynb notebook
- Added test script to verify fixes work correctly
- Maintains backward compatibility with Pydantic v2
- Resolves PydanticDeprecatedSince20 warnings

Resolves #934

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 17:07:16 +00:00
Mervin Praison 6a13d2b971 Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-15 17:57:12 +01:00
Mervin PraisonandMervin Praison fdc747a9c9 fix: Enhance session memory persistence and improve chat history management
- Improved session memory persistence for agent chat history
- Optimised methods for saving and restoring agent chat histories
- Ensured compatibility with existing code while enhancing functionality
- Addressed issues related to chat history loss during session transitions

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 17:57:11 +01:00
Mervin Praison f381b93ae9 Merge pull request #933 from MervinPraison/claude/issue-931-20250715_160543
feat: add comprehensive examples for missing PraisonAI features
2025-07-15 17:52:35 +01:00
Mervin Praison 9c44945123 chore: update numpy and scipy version constraints in dependency files 2025-07-15 17:50:01 +01:00
Mervin Praison a19748902d Merge pull request #932 from MervinPraison/claude/issue-930-20250715_141043
fix: Add session memory persistence for agent chat history
2025-07-15 17:48:02 +01:00
Mervin Praison fa1061a7aa Rename SESSION_MEMORY_FIX_REVIEW.md to examples/python/session/SESSION_MEMORY_FIX_REVIEW.md 2025-07-15 17:47:26 +01:00
github-actions[bot]andClaude 775d97671f fix: Fix session memory persistence by disabling embeddings and correcting search queries
- Fixed session memory persistence issue where state was not being restored correctly
- Disabled embeddings in default memory config to avoid OpenAI API key requirement
- Changed search queries from metadata-based to content-based for SQLite compatibility
- Fixed search query from "type:session_state" to "Session state:" to match stored content
- Fixed agent chat history search from "type:agent_chat_history" to "Agent chat history for"
- Session state persistence now works correctly across session save/restore cycles
- Agent chat history persistence now works correctly across session save/restore cycles
- Maintains backward compatibility with existing code
- Fixes #930

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Claude <[email protected]>
2025-07-15 16:31:10 +00:00
github-actions[bot]andMervin Praison 6c4f4d981d fix: Add missing increment_state method and improve chat history persistence
- Fix f-string issues on lines 275, 300 (remove unnecessary f prefixes)
- Fix chat history initialization to always set agent.chat_history even if empty
- Fix critical data loss bug in _save_agent_chat_histories method
- Now prioritizes live agent history but falls back to restored history
- Prevents chat history loss when agents aren''t re-instantiated after restore

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 16:19:19 +00:00
github-actions[bot]andMervin Praison 72395efd32 feat: add comprehensive examples for missing PraisonAI features
- Add hierarchical process orchestration examples
- Add RouterAgent cost optimization patterns
- Add advanced knowledge management with semantic chunking
- Add comprehensive guardrails with validation workflows
- Add advanced task management with conditional execution
- Add session management with persistence and recovery
- Add production telemetry with OpenTelemetry integration
- Add advanced workflow patterns with decision trees

Resolves #931: Create missing examples where features exist

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 16:16:49 +00:00
efa29de346 fix: Add missing increment_state method and improve chat history persistence
- Add increment_state method (lines 356-359) for numeric state management
- Fix empty chat history save condition to track agent existence properly
- Improve robustness by checking agent is not None before accessing chat_history
- Enhance memory persistence by saving even empty chat histories to track agent state

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Mervin Praison <[email protected]>
Co-Authored-By: Claude <[email protected]>
2025-07-15 16:13:36 +00:00
github-actions[bot]andMervin Praison 75462584b3 fix: Add session memory persistence for agent chat history
- Agent chat history now persists across session save/restore cycles
- Added _agents dictionary to track agent instances and chat histories
- Enhanced Agent() method to restore chat history in restored sessions
- Added helper methods for saving/restoring agent chat histories
- Modified save_state() to automatically save agent chat histories
- Modified restore_state() to automatically restore agent chat histories
- Backward compatible - existing code continues to work unchanged
- Fixes #930

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 14:31:24 +00:00
Mervin Praison d352adbdac chore: bump praisonai to v2.2.62 and praisonaiagents to v0.0.135 across all files 2025-07-15 15:11:32 +01:00
Mervin Praison 68664fdd55 Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-15 15:04:19 +01:00
Mervin Praison bcda52cc44 chore: bump praisonai version from 2.2.60 to 2.2.61 across Docker and deployment files 2025-07-15 15:04:17 +01:00
Mervin Praison 4888dfb993 Merge pull request #928 from MervinPraison/claude/issue-926-20250715_112808
refactor: inline Gemini internal tools detection logic
2025-07-15 14:36:45 +01:00
github-actions[bot]andMervin Praison c5a3fc89cc refactor: inline Gemini internal tools detection logic
- Remove import dependency on is_gemini_internal_tool helper function
- Inline logic directly in llm.py and openai_client.py
- Maintain identical functionality and debug logging
- Handle valid tools: googleSearch, urlContext, codeExecution
- Properly skip unknown tools with appropriate logging
- Preserve backward compatibility

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 11:33:22 +00:00
Mervin Praison ca33559d4c Merge pull request #922 from MervinPraison/claude/issue-918-20250715_061404
fix: resolve Ollama sequential tool calling argument mixing issue
2025-07-15 12:24:28 +01:00
Mervin Praison b02449dfdb chore: bump praisonai to v2.2.60 and praisonaiagents to v0.0.134 2025-07-15 09:55:35 +01:00
Mervin Praison c4c0c1c146 Merge pull request #923 from MervinPraison/claude/issue-919-20250715_061529
fix: Add support for Gemini internal tools in tool formatting
2025-07-15 09:52:47 +01:00
Mervin Praison aa2d7e4c31 Merge pull request #924 from Dhivya-Bharathy/kimi2-agents-final
Add Kimi2_Intelligent_Agents: 10 specialized AI agents
2025-07-15 09:51:35 +01:00
Dhivya-Bharathy b339c5b3aa Add Kimi2_Intelligent_Agents: 10 specialized AI agents 2025-07-15 12:57:48 +05:30
github-actions[bot]andMervin Praison e535f98f4e refactor: optimize Gemini tools handling with shared utilities
- Extract GEMINI_INTERNAL_TOOLS constant to eliminate loop recreation
- Create shared is_gemini_internal_tool() utility function
- Remove code duplication between llm.py and openai_client.py
- Improve maintainability and performance based on reviewer feedback

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 06:40:42 +00:00
Mervin Praison bc349ad09f feat: restrict review comments to PRs only and update Claude bot instructions 2025-07-15 07:38:32 +01:00
Mervin Praison 5b8da1180e chore: clarify Claude review instructions to specify branch context 2025-07-15 07:29:21 +01:00
github-actions[bot]andMervin Praison 8f6d8f0a48 fix: Add support for Gemini internal tools in tool formatting
- Add explicit handling for Gemini internal tools (googleSearch, urlContext, codeExecution)
- Update docstrings to document new capability
- Resolves issue where Gemini internal tools were skipped with ''unsupported type'' error
- Maintains backward compatibility with existing tool formats

Closes #919

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 06:23:32 +00:00
github-actions[bot]andMervin Praison 3730b4d094 fix: resolve Ollama sequential tool calling argument mixing issue
- Add parameter validation method _validate_and_filter_ollama_arguments()
- Filter invalid parameters before tool execution for Ollama providers
- Prevent functions from receiving unexpected keyword arguments
- Add comprehensive test suite for validation functionality
- Maintain backward compatibility and graceful error handling

Fixes #918: Ollama sequential failing 3

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-15 06:21:08 +00:00
Mervin Praison 7f3810ed98 chore: add gh tool permission and update claude review comment with PR creation request 2025-07-15 07:19:48 +01:00
Mervin Praison c84a59ab4d feat: implement PostHog event tracking in telemetry module 2025-07-15 07:13:38 +01:00
Mervin Praison c6a6bfa827 Merge pull request #917 from MervinPraison/claude/issue-867-20250714_201010
feat: Add Gemini internal tools examples
2025-07-14 22:22:48 +01:00
github-actions[bot]andMervin Praison a604cd9584 fix: Replace eval() with safe AST-based expression evaluator and remove unused imports
- Replace insecure eval() usage with ast-based safe expression evaluator in README and combined examples
- Remove unused os imports from all Python example files
- Update comments to show import os inline for API key setup
- Addresses security vulnerabilities identified in code review

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 20:38:04 +00:00
github-actions[bot]andMervin Praison cb043da19c feat: Add Gemini internal tools examples
- Add Google Search tool example (gemini-google-search.py)
- Add URL Context tool example (gemini-url-context.py)
- Add Code Execution tool example (gemini-code-execution.py)
- Add combined tools example (gemini-internal-tools-combined.py)
- Add comprehensive documentation (README_GEMINI_INTERNAL_TOOLS.md)

These examples demonstrate how to use Gemini''s built-in internal tools
through PraisonAI''s existing tool system without any code changes.

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 20:17:41 +00:00
Mervin Praison 7b41444aad refactor: move test files into src/praisonai-agents directory for better organization 2025-07-14 21:13:34 +01:00
Mervin Praison 7c8b77063f chore: bump praisonai to v2.2.58 and praisonaiagents to v0.0.132 across all files 2025-07-14 21:12:31 +01:00
Mervin Praison 59eb8d6356 Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-14 21:07:47 +01:00
Mervin Praison 89bddc31c6 feat: add allowed tools configuration for Claude GitHub action workflow 2025-07-14 21:07:45 +01:00
Mervin Praison e6e908298b Merge pull request #916 from MervinPraison/claude/issue-878-20250714_190400
Fix: resolve duplicate callback execution in LLM interaction handling
2025-07-14 20:53:39 +01:00
Mervin Praison 2b3509c1d1 Delete reproduce_issue_878.py 2025-07-14 20:52:36 +01:00
Mervin Praison 6e8029055a Delete ISSUE_878_REPRODUCTION_GUIDE.md 2025-07-14 20:52:15 +01:00
Mervin Praison d2c9e8825a Merge pull request #915 from MervinPraison/claude/issue-911-20250714_190025
fix: add execute and aexecute methods to Agent class
2025-07-14 20:44:41 +01:00
Mervin Praison cd9816e218 Merge pull request #914 from MervinPraison/claude/issue-901-20250714_143624
Issue #901: Fix self-reflection with tools
2025-07-14 20:43:29 +01:00
Mervin Praison d400149075 chore: update claude review request message to include follow-up action 2025-07-14 20:42:15 +01:00
Mervin Praison 73b08e6b76 feat: add PR creation permissions and request in auto-issue workflow 2025-07-14 20:32:59 +01:00
github-actions[bot]andClaude 19c3a1f6a7 fix: address code review issues in test files
- Add missing sys.path setup to test_self_reflection_with_tools_verification.py
- Remove unnecessary f-string prefixes from print statements
- Add missing newlines at end of files for PEP 8 compliance

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>
2025-07-14 19:24:12 +00:00
71a7968318 fix: resolve duplicate callback execution in LLM interaction handling
- Fixed duplicate callback issue in src/praisonai-agents/praisonaiagents/llm/llm.py where callbacks were executed multiple times
- Added callback_executed flag to track and prevent duplicate callback invocations
- Ensured callbacks are only executed once per interaction, either through display_interaction or direct execute_sync_callback calls
- Improved generation_time consistency by calculating once and reusing the value
- Added comprehensive test files and reproduction scripts to verify the fix
- Added ISSUE_878_REPRODUCTION_GUIDE.md documenting the issue and solution

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>
Co-Authored-By: Mervin Praison <[email protected]>
2025-07-14 19:15:39 +00:00
Mervin Praison 1b258229c1 Merge pull request #913 from MervinPraison/claude/issue-906-20250714_175631
fix: Add callback execution to synchronous display functions
2025-07-14 20:12:26 +01:00
github-actions[bot]andMervin Praison d0e09f51f3 fix: Ensure consistent callback execution across all display functions
- Fix async functions to use execute_callback() instead of direct async_display_callbacks access
- Add missing execute_sync_callback() calls to sync display functions (instruction, tool_call, error, generating)
- Ensures both sync and async callbacks are triggered consistently for all display types
- Maintains backward compatibility through existing parameter filtering

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 19:04:19 +00:00
github-actions[bot]andMervin Praison 5adc862ead fix: add execute and aexecute methods to Agent class for backward compatibility
Resolves AttributeError: type object ''Agent'' has no attribute ''execute''

- Added execute(task, context=None) method that delegates to chat()
- Added aexecute(task, context=None) method that delegates to achat()
- Both methods handle task objects with description attribute or string inputs
- Maintains complete backward compatibility with existing code
- Fixes issue #911

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 19:03:24 +00:00
Mervin Praison 856db54334 Merge pull request #912 from MervinPraison/claude/issue-904-20250714_171858
Fix issue #904: Loop task callback inheritance and enhanced metadata support
2025-07-14 19:58:26 +01:00
github-actions[bot]andMervin Praison f94e5bb3d6 fix: Add callback execution to synchronous display functions
Fixes the duplicate callback issue by:

- Adding execute_sync_callback calls to display_interaction and display_self_reflection
- Fixing test to patch the correct function location
- Ensuring backward compatibility by maintaining existing function signatures
- Adding comprehensive tests for callback system verification

This ensures that callbacks are called exactly once per LLM response while maintaining full backward compatibility.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 18:06:08 +00:00
Mervin Praison 1cece603ec Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-14 18:59:38 +01:00
Mervin Praison 9e2bbb9bbb refactor: update tool imports in test files from calculator to evaluate for improved clarity and consistency 2025-07-14 18:59:36 +01:00
Mervin Praison 35b4cff7f7 Merge pull request #910 from MervinPraison/claude/issue-905-20250714_163222
fix: resolve litellm.BadRequestError with unrecognized output_json/output_pydantic parameters
2025-07-14 18:44:18 +01:00
Mervin Praison a094d1ff30 Merge pull request #908 from MervinPraison/claude/issue-907-20250714_165230
fix: add string LLM support to LLMGuardrail for Gemini models
2025-07-14 18:43:11 +01:00
Mervin Praison 000bdf3418 refactor: move test files to src/praisonai-agents and bump version to 2.2.57 2025-07-14 18:40:18 +01:00
github-actions[bot]andMervin Praison 826ff56fc5 refactor: simplify LLMGuardrail string handling and consolidate imports
- Remove redundant if/else logic in string LLM initialization
- Consolidate duplicate LLM imports using _get_llm_class helper
- Maintain all functionality while improving code clarity
- Addresses code review feedback for cleaner, more maintainable code

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 17:31:52 +00:00
github-actions[bot]andClaude 4bd0195d5d Fix issue #904: Loop task callback inheritance and enhanced metadata support
- Add callback inheritance in all 6 locations where loop tasks create subtasks in process.py
- Implement enhanced callback metadata support in task.py with run_context tracking
- Ensure backward compatibility with existing callback implementations
- Maintain proper callback chain for nested loop executions

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>
2025-07-14 17:28:28 +00:00
Mervin Praison 812abec2eb Merge pull request #897 from MervinPraison/claude/issue-896-20250714_104106
feat: enhance callbacks with agent and task context (fixes #896)
2025-07-14 18:27:25 +01:00
github-actions[bot]andMervin Praison e58f6deb4b fix: resolve critical runtime issues in callback enhancement
- Fix undefined variables in llm.py response() and aresponse() methods
  by adding missing parameters to method signatures
- Fix backward compatibility by adding signature checking to callbacks
  in display_interaction(), execute_callback(), and execute_sync_callback()
- Ensure consistent parameter propagation in agent.py get_response calls
- Legacy callbacks now work without TypeError, new callbacks get full context

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 17:21:11 +00:00
Mervin Praison f23680c121 Merge pull request #903 from MervinPraison/claude/issue-899-20250714_122107
fix: exclude response_format parameter for Gemini image models
2025-07-14 18:01:38 +01:00
Mervin Praison e569400260 Merge pull request #902 from MervinPraison/claude/issue-901-20250714_140528
Issue #901: Changes from Claude
2025-07-14 17:57:58 +01:00
github-actions[bot]andMervin Praison 75c4d8432c fix: add string LLM support to LLMGuardrail for Gemini models
- Add _initialize_llm method to convert string identifiers to LLM instances
- Update __init__ to handle string LLM identifiers like ''gemini/gemini-2.5-flash-lite-preview-06-17''
- Maintain backward compatibility with existing LLM instances, dict configs, and None values
- Use same LLM initialization logic as Agent class for consistency
- Fixes issue #907 where LLMGuardrail failed with ''Unsupported LLM type'' error

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 16:57:24 +00:00
Mervin Praison 6249d14d3f Merge pull request #895 from MervinPraison/claude/issue-893-20250714_084251
fix: resolve Ollama sequential tool calling redundant execution issue
2025-07-14 17:55:05 +01:00
github-actions[bot]andMervin Praison 32537a227e fix: resolve litellm.BadRequestError with unrecognized output_json/output_pydantic parameters
- Fixed bug in _build_completion_params method where output_json and output_pydantic
  parameters were conditionally removed only when not None
- These custom parameters are now always removed before LiteLLM calls to prevent
  "Unrecognized request arguments supplied" errors
- Maintains backward compatibility and preserves all existing functionality
- Addresses issue #905

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 16:43:03 +00:00
Mervin Praison f83ffcdafe Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-14 17:27:52 +01:00
Mervin Praison 519aebd271 chore: bump praisonai to v2.2.56 and praisonaiagents to v0.0.130 across all files 2025-07-14 17:27:51 +01:00
github-actions[bot]andMervin Praison fb4ecf1893 fix: Address reviewer feedback on self-reflection test files
- Move test files from src/praisonai-agents/ to tests/ directory per repository guidelines
- Replace print statements and return values with proper assert statements
- Fix f-string issues without placeholders
- Add comprehensive exception handling for test failures
- Create improved test structure following repository patterns

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 14:55:27 +00:00
github-actions[bot]andMervin Praison 5a4cbf8aeb fix: improve robustness of Gemini model detection in ImageAgent
- Use self.llm_model property instead of self.llm for more robust model name resolution
- Handles all LLM configuration types (strings, dicts, custom instances, base_url configs)
- Fixes potential AttributeError when self.llm is not set
- Maintains backward compatibility for all existing configurations

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 14:54:16 +00:00
github-actions[bot]andMervin Praison 885a73f697 fix: add intelligent final answer detection to async method
Apply the same intelligent final answer detection logic that was added to the sync method (get_response) to the async method (get_response_async). This ensures consistent behavior between sync and async methods when detecting substantive final answers alongside tool calls.

The fix prevents redundant tool execution loops in async contexts by checking if the LLM provides substantive content (>10 characters) after tool execution, breaking the loop when appropriate.

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 14:48:03 +00:00
Mervin Praison 9336f15e67 Merge pull request #900 from Dhivya-Bharathy/add-Notion_Workspace_Assistant
feat: Add Notion_Workspace_Assistant notebook
2025-07-14 15:47:31 +01:00
Mervin Praison 6d55cfa87e Merge branch 'main' into claude/issue-896-20250714_104106 2025-07-14 15:46:44 +01:00
Mervin Praison 0b431b1fe1 Merge pull request #894 from MervinPraison/claude/issue-876-20250714_062122
fix: handle single function as tools parameter to prevent iteration error
2025-07-14 15:43:51 +01:00
github-actions[bot]andMervin Praison e252f86ac9 fix: self-reflection now works with tools
- Fixed issue where self-reflection failed when tools were present
- Added tools=formatted_tools parameter to reflection regeneration calls
- Ensures tools remain available throughout the reflection process
- Maintains backward compatibility with existing code
- Resolves issue #901

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 14:43:12 +00:00
Mervin Praison ed395de6f3 Merge pull request #898 from Dhivya-Bharathy/add-Natural_Language_Codebase
Add Natural_Language_Codebase notebook
2025-07-14 15:42:26 +01:00
Mervin Praison 633469d12e Merge pull request #886 from MervinPraison/claude/issue-877-20250714_005101
fix: ensure callbacks work regardless of verbose setting
2025-07-14 15:41:40 +01:00
Mervin Praison c5c57dff26 Merge pull request #891 from MervinPraison/claude/issue-870-20250714_062116
feat: add support for Gemini embedding models
2025-07-14 15:40:47 +01:00
Mervin Praison 573b9efcc1 Rename docs/gemini-embeddings.md to examples/python/memory/gemini-embeddings.md 2025-07-14 15:40:21 +01:00
Mervin Praison dc4e51ac81 Merge pull request #883 from MervinPraison/claude/issue-872-20250714_005257
fix: enable true parallel execution for async tasks in sequential mode
2025-07-14 15:38:00 +01:00
Mervin Praison 08e211c693 Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-14 15:33:32 +01:00
Mervin Praison abd8021925 chore: increase GitHub action comment delay from 20s to 30s 2025-07-14 15:33:32 +01:00
Mervin Praison 8a6c4eec7f Rename examples/python/tools/searxng-search.py to examples/python/tools/searxng/searxng-search.py 2025-07-14 15:31:47 +01:00
Mervin Praison 4e77a6a019 Rename examples/python/tools/trafilatura_example.py to examples/python/tools/trafilatura/trafilatura_example.py 2025-07-14 15:31:03 +01:00
Mervin Praison 41e33c09e0 Rename examples/python/tools/custom_trafilatura_tools.py to examples/python/tools/trafilatura/custom_trafilatura_tools.py 2025-07-14 15:30:54 +01:00
Mervin Praison 96f00e2172 Merge pull request #881 from MervinPraison/claude/issue-869-20250713_205041
feat: integrate Trafilatura for high-quality URL content extraction
2025-07-14 15:29:39 +01:00
github-actions[bot]andMervin Praison e858cebed9 fix: Enable self-reflection with tools in LLM class
- Fixed issue where self-reflection was not working when tools were present
- Added tools=formatted_tools parameter to completion calls during reflection regeneration
- Self-reflection now maintains tool access throughout the reflection process
- Backward compatibility maintained, no existing features removed
- Resolves issue #901

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 14:17:28 +00:00
Dhivya-Bharathy cca79bd9fc Add Notion_Workspace_Assistant notebook 2025-07-14 12:56:32 +00:00
github-actions[bot]andMervin Praison 4ffbc1b52a fix: exclude response_format parameter for Gemini image models
- Add detection logic to identify Gemini models (case-insensitive)
- Exclude response_format parameter for Gemini models to prevent litellm.UnsupportedParamsError
- Maintain backward compatibility for non-Gemini models
- Fixes #899

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 12:24:35 +00:00
DhivyaBharathy b948151a14 Update Natural_Language_Codebase.ipynb 2025-07-14 17:51:41 +05:30
Dhivya-Bharathy 11eee70d9a Add Natural_Language_Codebase notebook 2025-07-14 12:18:39 +00:00
github-actions[bot]andClaude da6a2c1c04 fix: improve tools parameter backward compatibility handling
Refactor tools parameter initialization to better handle falsy values while
maintaining single function support. Changed from explicit None check to
using ''tools or []'' pattern, which properly handles all falsy values
(None, False, 0, "", etc.) without breaking existing functionality.

This resolves backward compatibility regression where falsy values other
than None were not handled correctly in the tools parameter.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>
2025-07-14 11:59:51 +00:00
Mervin Praison 731268df5c Merge branch 'main' into claude/issue-877-20250714_005101 2025-07-14 12:55:28 +01:00
github-actions[bot]andMervin Praison d60021cb40 refactor: move Trafilatura from core to custom tool example
- Remove Trafilatura from core praisonaiagents package
- Create custom tool implementation in examples/python/tools/
- Update example to demonstrate custom tool usage with agents
- Remove dependencies from pyproject.toml
- Remove from tools __init__.py and README.md
- Keep all functionality as standalone custom tool

This change follows the request to not integrate into core package
but provide as custom tool example instead.

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 11:55:10 +00:00
Mervin Praison 75894f26c1 Merge pull request #884 from MervinPraison/claude/issue-874-20250714_005622
fix: prevent infinite loop in parallel task execution
2025-07-14 12:52:35 +01:00
Mervin Praison c53e09480a Rename test_async_sequential.py to src/praisonai-agents/test_async_sequential.py 2025-07-14 12:51:46 +01:00
2ef506fa8a feat: enhance callbacks with agent and task context (fixes #896)
- Add agent_name, agent_role, agent_tools, task_name, task_description, task_id to callbacks
- Update display_interaction() and adisplay_interaction() to accept new parameters
- Enhance LLM.get_response() to accept and pass task context
- Modify Agent.chat() to accept task parameters
- Update task execution in agents.py to pass task information
- Maintain backward compatibility with existing callbacks
- Add test scripts to verify functionality

Co-authored-by: Mervin Praison <[email protected]>

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>
2025-07-14 10:53:46 +00:00
Mervin Praison 7ce783e59b Merge branch 'main' into claude/issue-893-20250714_084251 2025-07-14 11:11:05 +01:00
Mervin Praison 95ffd0c53f chore: bump PraisonAI to 2.2.55 and PraisonAI Agents to 0.0.129 across all files 2025-07-14 11:09:35 +01:00
Mervin Praison e1c90bf569 Merge pull request #880 from MervinPraison/claude/issue-866-20250713_204435
feat: Add multi-provider/multi-model support with intelligent routing
2025-07-14 11:07:02 +01:00
Mervin Praison 5fbd8aa783 Merge pull request #892 from MervinPraison/claude/pr-879-20250714_063758
fix: Improve Ollama sequential tool calling handling
2025-07-14 11:02:12 +01:00
381c86de79 fix: resolve callback duplication and validation timing issues
- Add callback_executed tracking in llm.py to prevent duplicate callbacks
- Fix validation timing in agent.py for output_json/output_pydantic flows
- Ensure guardrail validation runs even for JSON outputs
- Improve interaction display logic to avoid redundant callbacks

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>
Co-Authored-By: Mervin Praison <[email protected]>
2025-07-14 08:52:55 +00:00
github-actions[bot]andClaude 35250e429e fix: clean up redundant conditionals and improve tool execution logic in llm.py
- Remove redundant conditional expressions that were causing unnecessary repetition
- Add intelligent final answer detection after tool execution
- Prevent unnecessary tool execution loops when LLM provides substantive response

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>
2025-07-14 08:51:02 +00:00
github-actions[bot]andClaude 3027937057 fix: prevent event loop blocking and refactor async task handling
- Replace deprecated asyncio.get_event_loop() with asyncio.get_running_loop()
- Fix event loop blocking issues in API and MCP server implementations
- Improve async handling consistency across the codebase

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>
2025-07-14 08:50:15 +00:00
github-actions[bot]andMervin Praison d79443868f refactor: Rename MultiModelAgent to RouterAgent for better clarity
- Renamed MultiModelAgent class to RouterAgent throughout codebase
- Updated multi_model_agent.py -> router_agent.py
- Updated all imports, exports, and references in:
  - src/praisonai-agents/praisonaiagents/agent/__init__.py
  - examples/python/agents/multi-provider-agent.py
  - examples/python/agents/MULTI_PROVIDER_README.md
  - src/praisonai-agents/test_multi_provider.py
- RouterAgent better reflects the primary purpose of intelligent model routing
- Maintains all existing functionality and backward compatibility

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 08:50:14 +00:00
github-actions[bot]andMervin Praison 6705e753e8 fix: Remove hardcoded regex logic from Ollama tool calling
- Remove explicitly prohibited hardcoded regex patterns
- Move import re to top of file (PEP 8 compliance)
- Extract duplicated message-building logic into helper method
- Define prompt constants at class level for maintainability
- Simplify tool result formatting without numeric extraction
- Maintain backward compatibility and existing functionality

Fixes code quality issues identified by automated review tools
while ensuring compliance with explicit PR requirements.

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 08:49:03 +00:00
Mervin Praison b4d1532512 ci: prevent bot accounts from triggering Claude workflow action 2025-07-14 08:51:51 +01:00
Mervin Praison 4b5d3b2eda ci: remove max_turns limit from Claude code review workflow 2025-07-14 08:50:09 +01:00
Mervin Praison a1d4f3ee43 feat: add repository checkout step with fetch history in Claude workflow 2025-07-14 08:23:09 +01:00
Mervin Praison b5e930ff2e Merge pull request #889 from MervinPraison/claude/issue-873-20250714_005616
fix: use LLM abstraction instead of hardcoded OpenAI in hierarchical process
2025-07-14 08:13:08 +01:00
Mervin Praison 1a6a14f8f2 Merge pull request #887 from MervinPraison/claude/issue-878-20250714_005448
fix: prevent duplicate callback invocations in LLM response generation
2025-07-14 08:11:52 +01:00
Mervin Praison ca69de4c6a Merge pull request #885 from MervinPraison/claude/issue-875-20250714_005802
fix: Support GuardrailResult return type in guardrail functions (#875)
2025-07-14 08:09:05 +01:00
Mervin Praison 9b3d36e6d7 ci: remove redundant repository checkout step from Claude workflow 2025-07-14 08:07:07 +01:00
Mervin Praison a9359a0785 Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-14 08:05:10 +01:00
Mervin Praison 4cb016c3ca feat: add Gemini structured output agent and enhance Claude workflow checkout 2025-07-14 08:05:08 +01:00
Mervin Praison d2e4776884 Merge pull request #888 from MervinPraison/claude/issue-871-20250714_005124
feat: Add Gemini structured output support
2025-07-14 07:52:18 +01:00
claude[bot]andMervin Praison 193b82aeba fix: Improve Ollama sequential tool calling handling
- Extract numeric values from tool results for better clarity
- Add explicit prompting for final answers after tool execution
- Handle empty initial responses from Ollama with retry logic
- Fix duplicate strip calls in response handling

This improves Ollama's ability to:
1. Understand numeric values from tool results
2. Generate proper final answers instead of returning None
3. Avoid getting stuck in infinite loops

Fixes the issue where Ollama was unable to properly chain tool calls
for sequential operations like getting a stock price and multiplying it.

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:42:14 +00:00
claude[bot]andMervin Praison 029de283c4 refactor: extract duplicate workflow completion logic into helper method
- Created _check_all_tasks_completed() helper method to eliminate code duplication
- Both async and sync workflow methods now use the same completion check logic
- No functional changes, pure refactoring for better maintainability

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:40:00 +00:00
Mervin Praison 26376a77af Merge pull request #890 from Dhivya-Bharathy/grok4-agents-batch-upload
feat: Add 10 Grok4 Intelligent Agent Examples
2025-07-14 07:31:17 +01:00
claude[bot]andMervin Praison 7cb580bb12 fix: handle single function as tools parameter to prevent iteration error
- Fix LLM._log_llm_config to properly check if tools is a list/tuple before iterating
- Fix Agent.__init__ to wrap single callable tools in a list automatically
- Resolves "'function' object is not iterable" error when using LLMGuardrail
- Maintains full backward compatibility

Fixes #876

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:30:54 +00:00
claude[bot]andMervin Praison 06b458cc54 test: add test script for async task parallel execution
- Tests that async tasks run in parallel in sequential mode
- Verifies sync tasks wait for async tasks to complete
- Measures execution times to confirm parallel execution
- Helps validate the async task execution fix

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:28:30 +00:00
claude[bot]andMervin Praison e29c2fbe90 fix: prevent event loop blocking and refactor async task handling
- Use run_in_executor for sync tasks in both sequential and hierarchical processes
- Refactor duplicated async task gathering logic into flush_async_tasks helper
- Update to asyncio.get_running_loop() from deprecated get_event_loop()
- Maintains backward compatibility while fixing performance issues

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:28:21 +00:00
claude[bot]andMervin Praison 76032be618 feat: add support for Gemini embedding models
- Extract embedding model from embedder configuration in Memory class
- Replace hardcoded "text-embedding-3-small" with configurable model
- Support any embedding model available through LiteLLM
- Maintain backward compatibility (defaults to OpenAI model)
- Add documentation and examples for Gemini embeddings
- Update memory_example.py to show configuration options

Fixes #870

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:27:18 +00:00
claude[bot]andMervin Praison 80f5ba940b fix: address code review issues for Trafilatura integration
- Fix asyncio blocking by using asyncio.to_thread() for all synchronous calls in async functions
- Remove dead code: module-level __call__ function that would never be invoked
- Replace global variable with module-level trafilatura import for cleaner code
- Remove unused json import in test file
- Improve error detection from fragile string check to robust pattern
- Add clarifying comments about safe circular imports due to lazy loading

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:26:49 +00:00
devops-vestradata 65791781e2 Add 10 grok4 intelligent agent files 2025-07-14 11:56:23 +05:30
claude[bot]andMervin Praison ee66154b73 fix: ensure interaction_displayed flag prevents all duplicate callbacks
- Added missing interaction_displayed checks to reasoning steps display
- Added missing checks to self-reflection display logic
- Ensured flag is set after each display interaction
- Fixed both sync and async methods
- Added comprehensive test to verify the fix

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:24:34 +00:00
claude[bot]andMervin Praison 0dc51060d8 fix: resolve callback duplication and validation timing issues
- Remove callback execution from display functions to prevent duplication when verbose=True
- Add missing callback execution after streaming completes when verbose=False
- Move callbacks to execute AFTER guardrail validation to prevent inconsistent state
- Add _execute_callback_and_display helper method to reduce code duplication
- Ensure callbacks work consistently regardless of verbose setting

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:23:51 +00:00
claude[bot]andMervin Praison 2daae1f464 fix: Address critical issues with Gemini structured output implementation
- Add hasattr check before calling model_json_schema() to prevent AttributeError
- Centralize Gemini model detection logic with new _is_gemini_model() method
- Fix critical bug: Always remove output_json/output_pydantic from params before passing to litellm
- Fix model recognition to handle provider-prefixed models (e.g., 'google/gemini-1.5-pro')
- Ensure backward compatibility for all existing models

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:23:29 +00:00
claude[bot]andMervin Praison 81bcda0440 fix: correct LLM interface usage and improve code quality
- Remove duplicate output_json=True when using output_pydantic
- Add helper methods to reduce code duplication
- Improve test file to use proper assertions instead of sys.exit()
- Add unit tests for the new helper methods
- Fix TypeError handling in JSON parsing

This ensures the hierarchical workflow works correctly with all LLM providers.

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:23:13 +00:00
claude[bot]andMervin Praison e269cc7bf2 fix: Improve guardrail validation and add missing type hints
- Fixed critical bug: validation now properly checks for GuardrailResult type instead of accepting any class with the same name
- Added missing TaskOutput type hints in example file
- Refactored conditional logic for better readability
- Maintains full backward compatibility

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 06:21:30 +00:00
Mervin Praison 66e5dfcb41 refactor: update Claude GitHub Action with simplified OAuth authentication and improved triggers 2025-07-14 03:08:02 +01:00
Mervin Praison ae3e69a286 feat: exclude @gemini triage command from issue review workflow 2025-07-14 02:48:11 +01:00
Mervin Praison 40ea6fe1aa ci: simplify Gemini trigger command from '@gemini review' to '@gemini' 2025-07-14 02:46:58 +01:00
Mervin Praison e30a7564be Merge pull request #879 from MervinPraison/claude/issue-854-20250713_205120
fix: Implement Ollama sequential tool calling fix
2025-07-14 02:20:07 +01:00
claude[bot]andMervin Praison fbd0a736f0 fix: Fix critical issues in multi-provider implementation
- Changed TaskComplexity from string Enum to IntEnum for proper ordering comparisons
- Fixed manual routing logic to use llm_model property correctly
- Removed private attribute access by storing api_key and base_url locally
- Improved context size estimation from char count to token approximation
- Removed unused imports (Callable, TaskComplexity, ModelProfile)
- Fixed unnecessary f-string in error message
- Added TODO for token tracking implementation
- Fixed inconsistent model names in documentation

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 01:16:18 +00:00
claude[bot]andMervin Praison eea9f757af fix: use LLM abstraction instead of hardcoded OpenAI in hierarchical process
- Replace direct OpenAI client instantiation with LLM class
- Support all LLM providers (Gemini, Anthropic, etc.) in manager tasks
- Remove unused OpenAI imports from process.py
- Fix issue #873 where hierarchical workflow required OpenAI API key

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 01:08:57 +00:00
claude[bot]andMervin Praison 02a8f075a6 fix: prevent duplicate callback invocations in LLM response generation
- Add interaction_displayed flag to track display_interaction calls
- Prevent duplicate callbacks in both sync and async get_response methods
- Fix addresses issue where callbacks fired twice for single task execution
- Maintain backward compatibility with no API changes

Fixes #878

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 01:08:54 +00:00
claude[bot]andMervin Praison 57c39653d6 feat: Add Gemini structured output support
- Add Gemini models to MODELS_SUPPORTING_STRUCTURED_OUTPUTS
- Implement native Gemini structured output using response_schema
- Maintain backward compatibility for OpenAI and other models
- Update _build_messages to skip prompt modification for Gemini
- Add response_mime_type and response_schema parameters for Gemini
- Pass output_json/output_pydantic through all completion calls

Fixes #871

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 01:06:59 +00:00
claude[bot]andMervin Praison 2f21d0f907 fix: Support GuardrailResult return type in guardrail functions (#875)
- Modified task.py to accept both GuardrailResult and Tuple[bool, Any] return types
- Updated _process_guardrail to handle GuardrailResult directly without conversion
- Added comprehensive test cases for both return types
- Updated documentation with GuardrailResult examples
- Maintains backward compatibility with existing Tuple[bool, Any] guardrails

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 01:05:39 +00:00
claude[bot]andMervin Praison e0caaf8421 fix: enable true parallel execution for async tasks in sequential mode
Previously, async tasks in sequential mode were awaited one-by-one, defeating 
the purpose of async execution. This fix collects async tasks and runs them 
in parallel using asyncio.gather(), ensuring:

- Async tasks run concurrently as intended
- Sync tasks still execute in order after async tasks complete
- Proper task completion and dependency handling

Fixes #872

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 01:05:27 +00:00
claude[bot]andMervin Praison 4bc35c3b6e test: add test scripts for callback fix verification
- test_callback_fix.py: Simple test to verify callbacks work with verbose=False
- test_issue_877.py: Comprehensive test reproducing the exact issue scenario
- Both tests verify that callbacks are executed regardless of verbose setting

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 01:05:20 +00:00
claude[bot]andMervin Praison 4fd9b863a6 fix: ensure callbacks work regardless of verbose setting
- Added execute_sync_callback helper function to trigger callbacks without display
- Updated LLM class to always execute interaction callbacks when responses are generated
- Updated Agent class to ensure callbacks work for direct OpenAI client usage
- Maintains backward compatibility - display output still controlled by verbose flag
- Fixes issue #877 where callbacks only worked when verbose=True

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 01:05:13 +00:00
claude[bot]andMervin Praison fc02392ddb fix: prevent infinite loop in parallel task execution by exiting workflow when all tasks complete
When all tasks are completed and workflow_finished is set to True, the workflow now immediately exits to prevent tasks from being reset to "not started" state. This fixes the issue where parallel tasks would repeat in a loop and the aggregator task would never run.

Fixes #874

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 01:03:39 +00:00
claude[bot]andMervin Praison 4564639f18 enhance: Improve Ollama provider detection
- Add check for "ollama" in base_url for custom configurations
- Support detection of custom Ollama ports beyond 11434
- More robust detection for various deployment scenarios

Co-authored-by: Mervin Praison <[email protected]>
2025-07-14 00:58:57 +00:00
claude[bot]andMervin Praison 16598104ab feat: integrate Trafilatura for high-quality URL content extraction
- Add trafilatura_tools.py with extract_content, extract_metadata, extract_text_only, and compare_extraction functions
- Register Trafilatura tools in __init__.py for easy imports
- Add trafilatura and newspaper3k as optional [web] dependencies
- Include comprehensive URL validation to prevent SSRF attacks
- Update tools README with Trafilatura documentation
- Add detailed example file showing various usage patterns
- Create test script for verification
- Maintain backward compatibility with existing URL extraction tools

Resolves #869

Co-authored-by: Mervin Praison <[email protected]>
2025-07-13 21:05:30 +00:00
claude[bot]andMervin Praison b5d13a2b7f fix: Implement Ollama sequential tool calling fix
Modified conversation format for Ollama to handle tool results properly:
- Assistant messages no longer include 'tool_calls' field for Ollama
- Tool results are passed as user messages with natural language format
- Prevents infinite loop of repeated tool calls

This allows Ollama to properly execute sequential tool calls as intended.

Fixes #854

Co-authored-by: Mervin Praison <[email protected]>
2025-07-13 20:55:34 +00:00
claude[bot]andMervin Praison 4622bcda43 feat: Add multi-provider/multi-model support with intelligent routing
- Add ModelRouter for intelligent model selection based on task complexity
- Create MultiModelAgent class supporting multiple models per agent
- Implement routing strategies: auto, cost-optimized, performance-optimized
- Add pre-configured profiles for 8+ popular models
- Include usage tracking and cost estimation
- Maintain full backward compatibility
- Add comprehensive examples and documentation

Fixes #866

Co-authored-by: Mervin Praison <[email protected]>
2025-07-13 20:53:06 +00:00
Mervin Praison 31bae3a231 chore: bump PraisonAI to 2.2.54 and update Gemini workflows 2025-07-13 21:49:21 +01:00
Mervin Praison 466bcfedd7 Merge pull request #863 from MervinPraison/gemini/issue-862-fix-async-loop
Fix #862: Agent gets stuck in infinite loop when running multiple async tasks
2025-07-13 20:46:57 +01:00
Mervin Praison c24bf80314 Merge pull request #858 from MervinPraison/gemini/issue-854-fix-ollama-sequential-1690000000
Fix #854: Ollama sequential still failing
2025-07-13 20:46:21 +01:00
Mervin Praison c36b3aee6e Merge pull request #865 from MervinPraison/gemini/issue-864-fix-parallel-task-loop
Fix #864: Fix parallel task loop
2025-07-13 20:45:55 +01:00
Mervin Praison 08512ba6d3 Merge pull request #861 from MervinPraison/gemini/issue-860-fix-hierarchical-llm-fallback
Fix #860: Hierarchical Process LLM Fallback
2025-07-13 20:43:53 +01:00
claude[bot]andMervin Praison 63ebfc0b54 fix: Complete removal of Ollama special handling for async method
This completes the fix for issue #854 by removing the Ollama special
handling from the async method (get_response_async) to match the sync
method fix. Also removes the now-unused _handle_ollama_model method.

The special handling was causing infinite loops in sequential tool
calling. By removing it, Ollama models now work correctly through
the standard litellm interface.

Co-authored-by: Mervin Praison <[email protected]>
2025-07-13 19:21:40 +00:00
claude[bot]andMervin Praison b29269071d fix: properly handle parallel task reset logic to prevent infinite loops
- Uncommented the task reset logic that was causing issue #864
- Added check for async_execution to prevent resetting parallel tasks
- Preserves backward compatibility for non-parallel tasks that need rerun
- Fixes infinite loop issue while maintaining existing functionality

Co-authored-by: Mervin Praison <[email protected]>
2025-07-13 19:14:59 +00:00
Gemini Bot b6a86bf348 Gemini: [Issue #864] Fix parallel task loop 2025-07-13 12:49:40 +00:00
Gemini Bot 02ba71be3d Gemini: [Issue #862] Fix infinite loop in async workflow 2025-07-13 12:17:37 +00:00
Gemini Bot e60f1c1aa1 Gemini: [Issue #860] Fix hierarchical process LLM fallback 2025-07-13 11:31:08 +00:00
Mervin Praison 450f237805 feat: add description field to TaskOutput in validation feedback tests 2025-07-13 01:17:26 +01:00
Mervin Praison 66e53fc82e refactor: move validation test files to src directory and update LLM model references 2025-07-13 01:09:52 +01:00
Mervin Praison 7e6ee5f9f2 chore: enhance variable handling in PR review workflow for improved clarity 2025-07-13 01:04:41 +01:00
Mervin Praison b669b9815d fix: correct variable interpolation syntax in PR review workflow 2025-07-13 00:50:39 +01:00
Mervin Praison 6d89cd603e chore: remove debug workflow trigger step from Gemini PR review action 2025-07-13 00:44:38 +01:00
Mervin Praison 6e955b5941 feat: restrict Gemini PR review to repo owners, members and collaborators 2025-07-13 00:34:48 +01:00
Mervin Praison 87a67b5f5d refactor: improve PR checkout logic in Gemini review workflow using dedicated step 2025-07-13 00:25:08 +01:00
Mervin Praison baec9cd0a4 fix: update checkout ref to handle both PR and issue comment triggers 2025-07-13 00:23:16 +01:00
Mervin Praison e144dc497d feat: allow all users to trigger Gemini PR review and add debug logging 2025-07-13 00:19:06 +01:00
Gemini Bot 8a77ab0ab5 Gemini: [Issue #854] Fix Ollama sequential tool calling loop 2025-07-12 23:10:23 +00:00
Mervin Praison 0034b5c964 Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-13 00:04:14 +01:00
Mervin Praison 543c627374 feat: expand GitHub Actions permissions to include PR list and granular git commands 2025-07-13 00:04:13 +01:00
Mervin Praison 93154adf9e Merge pull request #857 from MervinPraison/gemini/issue-854-fix-ollama-sequential-fail
Fix #854: Ollama sequential tool calling failing
2025-07-13 00:03:55 +01:00
Gemini Bot 3ab2462d96 Gemini: [Issue #854] Fix Ollama sequential tool calling loop 2025-07-12 23:00:05 +00:00
Mervin Praison aecf65a130 Merge branch 'main' of https://github.com/MervinPraison/PraisonAI 2025-07-12 23:52:47 +01:00
Mervin Praison 0a453d30a2 chore: add .gemini and issue_details.json to gitignore 2025-07-12 23:52:46 +01:00
Mervin Praison 354a2c717b Merge pull request #856 from MervinPraison/gemini/issue-854-fix-ollama-sequential
Fix #854: Ollama sequential tool calling failing
2025-07-12 23:52:38 +01:00
Mervin Praison 62514fbef4 Delete issue_details.json 2025-07-12 23:51:43 +01:00
Mervin Praison f3c77a2e8b Delete .gemini/settings.json 2025-07-12 23:51:32 +01:00
Gemini Bot 50dde4fa02 Gemini: [Issue #854] Fix Ollama sequential tool calling 2025-07-12 22:45:09 +00:00
Mervin Praison bbc4530329 chore: bump version to 2.2.53 and update dependencies across package files 2025-07-12 23:43:01 +01:00
Mervin Praison aab975bec7 Merge pull request #851 from MervinPraison/claude/issue-846-20250712_220811
fix: handle None response_text in Ollama sequential tool calls
2025-07-12 23:28:38 +01:00
Mervin Praison 50c50fa567 Merge pull request #849 from MervinPraison/claude/issue-845-20250712_183232
fix: enable sequential tool execution in OpenAI client
2025-07-12 23:27:54 +01:00
claude[bot]andMervin Praison 68f280634a refactor: remove dead code from sequential tool execution logic
- Remove unused should_continue variable and related logic
- Simplify code by removing special handling for sequentialthinking tool
- Add clear comments explaining the sequential execution behavior

Co-authored-by: Mervin Praison <[email protected]>
2025-07-12 22:16:05 +00:00
Mervin Praison 346488fbec Delete test_sequential_fix.py 2025-07-12 23:15:10 +01:00
claude[bot]andMervin Praison 57b5a1fd8d fix: handle None response_text in Ollama sequential tool calls
- Add None checks before calling .strip() on response_text
- Fixes 'NoneType' object has no attribute 'strip' error with Ollama models
- Applied pattern: response_text.strip() if response_text else ""
- Fixed 14 locations where .strip() was called without None checks
- Ensures backward compatibility by returning empty string instead of None

Fixes issue #846

Co-authored-by: Mervin Praison <[email protected]>
2025-07-12 22:11:55 +00:00
claude[bot]andMervin Praison d75a5bb728 test: add assertions to test_sequential_fix.py for automated testing
- Add proper assertions to verify sequential tool execution results
- Improve multiply function docstring with complete Args and Returns sections
- Add success message when all assertions pass

Co-authored-by: Mervin Praison <[email protected]>
2025-07-12 22:10:09 +00:00
Mervin Praison 519ecc025a feat: update allowed commands in GitHub workflow files for Gemini reviews 2025-07-12 21:21:23 +01:00
Mervin Praison 8290b95132 refactor: update issue review prompt with clearer instructions and dynamic commands 2025-07-12 21:01:13 +01:00
Mervin Praison 3d95e1ff0e feat: add read_file permission to Gemini PR review workflow 2025-07-12 20:55:41 +01:00
Mervin Praison 74405d1bbe feat: add read_file permission to Gemini issue review workflow 2025-07-12 20:54:59 +01:00
Mervin Praison d307c3857c feat: add environment variables for Gemini CLI action in GitHub workflow 2025-07-12 20:54:31 +01:00
Mervin Praison fbe5dddaae chore: disable telemetry in GitHub workflow configurations 2025-07-12 20:45:20 +01:00
Mervin Praison 7e4108918d fix: move GH_TOKEN to local env scope in GitHub workflow 2025-07-12 20:40:04 +01:00
Mervin Praison da7e38100e feat: add GitHub workflow for automated issue review using Gemini AI 2025-07-12 20:36:37 +01:00
Mervin Praison 848459b816 feat: add edit, web-fetch, web-search and memoryTool permissions while removing telemetry setup 2025-07-12 20:35:19 +01:00
Mervin Praison cd63845398 feat: add git command to allowed shell commands in PR review workflow 2025-07-12 20:31:18 +01:00
Mervin Praison 7827427dc0 feat: move Gemini PR review action to dedicated directory with configuration 2025-07-12 20:16:47 +01:00
Mervin Praison 0e378d0754 feat: add GitHub workflows for Gemini issue triage and PR review automation 2025-07-12 20:03:05 +01:00
Mervin Praison 206022eb9c refactor: simplify auto-comment workflows by removing multi-agent review requirement 2025-07-12 19:48:23 +01:00
Mervin Praison 901ba8074d Delete test_sequential_fix.py 2025-07-12 19:40:42 +01:00
claude[bot]andMervin Praison 953ee6e7bd fix: enable sequential tool execution in OpenAI client
Removes premature loop termination after tool execution to allow
sequential tool calls. The model can now see tool results and make
additional tool calls as needed.

Fixes #845

Co-authored-by: Mervin Praison <[email protected]>
2025-07-12 18:36:00 +00:00
209 changed files with 18425 additions and 4049 deletions

No files matched your search

+81
View File
@@ -0,0 +1,81 @@
name: 'Gemini CLI Action'
description: 'Uses the Gemini CLI to automate software development tasks.'
author: 'Gemini'
inputs:
prompt:
description: 'A specific prompt to guide Gemini.'
required: false
default: 'You are a helpful assistant.'
GEMINI_API_KEY:
description: 'Your Gemini API key.'
required: true
OTLP_GCP_WIF_PROVIDER:
description: 'The workload identity provider for GCP authentication.'
required: false
OTLP_GOOGLE_CLOUD_PROJECT:
description: 'The Google Cloud project for telemetry.'
required: false
settings_json:
description: |
A JSON string to configure the Gemini CLI. This will be written to .gemini/settings.json.
To enable observability with Google Cloud, include:
{"telemetry": {"enabled": true, "target": "gcp"}, "sandbox": false}
required: false
version:
description: |
The version of @google/gemini-cli to execute.
Can be a specific version from npm (e.g., '0.1.0', 'latest'), a branch name (e.g., 'main'), or a commit hash.
required: false
default: 'latest'
runs:
using: "composite"
steps:
- name: Configure Gemini CLI
if: inputs.settings_json != ''
run: |
mkdir -p .gemini
echo "$SETTINGS_JSON" > .gemini/settings.json
shell: bash
env:
SETTINGS_JSON: ${{ inputs.settings_json }}
- name: Authenticate to Google Cloud for Telemetry
if: inputs.OTLP_GCP_WIF_PROVIDER != ''
id: auth
uses: 'google-github-actions/auth@v2'
with:
project_id: ${{ inputs.OTLP_GOOGLE_CLOUD_PROJECT }}
workload_identity_provider: ${{ inputs.OTLP_GCP_WIF_PROVIDER }}
- name: Install Gemini CLI
id: install
run: |
if [[ "${{ inputs.version }}" == "latest" || "${{ inputs.version }}" =~ ^[0-9]+\.[0-9]+\.[0-9]+(-[a-zA-Z0-9\.-]+)?(\+[a-zA-Z0-9\.-]+)?$ ]]; then
echo "Installing Gemini CLI from npm: @google/gemini-cli@${{ inputs.version }}"
npm install -g @google/gemini-cli@${{ inputs.version }}
else
echo "Installing Gemini CLI from GitHub: github:google-gemini/gemini-cli#${{ inputs.version }}"
npm install -g github:google-gemini/gemini-cli#${{ inputs.version }}
fi
echo "Verifying installation:"
if command -v gemini >/dev/null 2>&1; then
gemini --version || echo "Gemini CLI installed successfully (version command not available)"
else
echo "Error: Gemini CLI not found in PATH"
exit 1
fi
shell: bash
- name: Run Gemini CLI
run: |
gemini --yolo --prompt "$PROMPT"
shell: bash
env:
GEMINI_API_KEY: ${{ inputs.GEMINI_API_KEY }}
PROMPT: ${{ inputs.prompt }}
branding:
icon: 'terminal'
color: 'blue'
+6 -3
View File
@@ -1,4 +1,4 @@
name: Auto Issue Comment
name: Auto Issue Comment with PR Creation
on:
issues:
@@ -9,6 +9,9 @@ jobs:
runs-on: ubuntu-latest
permissions:
issues: write
contents: write
pull-requests: write
actions: read
steps:
- name: Add automated comment
uses: actions/github-script@v7
@@ -19,5 +22,5 @@ jobs:
issue_number: context.issue.number,
owner: context.repo.owner,
repo: context.repo.repo,
body: '@claude review this issue and do a detailed analysis and fix this if the existing code doesn\'t have the solution implemented. Making sure it has backward compatibility, no existing features removed. After making those changes, again use multi agents to review the applied changes. Use @web to search if you dont know any information or to find the latest documentation or to find the latest version. Run the code if you think you need to run it to test it. Minimal code change to start with if required any changes.'
})
body: '@claude review this issue and do a detailed analysis and fix this if the existing code doesn\'t have the solution implemented. Making sure it has backward compatibility, no existing features removed. After making those changes, again review the applied changes. Use @web to search if you dont know any information or to find the latest documentation or to find the latest version. Run the code if you think you need to run it to test it. Minimal code change to start with if required any changes. **Please create a PR using gh tool with your changes**.'
})
+1 -1
View File
@@ -19,5 +19,5 @@ jobs:
issue_number: context.issue.number,
owner: context.repo.owner,
repo: context.repo.repo,
body: '@claude review this pull request and do a detailed analysis and fix this if the existing code doesn\'t have the solution implemented. Making sure it has backward compatibility, no existing features removed. After making those changes, again use multi agents to review the applied changes. Use @web to search if you dont know any information or to find the latest documentation or to find the latest version. Run the code if you think you need to run it to test it. Minimal code change to start with if required any changes.'
body: '@claude review this pull request and do a detailed analysis and fix this if the existing code doesn\'t have the solution implemented. Making sure it has backward compatibility, no existing features removed. After making those changes, again review the applied changes. Use @web to search if you dont know any information or to find the latest documentation or to find the latest version. Run the code if you think you need to run it to test it. Minimal code change to start with if required any changes.'
})
+78 -38
View File
@@ -1,4 +1,4 @@
name: Claude Code
name: Claude Assistant
on:
issue_comment:
@@ -6,41 +6,60 @@ on:
pull_request_review_comment:
types: [created]
issues:
types: [opened, assigned]
types: [opened, assigned, labeled]
pull_request_review:
types: [submitted]
jobs:
claude:
claude-response:
# Prevent bots from triggering this action
if: |
(github.event_name == 'issue_comment' && contains(github.event.comment.body, '@claude')) ||
(github.event_name == 'pull_request_review_comment' && contains(github.event.comment.body, '@claude')) ||
(github.event_name == 'pull_request_review' && contains(github.event.review.body, '@claude')) ||
(github.event_name == 'issues' && (contains(github.event.issue.body, '@claude') || contains(github.event.issue.title, '@claude')))
github.actor != 'dependabot[bot]' &&
github.actor != 'github-actions[bot]' &&
github.actor != 'cursor[bot]' &&
!contains(github.actor, '[bot]') &&
github.actor != 'renovate[bot]'
runs-on: ubuntu-latest
permissions:
contents: read
pull-requests: read
issues: read
id-token: write
contents: write
pull-requests: write
issues: write
actions: read
steps:
- name: Checkout repository
uses: actions/checkout@v4
with:
fetch-depth: 1
- name: Run Claude Code (OAuth)
id: claude
uses: grll/claude-code-action@beta
fetch-depth: 0
- uses: anthropics/claude-code-action@beta
with:
use_oauth: true
claude_access_token: ${{ secrets.CLAUDE_ACCESS_TOKEN }}
claude_refresh_token: ${{ secrets.CLAUDE_REFRESH_TOKEN }}
claude_expires_at: ${{ secrets.CLAUDE_EXPIRES_AT }}
claude_code_oauth_token: ${{ secrets.CLAUDE_ACCESS_TOKEN }}
github_token: ${{ secrets.GITHUB_TOKEN }}
trigger_phrase: "@claude"
allowed_tools: |
Bash(git:*)
Bash(python:*)
Bash(pip:*)
Bash(conda:*)
Bash(pytest:*)
Bash(gh:*)
Bash(python -m pytest:*)
Bash(python -m pip:*)
Bash(poetry:*)
View
GlobTool
GrepTool
BatchTool
Edit
Replace
mcp__github__get_issue
mcp__github__get_issue_comments
mcp__github__update_issue
timeout_minutes: 30
post-review-comment:
needs: claude
if: success() && needs.claude.result == 'success'
needs: claude-response
if: success() && needs.claude-response.result == 'success'
runs-on: ubuntu-latest
permissions:
contents: read
@@ -53,7 +72,7 @@ jobs:
with:
github-token: ${{ secrets.GH_TOKEN }}
script: |
const reviewText = '@claude review if the changes made are as per the requirements';
const reviewText = '@claude review if the changes made in the above branch are as per the requirements and make changes to the above branch if issues still exist.';
let comments = [];
let issueNumber;
@@ -75,22 +94,34 @@ jobs:
issue_number: issueNumber
});
const existingComment = comments.data.find(comment =>
// Look for Claude's completed review comments
const claudeReviewComments = comments.data.filter(comment =>
(comment.user.login === 'claude[bot]' || comment.body.includes('✅') || comment.body.includes('## Summary')) &&
!comment.body.includes('⏳') &&
!comment.body.includes('🔄') &&
!comment.body.includes('Loading...')
);
// Check if our review request comment already exists
const existingReviewRequest = comments.data.find(comment =>
comment.body.includes(reviewText)
);
const shouldPost = !existingComment;
// Post if Claude has completed a review AND we haven't posted the review request yet
const shouldPost = claudeReviewComments.length > 0 && !existingReviewRequest;
core.setOutput('should-post', shouldPost);
console.log(`Claude review completed: ${claudeReviewComments.length > 0}`);
console.log(`Review request exists: ${!!existingReviewRequest}`);
console.log(`Should post comment: ${shouldPost}`);
} catch (error) {
console.log('Error checking comments:', error);
core.setOutput('should-post', true); // If we can't check, proceed with posting
core.setOutput('should-post', false); // If we can't check, don't post
}
- name: Wait 20 seconds before posting comment
- name: Wait 30 seconds before posting comment
if: steps.check-comment.outputs.should-post == 'true'
run: sleep 20
run: sleep 30
- name: Post review comment
if: steps.check-comment.outputs.should-post == 'true'
@@ -99,23 +130,32 @@ jobs:
github-token: ${{ secrets.GH_TOKEN }}
script: |
let issueNumber;
let isPR = false;
// Determine issue/PR number based on event type
// Determine issue/PR number and type based on event type
if (context.eventName === 'pull_request_review_comment' || context.eventName === 'pull_request_review') {
issueNumber = context.payload.pull_request?.number || context.issue.number;
isPR = true;
} else if (context.eventName === 'issue_comment' && context.payload.issue.pull_request) {
// This is a comment on a PR (not an issue)
issueNumber = context.payload.issue.number;
isPR = true;
} else {
// This is a regular issue
// This is a regular issue - don't post review comment
issueNumber = context.issue.number;
isPR = false;
}
console.log(`Posting review comment on issue/PR #${issueNumber}`);
await github.rest.issues.createComment({
issue_number: issueNumber,
owner: context.repo.owner,
repo: context.repo.repo,
body: '@claude review if the changes made are as per the requirements'
});
// Only post review comment on PRs, not regular issues
if (isPR) {
console.log(`Posting review comment on PR #${issueNumber}`);
await github.rest.issues.createComment({
issue_number: issueNumber,
owner: context.repo.owner,
repo: context.repo.repo,
body: '@claude review if the changes made in the above branch are as per the requirements and make changes to the above branch if issues still exist.'
});
} else {
console.log(`Skipping review comment for regular issue #${issueNumber}`);
}
+1
View File
@@ -40,5 +40,6 @@ jobs:
uses: docker/build-push-action@v2
with:
context: .
file: ./docker/Dockerfile
push: true
tags: ghcr.io/mervinpraison/praisonai:latest
@@ -0,0 +1,76 @@
name: Gemini Automated Issue Triage
on:
issue_comment:
types: [created]
jobs:
triage-issue:
if: >
github.event_name == 'issue_comment' &&
!github.event.issue.pull_request &&
contains(github.event.comment.body, '@gemini triage') &&
(github.event.comment.author_association == 'OWNER' ||
github.event.comment.author_association == 'MEMBER' ||
github.event.comment.author_association == 'COLLABORATOR')
timeout-minutes: 5
permissions:
issues: write
contents: read
id-token: write
concurrency:
group: ${{ github.workflow }}-${{ github.event.issue.number }}
cancel-in-progress: true
runs-on: ubuntu-latest
steps:
- name: Generate GitHub App Token
id: generate_token
uses: actions/create-github-app-token@v1
with:
app-id: ${{ secrets.APP_ID }}
private-key: ${{ secrets.PRIVATE_KEY }}
- name: Checkout repository
uses: actions/checkout@v4
with:
token: ${{ steps.generate_token.outputs.token }}
- name: Run Gemini Issue Triage
uses: google-gemini/gemini-cli-action@main
env:
GITHUB_TOKEN: ${{ steps.generate_token.outputs.token }}
ISSUE_TITLE: ${{ github.event.issue.title }}
ISSUE_BODY: ${{ github.event.issue.body }}
ISSUE_NUMBER: ${{ github.event.issue.number }}
REPOSITORY: ${{ github.repository }}
with:
version: 0.1.8-rc.0
GEMINI_API_KEY: ${{ secrets.GEMINI_API_KEY }}
OTLP_GCP_WIF_PROVIDER: ${{ secrets.OTLP_GCP_WIF_PROVIDER }}
OTLP_GOOGLE_CLOUD_PROJECT: ${{ secrets.OTLP_GOOGLE_CLOUD_PROJECT }}
settings_json: |
{
"coreTools": [
"run_shell_command(gh label list)",
"run_shell_command(gh issue edit)"
],
"telemetry": {
"enabled": true,
"target": "gcp"
},
"sandbox": false
}
prompt: |
You are an issue triage assistant. Analyze the current GitHub issue and apply the most appropriate existing labels.
Steps:
1. Run: `gh label list --limit 100` to get all available labels.
2. Review the issue title and body provided in the environment variables.
3. Select the most relevant labels from the existing labels, focusing on kind/*, area/*, and priority/*.
4. Apply the selected labels to this issue using: `gh issue edit ISSUE_NUMBER --add-label "label1,label2"`
Guidelines:
- Only use labels that already exist in the repository.
- Do not add comments or modify the issue content.
- Triage only the current issue.
- Assign all applicable kind/*, area/*, and priority/* labels based on the issue content.
+198
View File
@@ -0,0 +1,198 @@
name: Gemini Issue Review
on:
workflow_dispatch:
issue_comment:
types: [created]
permissions:
contents: write
issues: write
pull-requests: read
jobs:
review-issue:
if: >
github.event_name == 'workflow_dispatch' ||
(github.event_name == 'issue_comment' &&
!github.event.issue.pull_request &&
contains(github.event.comment.body, '@gemini') &&
!contains(github.event.comment.body, '@gemini triage') &&
(github.event.comment.author_association == 'OWNER' ||
github.event.comment.author_association == 'MEMBER' ||
github.event.comment.author_association == 'COLLABORATOR'))
runs-on: ubuntu-latest
env:
GEMINI_API_KEY: ${{ secrets.GEMINI_API_KEY }}
steps:
- name: Generate token
id: generate_token
uses: actions/create-github-app-token@v1
with:
app-id: ${{ secrets.APP_ID }}
private-key: ${{ secrets.PRIVATE_KEY }}
- name: Checkout repository
uses: actions/checkout@v4
with:
token: ${{ steps.generate_token.outputs.token }}
- name: Set up Git
run: |
git config --local user.email "[email protected]"
git config --local user.name "Gemini Bot"
- name: Get issue data
id: issue_data
env:
GH_TOKEN: ${{ steps.generate_token.outputs.token }}
run: |
# Get issue number from event or context
if [ "${{ github.event_name }}" = "issue_comment" ]; then
ISSUE_NUMBER="${{ github.event.issue.number }}"
elif [ "${{ github.event_name }}" = "issues" ]; then
ISSUE_NUMBER="${{ github.event.issue.number }}"
else
echo "Unable to determine issue number"
exit 1
fi
# Get issue details
gh issue view $ISSUE_NUMBER --json title,body,labels,assignees > issue_details.json
# Extract additional instructions from comment if triggered by comment
ADDITIONAL_INSTRUCTIONS=""
if [ "${{ github.event_name }}" = "issue_comment" ]; then
COMMENT_BODY="${{ github.event.comment.body }}"
ADDITIONAL_INSTRUCTIONS=$(echo "$COMMENT_BODY" | sed -n 's/.*@gemini[[:space:]]*\(.*\)/\1/p' | head -1)
fi
echo "ISSUE_NUMBER=$ISSUE_NUMBER" >> $GITHUB_OUTPUT
echo "ADDITIONAL_INSTRUCTIONS=$ADDITIONAL_INSTRUCTIONS" >> $GITHUB_OUTPUT
- name: Run Gemini CLI
uses: ./.github/actions/gemini
env:
GITHUB_TOKEN: ${{ steps.generate_token.outputs.token }}
ISSUE_NUMBER: ${{ steps.issue_data.outputs.ISSUE_NUMBER }}
ADDITIONAL_INSTRUCTIONS: ${{ steps.issue_data.outputs.ADDITIONAL_INSTRUCTIONS }}
REPOSITORY: ${{ github.repository }}
with:
GEMINI_API_KEY: ${{ secrets.GEMINI_API_KEY }}
OTLP_GCP_WIF_PROVIDER: ${{ secrets.OTLP_GCP_WIF_PROVIDER }}
OTLP_GOOGLE_CLOUD_PROJECT: ${{ secrets.OTLP_GOOGLE_CLOUD_PROJECT }}
prompt: |
You are a helpful GitHub issue assistant. Your task is to analyze and help with GitHub issues.
**Issue Details:**
- Issue Number: #${ISSUE_NUMBER}
- Repository: ${REPOSITORY}
- Issue URL: ${{ github.event.issue.html_url }}
**Additional Instructions:** ${ADDITIONAL_INSTRUCTIONS}
**CRITICAL: When making code modifications, follow this EXACT workflow:**
**STEP 1: Analyze the Issue**
- Use: `gh issue view ${ISSUE_NUMBER} --json title,body,labels,assignees`
- Analyze issue content thoroughly
- Search codebase for related files using `glob`, `search_file_content`, `list_directory`
**STEP 2: If Code Changes Are Needed**
Create a new branch with 'gemini/' prefix:
- Use: `git checkout -b gemini/issue-${ISSUE_NUMBER}-description`
- Example: `git checkout -b gemini/issue-123-fix-sequential-tools`
**STEP 3: Make Code Modifications**
- Use `read_file` to examine existing code
- Use `write_file` or `replace` to make changes
- Use `search_file_content` to find related code
- Make focused, targeted changes to fix the issue
**STEP 4: Test Changes (if applicable)**
- Run relevant tests if test files exist
- Use `run_shell_command` to execute test commands
**STEP 5: Commit and Push Changes**
- Use: `git add .` (or specific files)
- Use: `git commit -m "Gemini: [Issue #${ISSUE_NUMBER}] Brief description of fix"`
- Use: `git push -u origin gemini/issue-${ISSUE_NUMBER}-description`
**STEP 6: Create Pull Request**
- Use: `gh pr create --title "Fix #${ISSUE_NUMBER}: Issue description" --body "Fixes #${ISSUE_NUMBER}\n\nSummary of changes:\n- Change 1\n- Change 2" --head gemini/issue-${ISSUE_NUMBER}-description --base main`
- This will return a PR URL like: https://github.com/repo/pull/123
**STEP 7: Comment on Issue with Results**
Use: `gh issue comment ${ISSUE_NUMBER} -b "MESSAGE"`
**Your final comment MUST include:**
- Summary of what was analyzed
- What changes were made (if any)
- Link to the created PR (if applicable)
- Brief explanation of the solution
**Example final comment:**
```
Hi! I've analyzed and fixed this issue.
**Problem:** Sequential tool calling was getting stuck after first tool call.
**Solution:** Added proper prompt handling after each tool call in the Ollama model integration.
**Changes Made:**
- Modified `src/praisonai-agents/ollama-sequential.py`
- Added continuation prompts for multi-step tool execution
**Pull Request:** https://github.com/MervinPraison/PraisonAI/pull/856
The fix is ready for review. Please test and let me know if you encounter any issues!
```
**IMPORTANT NOTES:**
- Always create branches with 'gemini/' prefix
- Always include issue number in branch name and commit message
- Always create PR that references the issue
- Always provide PR link in your final response
- Use the GITHUB_TOKEN environment variable for authentication
Please proceed with helping resolve this issue following this exact workflow.
settings_json: |
{
"coreTools": [
"run_shell_command(echo)",
"run_shell_command(gh issue view)",
"run_shell_command(gh issue comment)",
"run_shell_command(gh issue list)",
"run_shell_command(gh issue edit)",
"run_shell_command(gh pr create)",
"run_shell_command(gh pr view)",
"run_shell_command(gh pr list)",
"run_shell_command(cat)",
"run_shell_command(head)",
"run_shell_command(tail)",
"run_shell_command(grep)",
"run_shell_command(find)",
"run_shell_command(git checkout)",
"run_shell_command(git add)",
"run_shell_command(git commit)",
"run_shell_command(git push)",
"run_shell_command(git status)",
"run_shell_command(git branch)",
"list_directory",
"read_file",
"read_many_files",
"write_file",
"replace",
"glob",
"search_file_content",
"web_fetch",
"google_web_search",
"save_memory"
],
"telemetry": {
"enabled": false,
"otlpGcpWifProvider": "${{ secrets.OTLP_GCP_WIF_PROVIDER }}",
"googleCloudProject": "${{ secrets.OTLP_GOOGLE_CLOUD_PROJECT }}"
}
}
+242
View File
@@ -0,0 +1,242 @@
name: Gemini PR Review
on:
issue_comment:
types: [created]
workflow_dispatch:
inputs:
pr_number:
description: 'PR number to review'
required: true
type: number
jobs:
review-pr:
if: >
github.event_name == 'workflow_dispatch' ||
(github.event_name == 'issue_comment' &&
github.event.issue.pull_request &&
contains(github.event.comment.body, '@gemini') &&
(github.event.comment.author_association == 'OWNER' ||
github.event.comment.author_association == 'MEMBER' ||
github.event.comment.author_association == 'COLLABORATOR'))
timeout-minutes: 15
runs-on: ubuntu-latest
permissions:
contents: read
id-token: write
pull-requests: write
issues: write
steps:
- name: Generate GitHub App Token
id: generate_token
uses: actions/create-github-app-token@v1
with:
app-id: ${{ secrets.APP_ID }}
private-key: ${{ secrets.PRIVATE_KEY }}
- name: Determine checkout ref
id: checkout_ref
run: |
if [ "${{ github.event_name }}" = "issue_comment" ]; then
echo "ref=refs/pull/${{ github.event.issue.number }}/head" >> "$GITHUB_OUTPUT"
else
echo "ref=${{ github.event.pull_request.head.sha }}" >> "$GITHUB_OUTPUT"
fi
- name: Checkout PR code
uses: actions/checkout@v4
with:
token: ${{ steps.generate_token.outputs.token }}
ref: ${{ steps.checkout_ref.outputs.ref }}
fetch-depth: 0
- name: Get PR details
id: get_pr
env:
GITHUB_TOKEN: ${{ steps.generate_token.outputs.token }}
run: |
if [ "${{ github.event_name }}" = "workflow_dispatch" ]; then
PR_NUMBER=${{ github.event.inputs.pr_number }}
elif [ "${{ github.event_name }}" = "issue_comment" ]; then
PR_NUMBER=${{ github.event.issue.number }}
else
PR_NUMBER=${{ github.event.pull_request.number }}
fi
echo "pr_number=$PR_NUMBER" >> "$GITHUB_OUTPUT"
# Extract additional instructions from comment (if triggered by comment)
ADDITIONAL_INSTRUCTIONS=""
if [ "${{ github.event_name }}" = "issue_comment" ]; then
COMMENT_BODY="${{ github.event.comment.body }}"
ADDITIONAL_INSTRUCTIONS=$(echo "$COMMENT_BODY" | sed 's/.*@gemini//' | xargs)
fi
echo "additional_instructions=$ADDITIONAL_INSTRUCTIONS" >> "$GITHUB_OUTPUT"
# Get PR details
PR_DATA=$(gh pr view $PR_NUMBER --json title,body,additions,deletions,changedFiles,baseRefName,headRefName)
echo "pr_data=$PR_DATA" >> "$GITHUB_OUTPUT"
# Get file changes
CHANGED_FILES=$(gh pr diff $PR_NUMBER --name-only)
echo "changed_files<<EOF" >> "$GITHUB_OUTPUT"
echo "$CHANGED_FILES" >> "$GITHUB_OUTPUT"
echo "EOF" >> "$GITHUB_OUTPUT"
- name: Run Gemini PR Review
uses: ./.github/actions/gemini
env:
GITHUB_TOKEN: ${{ steps.generate_token.outputs.token }}
PR_NUMBER: ${{ steps.get_pr.outputs.pr_number }}
PR_DATA: ${{ steps.get_pr.outputs.pr_data }}
CHANGED_FILES: ${{ steps.get_pr.outputs.changed_files }}
ADDITIONAL_INSTRUCTIONS: ${{ steps.get_pr.outputs.additional_instructions }}
REPOSITORY: ${{ github.repository }}
with:
version: 0.1.8-rc.0
GEMINI_API_KEY: ${{ secrets.GEMINI_API_KEY }}
OTLP_GCP_WIF_PROVIDER: ${{ secrets.OTLP_GCP_WIF_PROVIDER }}
OTLP_GOOGLE_CLOUD_PROJECT: ${{ secrets.OTLP_GOOGLE_CLOUD_PROJECT }}
settings_json: |
{
"coreTools": [
"run_shell_command(echo)",
"run_shell_command(gh pr view)",
"run_shell_command(gh pr diff)",
"run_shell_command(gh pr comment)",
"run_shell_command(gh pr list)",
"run_shell_command(cat)",
"run_shell_command(head)",
"run_shell_command(tail)",
"run_shell_command(grep)",
"run_shell_command(find)",
"run_shell_command(git checkout)",
"run_shell_command(git add)",
"run_shell_command(git commit)",
"run_shell_command(git push)",
"run_shell_command(git status)",
"run_shell_command(git branch)",
"run_shell_command(git fetch)",
"list_directory",
"read_file",
"read_many_files",
"write_file",
"replace",
"glob",
"search_file_content",
"web_fetch",
"google_web_search",
"save_memory"
],
"telemetry": {
"enabled": false,
"target": "gcp"
},
"sandbox": false
}
prompt: |
You are an expert code reviewer and AI assistant for Pull Requests. You can review code AND make modifications to improve the PR.
**PR Details:**
- PR Number: #$PR_NUMBER
- Repository: $REPOSITORY
- Additional Instructions: $ADDITIONAL_INSTRUCTIONS
**CRITICAL: When making code modifications, follow this EXACT workflow:**
**STEP 1: Analyze the PR**
- Use: `echo "$PR_DATA"` to get PR details (JSON format)
- Use: `echo "$CHANGED_FILES"` to get changed files list
- Use: `gh pr diff ${PR_NUMBER}` to see the full diff
- Use: `gh pr view ${PR_NUMBER} --json title,body,headRefName` to get PR info
- Analyze the code changes thoroughly
**STEP 2: If Code Improvements Are Needed**
**IMPORTANT:** DO NOT create a new branch! Work on the EXISTING PR branch.
- Use: `git fetch origin` to get latest changes
- Get PR branch name: `gh pr view ${PR_NUMBER} --json headRefName`
- Switch to PR branch: `git checkout BRANCH_NAME` (use actual branch name from PR)
- Use: `git pull origin BRANCH_NAME` to get latest PR changes
**STEP 3: Make Code Modifications**
- Use `read_file` to examine existing code
- Use `write_file` or `replace` to make improvements
- Use `search_file_content` to find related code
- Make focused, targeted improvements to the PR
**STEP 4: Test Changes (if applicable)**
- Run relevant tests if test files exist
- Use `run_shell_command` to execute test commands
**STEP 5: Commit and Push Changes to EXISTING PR Branch**
- Use: `git add .` (or specific files)
- Use: `git commit -m "Gemini: [PR #${PR_NUMBER}] Brief description of improvements"`
- Use: `git push origin BRANCH_NAME` (push to EXISTING PR branch)
**STEP 6: Review and Comment on PR**
Write comprehensive review and post comment with your analysis and any improvements made.
Use: `gh pr comment ${PR_NUMBER} -b "MESSAGE"`
Review Guidelines:
- Focus on code quality, security, performance, and maintainability
- Check for common issues: potential bugs, security vulnerabilities, performance bottlenecks
- Verify error handling and edge cases
- Look for code style and best practices
- Comment on architecture and design decisions if significant
- Be constructive and specific in feedback
- Highlight both issues and positive aspects
- Suggest improvements with examples when possible
Review Areas:
- **Security**: Authentication, authorization, input validation, data sanitization
- **Performance**: Algorithms, database queries, caching, resource usage
- **Reliability**: Error handling, logging, testing coverage, edge cases
- **Maintainability**: Code structure, documentation, naming conventions
- **Functionality**: Logic correctness, requirements fulfillment
Output Format:
Structure your review using this exact format with markdown:
## 📋 Review Summary
Provide a brief 2-3 sentence overview of the PR and overall assessment.
## 🔍 General Feedback
- List general observations about code quality
- Mention overall patterns or architectural decisions
- Highlight positive aspects of the implementation
- Note any recurring themes across files
## 🎯 Specific Feedback
Only include sections below that have actual issues. If there are no issues in a priority category, omit that entire section.
### 🔴 Critical
(Only include this section if there are critical issues)
Issues that must be addressed before merging (security vulnerabilities, breaking changes, major bugs):
- **File: `filename:line`** - Description of critical issue with specific recommendation
### 🟡 High
(Only include this section if there are high priority issues)
Important issues that should be addressed (performance problems, design flaws, significant bugs):
- **File: `filename:line`** - Description of high priority issue with suggested fix
### 🟢 Medium
(Only include this section if there are medium priority issues)
Improvements that would enhance code quality (style issues, minor optimizations, better practices):
- **File: `filename:line`** - Description of medium priority improvement
### 🔵 Low
(Only include this section if there are suggestions)
Nice-to-have improvements and suggestions (documentation, naming, minor refactoring):
- **File: `filename:line`** - Description of suggestion or enhancement
**Note**: If no specific issues are found in any category, simply state "No specific issues identified in this review."
## ✅ Highlights
(Only include this section if there are positive aspects to highlight)
- Mention specific good practices or implementations
- Acknowledge well-written code sections
- Note improvements from previous versions
---
*Review completed by Gemini CLI*
+3 -1
View File
@@ -75,4 +75,6 @@ agents/praisonaiagents/praisonaiagents.egg-info
# Local Netlify folder
.netlify
.qodo
CopilotKit*
CopilotKit*
.gemini
issue_details.json
+4 -4
View File
@@ -120,6 +120,10 @@ agent.start('Write a movie script about a robot in Mars');
![PraisonAI CLI Demo](docs/demo/praisonai-cli-demo.gif)
## Star History
[![Star History Chart](https://api.star-history.com/svg?repos=MervinPraison/PraisonAI&type=Date)](https://docs.praison.ai)
## AI Agents Flow
```mermaid
@@ -557,10 +561,6 @@ uv pip install -r pyproject.toml --extra "crewai,autogen"
- 🖼️ Vision Language Model (VLM) Support
- 🎙️ Real-time Voice Interaction
## Star History
[![Star History Chart](https://api.star-history.com/svg?repos=MervinPraison/PraisonAI&type=Date)](https://docs.praison.ai)
## Video Tutorials
| Topic | Video |
+1 -1
View File
@@ -16,7 +16,7 @@ RUN mkdir -p /root/.praison
# Install Python packages (using latest versions)
RUN pip install --no-cache-dir \
flask \
"praisonai>=2.2.52" \
"praisonai>=2.2.70" \
"praisonai[api]" \
gunicorn \
markdown
+1 -1
View File
@@ -16,7 +16,7 @@ RUN mkdir -p /root/.praison
# Install Python packages (using latest versions)
RUN pip install --no-cache-dir \
praisonai_tools \
"praisonai>=2.2.52" \
"praisonai>=2.2.70" \
"praisonai[chat]" \
"embedchain[github,youtube]"
+1 -1
View File
@@ -20,7 +20,7 @@ RUN mkdir -p /root/.praison
# Install Python packages (using latest versions)
RUN pip install --no-cache-dir \
praisonai_tools \
"praisonai>=2.2.52" \
"praisonai>=2.2.70" \
"praisonai[ui]" \
"praisonai[chat]" \
"praisonai[realtime]" \
+1 -1
View File
@@ -16,7 +16,7 @@ RUN mkdir -p /root/.praison
# Install Python packages (using latest versions)
RUN pip install --no-cache-dir \
praisonai_tools \
"praisonai>=2.2.52" \
"praisonai>=2.2.70" \
"praisonai[ui]" \
"praisonai[crewai]"
+2 -2
View File
@@ -121,7 +121,7 @@ healthcheck:
## 📦 Package Versions
All Docker images use consistent, up-to-date versions:
- PraisonAI: `>=2.2.52`
- PraisonAI: `>=2.2.70`
- PraisonAI Agents: `>=0.0.92`
- Python: `3.11-slim`
@@ -218,7 +218,7 @@ docker-compose up -d
### Version Pinning
To use specific versions, update the Dockerfile:
```dockerfile
RUN pip install "praisonai==2.2.52" "praisonaiagents==0.0.92"
RUN pip install "praisonai==2.2.70" "praisonaiagents==0.0.92"
```
## 🌐 Production Deployment
-70
View File
@@ -1,70 +0,0 @@
#!/usr/bin/env python3
"""Example demonstrating validation feedback in workflow retry logic"""
import sys
import os
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src/praisonai-agents'))
from praisonaiagents import Agent, Task, PraisonAIAgents
# This example shows how validation feedback is now passed to retry tasks
print("""
==========================================
Validation Feedback Feature Demonstration
==========================================
This example shows how the workflow retry logic now includes validation feedback.
Before: When validation failed, the retry task would start fresh without knowing why it failed.
After: The retry task receives detailed feedback about what went wrong.
Example Scenario:
1. A data collection task produces output
2. A validation task checks if the output meets criteria
3. If validation fails, the data collection task is retried WITH feedback about why it failed
The retry task will see:
- Previous attempt failed validation with reason: invalid
- Validation feedback: [specific reason for failure]
- Rejected output: [the output that failed validation]
- Please try again with a different approach based on this feedback.
This enables context-dependent tasks to improve on retry instead of repeating the same mistakes.
==========================================
Implementation Details:
1. Added 'validation_feedback' field to Task class
2. When routing from decision task to retry task, capture validation details
3. Include feedback in task context when building retry task description
4. Clear feedback after use to prevent persistence
The solution is minimal and backward compatible - existing workflows continue to work unchanged.
""")
# Example workflow structure (without actual execution)
print("\nExample Workflow Structure:")
print("""
collect_task = Task(
name="collect_data",
description="Collect data from source",
is_start=True,
next_tasks=["validate_data"]
)
validate_task = Task(
name="validate_data",
task_type="decision",
description="Validate if data meets criteria",
condition={
"valid": [], # End workflow
"invalid": ["collect_data"] # Retry with feedback
}
)
# When validation returns "invalid":
# 1. The workflow captures the validation output
# 2. Stores it in collect_task.validation_feedback
# 3. On retry, collect_data task sees the feedback in its context
# 4. Agent can adjust approach based on specific failure reason
""")
@@ -0,0 +1,293 @@
{
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "ZOWgx2fYqW8T"
},
"source": [
"# GitHub Repository Insights with PraisonAIAgents"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "XgJdLRiIqZ2L"
},
"source": [
"This notebook demonstrates how to explore and analyze GitHub repositories using PraisonAIAgents in Google Colab. You can query repository issues, pull requests, and activity patterns with natural language, leveraging the power of agentic tools—all without leaving your notebook."
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "Y_koMT6qzxne"
},
"source": [
"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/Dhivya-Bharathy/PraisonAI/blob/main/examples/cookbooks/Programming_Code_Analysis_Agents/GitHub_Repository_Insights.ipynb)\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "2ey7zWAJrIPx"
},
"source": [
"# Dependencies"
]
},
{
"cell_type": "code",
"execution_count": 24,
"metadata": {
"id": "_xFH-iLTxzd8"
},
"outputs": [],
"source": [
"!pip install praisonaiagents openai requests --quiet"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "Kzty0zxzsbZ_"
},
"source": [
"# Set Up Key"
]
},
{
"cell_type": "code",
"execution_count": 25,
"metadata": {
"id": "HCVP5fmvx2hH"
},
"outputs": [],
"source": [
"import os\n",
"\n",
"# Set your GitHub token here\n",
"GITHUB_TOKEN = \"your-github-token\" # <-- Replace with your actual token\n",
"\n",
"# Set your OpenAI API key here (required by PraisonAIAgents, even if not used for LLM)\n",
"OPENAI_API_KEY = \"your-openai-api-key\" # <-- Replace with your actual OpenAI API key\n",
"\n",
"# Set environment variable for OpenAI\n",
"os.environ[\"OPENAI_API_KEY\"] = OPENAI_API_KEY"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "10BTLbF4s_f-"
},
"source": [
"# Import Libraries"
]
},
{
"cell_type": "code",
"execution_count": 26,
"metadata": {
"id": "_sLahvfXyEb0"
},
"outputs": [],
"source": [
"from praisonaiagents import Agent\n",
"import requests\n",
"from IPython.display import display, Markdown"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "GRftdghxtUDs"
},
"source": [
"# Custom Tool"
]
},
{
"cell_type": "code",
"execution_count": 27,
"metadata": {
"id": "nT11EFNNyIHl"
},
"outputs": [],
"source": [
"def github_query_tool(query: str, repo: str, github_token: str) -> str:\n",
" \"\"\"\n",
" Custom tool to query GitHub repository data using the GitHub API.\n",
" Supports basic queries for issues, pull requests, and repository info.\n",
" \"\"\"\n",
" headers = {\n",
" \"Authorization\": f\"token {github_token}\",\n",
" \"Accept\": \"application/vnd.github.v3+json\"\n",
" }\n",
" base_url = f\"https://api.github.com/repos/{repo}\"\n",
" result = \"\"\n",
"\n",
" if \"issue\" in query.lower():\n",
" url = f\"{base_url}/issues\"\n",
" response = requests.get(url, headers=headers)\n",
" if response.ok:\n",
" issues = response.json()\n",
" result = f\"Found {len(issues)} issues in {repo}.\\n\"\n",
" for issue in issues[:5]:\n",
" result += f\"- #{issue['number']}: {issue['title']} (State: {issue['state']})\\n\"\n",
" else:\n",
" result = f\"Failed to fetch issues: {response.text}\"\n",
"\n",
" elif \"pull request\" in query.lower() or \"pr\" in query.lower():\n",
" url = f\"{base_url}/pulls\"\n",
" response = requests.get(url, headers=headers)\n",
" if response.ok:\n",
" pulls = response.json()\n",
" result = f\"Found {len(pulls)} pull requests in {repo}.\\n\"\n",
" for pr in pulls[:5]:\n",
" result += f\"- #{pr['number']}: {pr['title']} (State: {pr['state']})\\n\"\n",
" else:\n",
" result = f\"Failed to fetch pull requests: {response.text}\"\n",
"\n",
" elif \"activity\" in query.lower() or \"stats\" in query.lower():\n",
" url = f\"{base_url}/stats/contributors\"\n",
" response = requests.get(url, headers=headers)\n",
" if response.ok:\n",
" stats = response.json()\n",
" result = f\"Repository activity for {repo}:\\n\"\n",
" for contributor in stats[:5]:\n",
" result += f\"- {contributor['author']['login']}: {contributor['total']} commits\\n\"\n",
" else:\n",
" result = f\"Failed to fetch activity stats: {response.text}\"\n",
"\n",
" else:\n",
" # Default: basic repo info\n",
" response = requests.get(base_url, headers=headers)\n",
" if response.ok:\n",
" info = response.json()\n",
" result = (\n",
" f\"Repository: {info['full_name']}\\n\"\n",
" f\"Description: {info.get('description', 'No description')}\\n\"\n",
" f\"Stars: {info['stargazers_count']}\\n\"\n",
" f\"Forks: {info['forks_count']}\\n\"\n",
" f\"Open Issues: {info['open_issues_count']}\\n\"\n",
" )\n",
" else:\n",
" result = f\"Failed to fetch repository info: {response.text}\"\n",
"\n",
" return result"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "mAFg5SW-tacD"
},
"source": [
"# Register Tool and Create Agent"
]
},
{
"cell_type": "code",
"execution_count": 28,
"metadata": {
"id": "x_MmUc_AyMaB"
},
"outputs": [],
"source": [
"def github_tool_wrapper(inputs):\n",
" query = inputs.get(\"query\", \"\")\n",
" repo = inputs.get(\"repo\", \"\")\n",
" github_token = inputs.get(\"github_token\", \"\")\n",
" return github_query_tool(query, repo, github_token)\n",
"\n",
"agent = Agent(\n",
" instructions=\"\"\"\n",
" You are a GitHub assistant. Help users explore repositories and their activity.\n",
" - Provide organized, concise insights about the repository.\n",
" - Focus on facts and data from the GitHub API.\n",
" - Use markdown formatting for better readability.\n",
" - Present numerical data in tables when appropriate.\n",
" - Include links to relevant GitHub pages when helpful.\n",
" \"\"\",\n",
" tools={\n",
" \"github_query_tool\": github_tool_wrapper\n",
" }\n",
")"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "lPstz7bbtr0k"
},
"source": [
"# User Input and Run Agent"
]
},
{
"cell_type": "code",
"execution_count": 30,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 118
},
"id": "MymtRmlnyQqd",
"outputId": "33a23810-4a67-4615-f9e9-146b3030fcbf"
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Enter the GitHub repository (owner/repo): Dhivya-Bharathy/PraisonAI\n",
"Enter your question (e.g., Show me open issues): Show me open issues\n"
]
},
{
"data": {
"text/markdown": [
"### Results\n",
"Found 0 issues in Dhivya-Bharathy/PraisonAI.\n"
],
"text/plain": [
"<IPython.core.display.Markdown object>"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"# User input\n",
"repo = input(\"Enter the GitHub repository (owner/repo): \")\n",
"query = input(\"Enter your question (e.g., Show me open issues): \")\n",
"\n",
"# Call the tool directly (since agent.chat expects an LLM, but we want tool output)\n",
"response = agent.execute_tool(\"github_query_tool\", {\n",
" \"query\": query,\n",
" \"repo\": repo,\n",
" \"github_token\": GITHUB_TOKEN\n",
"})\n",
"\n",
"# Display the result\n",
"display(Markdown(f\"### Results\\n{response}\"))"
]
}
],
"metadata": {
"colab": {
"provenance": []
},
"kernelspec": {
"display_name": "Python 3",
"name": "python3"
},
"language_info": {
"name": "python"
}
},
"nbformat": 4,
"nbformat_minor": 0
}
@@ -0,0 +1,309 @@
{
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "vmQmHoy-5cAh"
},
"source": [
"# Notion Workspace Assistant with PraisonAIAgents"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "WOow52NM5dmh"
},
"source": [
"This notebook demonstrates how to interact with your Notion workspace using PraisonAIAgents in Google Colab. You can read, search, and update Notion pages with natural language queries, leveraging agentic tools for seamless document management—all within a notebook environment."
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "TlnFAv9P_lGL"
},
"source": [
"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/Dhivya-Bharathy/PraisonAI/blob/main/examples/cookbooks/Programming_Code_Analysis_Agents/Notion_Workspace_Assistant.ipynb)\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "zMn7hUGZ5hn5"
},
"source": [
"# Dependencies"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"id": "U6X3lKAu6iU3"
},
"outputs": [],
"source": [
"!pip install praisonaiagents openai requests --quiet"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "4_96-D7z6lrC"
},
"source": [
"# Set Up Key"
]
},
{
"cell_type": "code",
"execution_count": 2,
"metadata": {
"id": "kltUFVIW6rEO"
},
"outputs": [],
"source": [
"import os\n",
"\n",
"# Set your Notion and OpenAI API keys here\n",
"NOTION_API_KEY = \"ntn_..\" # <-- Replace with your actual Notion integration token\n",
"OPENAI_API_KEY = \"sk-proj-_..\" # <-- Replace with your actual OpenAI API key\n",
"\n",
"# Set environment variables for downstream libraries\n",
"os.environ[\"NOTION_API_KEY\"] = NOTION_API_KEY\n",
"os.environ[\"OPENAI_API_KEY\"] = OPENAI_API_KEY"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "FOLTBEXu635I"
},
"source": [
"# Tools"
]
},
{
"cell_type": "code",
"execution_count": 3,
"metadata": {
"id": "fLWxSASM5jlU"
},
"outputs": [],
"source": [
"from praisonaiagents import Agent\n",
"import requests\n",
"from IPython.display import display, Markdown"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "KN_Sxha37EQ0"
},
"source": [
"# Custom Tool"
]
},
{
"cell_type": "code",
"execution_count": 4,
"metadata": {
"id": "WWxcfO9L7Fsu"
},
"outputs": [],
"source": [
"def notion_read_page_tool(page_id: str, notion_token: str) -> str:\n",
" \"\"\"\n",
" Custom tool to read the content of a Notion page using the Notion API.\n",
" \"\"\"\n",
" url = f\"https://api.notion.com/v1/blocks/{page_id}/children\"\n",
" headers = {\n",
" \"Authorization\": f\"Bearer {notion_token}\",\n",
" \"Notion-Version\": \"2022-06-28\"\n",
" }\n",
" response = requests.get(url, headers=headers)\n",
" if response.ok:\n",
" data = response.json()\n",
" blocks = data.get(\"results\", [])\n",
" content = []\n",
" for block in blocks:\n",
" block_type = block.get(\"type\", \"\")\n",
" text = \"\"\n",
" if block_type == \"paragraph\":\n",
" text = block.get(\"paragraph\", {}).get(\"text\", [])\n",
" if text:\n",
" text = text[0].get(\"plain_text\", \"\")\n",
" elif block_type == \"heading_1\":\n",
" text = block.get(\"heading_1\", {}).get(\"text\", [])\n",
" if text:\n",
" text = \"# \" + text[0].get(\"plain_text\", \"\")\n",
" elif block_type == \"heading_2\":\n",
" text = block.get(\"heading_2\", {}).get(\"text\", [])\n",
" if text:\n",
" text = \"## \" + text[0].get(\"plain_text\", \"\")\n",
" elif block_type == \"heading_3\":\n",
" text = block.get(\"heading_3\", {}).get(\"text\", [])\n",
" if text:\n",
" text = \"### \" + text[0].get(\"plain_text\", \"\")\n",
" if text:\n",
" content.append(text)\n",
" return \"\\n\".join(content) if content else \"No readable content found on this page.\"\n",
" else:\n",
" return f\"Failed to fetch page content: {response.text}\""
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "ebuhw9z27Nae"
},
"source": [
"# YAML Prompt"
]
},
{
"cell_type": "code",
"execution_count": 5,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "eMjDDjzO7Pf3",
"outputId": "b935f96d-f147-44de-80ea-7f75d23f9e7b"
},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"\n",
"agent_instructions: |\n",
" You are a Notion workspace assistant. Help users read and manage their Notion pages.\n",
" - Provide clear, organized summaries of page content.\n",
" - Use the Notion API to fetch and display information.\n",
" - If a page cannot be read, explain the issue clearly.\n",
"\n",
"tools:\n",
" - name: notion_read_page_tool\n",
" description: >\n",
" Reads the content of a Notion page using its page ID and the user's Notion integration token.\n",
" parameters:\n",
" - name: page_id\n",
" type: string\n",
" description: The Notion page ID to read.\n",
" - name: notion_token\n",
" type: string\n",
" description: The user's Notion integration token.\n",
"\n"
]
}
],
"source": [
"yaml_prompt = \"\"\"\n",
"agent_instructions: |\n",
" You are a Notion workspace assistant. Help users read and manage their Notion pages.\n",
" - Provide clear, organized summaries of page content.\n",
" - Use the Notion API to fetch and display information.\n",
" - If a page cannot be read, explain the issue clearly.\n",
"\n",
"tools:\n",
" - name: notion_read_page_tool\n",
" description: >\n",
" Reads the content of a Notion page using its page ID and the user's Notion integration token.\n",
" parameters:\n",
" - name: page_id\n",
" type: string\n",
" description: The Notion page ID to read.\n",
" - name: notion_token\n",
" type: string\n",
" description: The user's Notion integration token.\n",
"\"\"\"\n",
"print(yaml_prompt)"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "1ozm6Z467WQK"
},
"source": [
"# Main"
]
},
{
"cell_type": "code",
"execution_count": 15,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
"height": 83
},
"id": "eNg-jwpx7X5s",
"outputId": "4440b52a-46b3-4683-9183-13cd21596e25"
},
"outputs": [
{
"data": {
"text/markdown": [
"### Notion Page Content\n",
"*No content found or page is not accessible. Please check your page ID and sharing settings.*"
],
"text/plain": [
"<IPython.core.display.Markdown object>"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"# Define a wrapper to match the agent's expected tool signature\n",
"def notion_tool_wrapper(inputs):\n",
" page_id = inputs.get(\"page_id\", \"\")\n",
" notion_token = inputs.get(\"notion_token\", \"\")\n",
" return notion_read_page_tool(page_id, notion_token)\n",
"\n",
"# Create the agent and register the tool\n",
"agent = Agent(\n",
" instructions=\"\"\"\n",
" You are a Notion workspace assistant. Help users read and manage their Notion pages.\n",
" - Provide clear, organized summaries of page content.\n",
" - Use the Notion API to fetch and display information.\n",
" - If a page cannot be read, explain the issue clearly.\n",
" \"\"\",\n",
" tools={\n",
" \"notion_read_page_tool\": notion_tool_wrapper\n",
" }\n",
")\n",
"\n",
"# Call the tool directly (since agent.chat expects an LLM, but we want tool output)\n",
"response = agent.execute_tool(\"notion_read_page_tool\", {\n",
" \"page_id\": page_id,\n",
" \"notion_token\": NOTION_API_KEY\n",
"})\n",
"\n",
"# Check for error pattern and display a friendly message\n",
"if response.startswith(\"Failed to fetch page content:\"):\n",
" display(Markdown(\"### Notion Page Content\\n*No content found or page is not accessible. Please check your page ID and sharing settings.*\"))\n",
"else:\n",
" display(Markdown(f\"### Notion Page Content\\n{response}\"))"
]
}
],
"metadata": {
"colab": {
"provenance": []
},
"kernelspec": {
"display_name": "Python 3",
"name": "python3"
},
"language_info": {
"name": "python"
}
},
"nbformat": 4,
"nbformat_minor": 0
}
@@ -1,355 +1,355 @@
{
"cells": [
{
"cell_type": "markdown",
"metadata": {
"id": "_T4g3t5sLoag"
},
"source": [
"# ZeroScript: AI-Powered Browser Test Automation using PraisonAI Agent & OpenAI"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "O4PKfpalLsmn"
},
"source": [
"# Description"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "CEG1pfumLysm"
},
"source": [
"AI-powered browser testing using PraisonAI Agent and OpenAI. Executes YAML-defined test cases, captures screenshots, logs interactions, and generates JSON reports. Ideal for intelligent, automated UI testing.\n",
"\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "lbgxOG4gL1Wf"
},
"source": [
"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/DhivyaBharathy-web/PraisonAI/blob/main/examples/cookbooks/ZeroScript_AI_TestExecutor.ipynb)\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "WCJqMUzXIndh"
},
"source": [
"# Dependencies"
]
},
{
"cell_type": "code",
"execution_count": 34,
"metadata": {
"id": "9AQhJ59cJPmR"
},
"outputs": [],
"source": [
"!pip install --quiet praisonaiagents browser-use python-dotenv pyyaml openai\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "QD-dF-6_It1z"
},
"source": [
"# Environment Setup"
]
},
{
"cell_type": "code",
"execution_count": 35,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "QkRQDli4JhCV",
"outputId": "4b8957ed-a37a-4b11-8585-3537a0437d40"
},
"outputs": [
"cells": [
{
"data": {
"text/plain": [
"True"
"cell_type": "markdown",
"metadata": {
"id": "_T4g3t5sLoag"
},
"source": [
"# ZeroScript: AI-Powered Browser Test Automation using PraisonAI Agent & OpenAI"
]
},
"execution_count": 35,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"import os\n",
"from dotenv import load_dotenv\n",
"\n",
"os.environ[\"OPENAI_API_KEY\"] = \"Enter your key\" # replace with actual key\n",
"load_dotenv()\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "ELuogWnrIw3o"
},
"source": [
"# Create my_secrets.py"
]
},
{
"cell_type": "code",
"execution_count": 36,
"metadata": {
"id": "sG6tO_v0Jm5N"
},
"outputs": [],
"source": [
"with open(\"my_secrets.py\", \"w\") as f:\n",
" f.write('''sensitive_data = {\n",
" \"username\": \"standard_user\",\n",
" \"password\": \"secret_sauce\"\n",
"}''')\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "rDB_viedIzRR"
},
"source": [
"# Create views.py"
]
},
{
"cell_type": "code",
"execution_count": 37,
"metadata": {
"id": "CtwcybdbJoV8"
},
"outputs": [],
"source": [
"with open(\"views.py\", \"w\") as f:\n",
" f.write('''\n",
"from pydantic import BaseModel\n",
"\n",
"class TestResult(BaseModel):\n",
" status: str\n",
" message: str = \"\"\n",
"''')\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "YTRmBpTiI1kd"
},
"source": [
"# Create agent.py"
]
},
{
"cell_type": "code",
"execution_count": 38,
"metadata": {
"id": "zdJJGt6TJpvr"
},
"outputs": [],
"source": [
"with open(\"agent.py\", \"w\") as f:\n",
" f.write('''\n",
"import os\n",
"from praisonaiagents import Agent\n",
"from browser_use import Browser, BrowserConfig, Controller\n",
"from browser_use.browser.context import BrowserContext, BrowserContextConfig\n",
"from openai import OpenAI\n",
"from views import TestResult\n",
"from my_secrets import sensitive_data\n",
"\n",
"# Browser setup\n",
"browser = Browser(config=BrowserConfig(headless=True))\n",
"context_cfg = BrowserContextConfig(save_recording_path=\"data/recordings\")\n",
"controller = Controller(output_model=TestResult)\n",
"\n",
"async def browser_use(task: str):\n",
" context = BrowserContext(browser=browser, config=context_cfg)\n",
" agent = Agent(\n",
" task=task,\n",
" llm=OpenAI(api_key=os.getenv(\"OPENAI_API_KEY\"), model=\"gpt-4\"),\n",
" browser_context=context,\n",
" sensitive_data=sensitive_data,\n",
" controller=controller,\n",
" save_conversation_path=\"data/conversations/convo\",\n",
" extend_system_message=(\n",
" \"You are a professional software tester. Execute browser test steps precisely.\"\n",
" )\n",
" )\n",
" response = await agent.run()\n",
" await context.close()\n",
" return response\n",
"''')\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "-LZtQIE3I4-_"
},
"source": [
"# Create Sample Test YAML"
]
},
{
"cell_type": "code",
"execution_count": 39,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "l1OdXYI-JreK",
"outputId": "62323fcd-c74f-498f-defc-2c170fa8983e"
},
"outputs": [
{
"data": {
"text/plain": [
"350"
"cell_type": "markdown",
"metadata": {
"id": "O4PKfpalLsmn"
},
"source": [
"# Description"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "CEG1pfumLysm"
},
"source": [
"AI-powered browser testing using PraisonAI Agent and OpenAI. Executes YAML-defined test cases, captures screenshots, logs interactions, and generates JSON reports. Ideal for intelligent, automated UI testing.\n",
"\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "lbgxOG4gL1Wf"
},
"source": [
"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/DhivyaBharathy-web/PraisonAI/blob/main/examples/cookbooks/ZeroScript_AI_TestExecutor.ipynb)\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "WCJqMUzXIndh"
},
"source": [
"# Dependencies"
]
},
{
"cell_type": "code",
"execution_count": 34,
"metadata": {
"id": "9AQhJ59cJPmR"
},
"outputs": [],
"source": [
"!pip install --quiet praisonaiagents browser-use python-dotenv pyyaml openai\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "QD-dF-6_It1z"
},
"source": [
"# Environment Setup"
]
},
{
"cell_type": "code",
"execution_count": 35,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "QkRQDli4JhCV",
"outputId": "4b8957ed-a37a-4b11-8585-3537a0437d40"
},
"outputs": [
{
"data": {
"text/plain": [
"True"
]
},
"execution_count": 35,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"import os\n",
"from dotenv import load_dotenv\n",
"\n",
"os.environ[\"OPENAI_API_KEY\"] = \"Enter your key\" # replace with actual key\n",
"load_dotenv()\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "ELuogWnrIw3o"
},
"source": [
"# Create my_secrets.py"
]
},
{
"cell_type": "code",
"execution_count": 36,
"metadata": {
"id": "sG6tO_v0Jm5N"
},
"outputs": [],
"source": [
"with open(\"my_secrets.py\", \"w\") as f:\n",
" f.write('''sensitive_data = {\n",
" \"username\": \"standard_user\",\n",
" \"password\": \"secret_sauce\"\n",
"}''')\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "rDB_viedIzRR"
},
"source": [
"# Create views.py"
]
},
{
"cell_type": "code",
"execution_count": 37,
"metadata": {
"id": "CtwcybdbJoV8"
},
"outputs": [],
"source": [
"with open(\"views.py\", \"w\") as f:\n",
" f.write('''\n",
"from pydantic import BaseModel\n",
"\n",
"class TestResult(BaseModel):\n",
" status: str\n",
" message: str = \"\"\n",
"''')\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "YTRmBpTiI1kd"
},
"source": [
"# Create agent.py"
]
},
{
"cell_type": "code",
"execution_count": 38,
"metadata": {
"id": "zdJJGt6TJpvr"
},
"outputs": [],
"source": [
"with open(\"agent.py\", \"w\") as f:\n",
" f.write('''\n",
"import os\n",
"from praisonaiagents import Agent\n",
"from browser_use import Browser, BrowserConfig, Controller\n",
"from browser_use.browser.context import BrowserContext, BrowserContextConfig\n",
"from openai import OpenAI\n",
"from views import TestResult\n",
"from my_secrets import sensitive_data\n",
"\n",
"# Browser setup\n",
"browser = Browser(config=BrowserConfig(headless=True))\n",
"context_cfg = BrowserContextConfig(save_recording_path=\"data/recordings\")\n",
"controller = Controller(output_model=TestResult)\n",
"\n",
"async def browser_use(task: str):\n",
" context = BrowserContext(browser=browser, config=context_cfg)\n",
" agent = Agent(\n",
" task=task,\n",
" llm=OpenAI(api_key=os.getenv(\"OPENAI_API_KEY\"), model=\"gpt-4\"),\n",
" browser_context=context,\n",
" sensitive_data=sensitive_data,\n",
" controller=controller,\n",
" save_conversation_path=\"data/conversations/convo\",\n",
" extend_system_message=(\n",
" \"You are a professional software tester. Execute browser test steps precisely.\"\n",
" )\n",
" )\n",
" response = await agent.run()\n",
" await context.close()\n",
" return response\n",
"''')\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "-LZtQIE3I4-_"
},
"source": [
"# Create Sample Test YAML"
]
},
{
"cell_type": "code",
"execution_count": 39,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "l1OdXYI-JreK",
"outputId": "62323fcd-c74f-498f-defc-2c170fa8983e"
},
"outputs": [
{
"data": {
"text/plain": [
"350"
]
},
"execution_count": 39,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"from pathlib import Path\n",
"tests_dir = Path(\"tests\")\n",
"tests_dir.mkdir(exist_ok=True)\n",
"yaml_content = \"\"\"\n",
"hooks:\n",
" beforeEach: |\n",
" Navigate to https://www.saucedemo.com\n",
" afterEach: |\n",
" Take screenshot\n",
"\n",
"tests:\n",
" - id: \"login_test_001\"\n",
" name: \"Valid Login Test\"\n",
" instructions: |\n",
" 1. Enter standard_user in username field\n",
" 2. Enter secret_sauce in password field\n",
" 3. Click login button\n",
" 4. Verify page contains text \"Products\"\n",
"\"\"\"\n",
"(Path(\"tests/login_test.yml\")).write_text(yaml_content)\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "sUjy-7OdI7p_"
},
"source": [
"# Run Tests"
]
},
{
"cell_type": "code",
"execution_count": 40,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
},
"id": "Agkiwh7xJtY3",
"outputId": "759aa77c-fd65-49b8-b239-550588059478"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"Exception ignored in: <function BrowserSession.__del__ at 0x7e2f540fb100>\n",
"Traceback (most recent call last):\n",
" File \"/usr/local/lib/python3.11/dist-packages/browser_use/browser/session.py\", line 497, in __del__\n",
" status = f'\ud83e\ude93 killing pid={self.browser_pid}...' if (self.browser_pid and owns_browser) else '\u2620\ufe0f'\n",
" ^^^^^^^^^^^^^^^^\n",
" File \"/usr/local/lib/python3.11/dist-packages/pydantic/main.py\", line 991, in __getattr__\n",
" raise AttributeError(f'{type(self).__name__!r} object has no attribute {item!r}')\n",
"AttributeError: 'BrowserSession' object has no attribute 'browser_pid'\n"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"\u2705 Report saved to data/reports/report_20250630_092319.json\n"
]
}
],
"source": [
"import asyncio, yaml, json, base64, shutil\n",
"from agent import browser_use, browser\n",
"from pathlib import Path\n",
"from datetime import datetime\n",
"\n",
"# Clean or create data dirs\n",
"for sub in [\"recordings\", \"conversations\", \"screenshots\", \"reports\"]:\n",
" d = Path(\"data\") / sub\n",
" if d.exists(): shutil.rmtree(d)\n",
" d.mkdir(parents=True, exist_ok=True)\n",
"\n",
"async def execute_test_file(path):\n",
" cfg = yaml.safe_load(Path(path).read_text())\n",
" hooks, results = cfg.get(\"hooks\", {}), []\n",
" for t in cfg[\"tests\"]:\n",
" task = f\"{hooks.get('beforeEach','')}\\n{t['instructions']}\\n{hooks.get('afterEach','')}\"\n",
" res = {\"id\": t[\"id\"], \"name\": t[\"name\"]}\n",
" try:\n",
" r = await browser_use(task)\n",
" imgs = r.screenshots()\n",
" if imgs:\n",
" out = Path(\"data/screenshots\") / f\"{t['id']}.png\"\n",
" out.write_bytes(base64.b64decode(imgs[-1]))\n",
" result_obj = r.final_result() if hasattr(r, \"final_result\") else str(r)\n",
" res[\"result\"] = result_obj.model_dump() if hasattr(result_obj, \"model_dump\") else str(result_obj)\n",
" except Exception as e:\n",
" res[\"error\"] = str(e)\n",
" results.append(res)\n",
" return results\n",
"\n",
"async def main():\n",
" all_results = []\n",
" for file in Path(\"tests\").glob(\"*.yml\"):\n",
" all_results.extend(await execute_test_file(str(file)))\n",
" report = Path(\"data/reports\") / f\"report_{datetime.now():%Y%m%d_%H%M%S}.json\"\n",
" report.write_text(json.dumps(all_results, indent=2))\n",
" await browser.close()\n",
" print(f\"\u2705 Report saved to {report}\")\n",
"\n",
"await main()\n"
]
},
"execution_count": 39,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"from pathlib import Path\n",
"tests_dir = Path(\"tests\")\n",
"tests_dir.mkdir(exist_ok=True)\n",
"yaml_content = \"\"\"\n",
"hooks:\n",
" beforeEach: |\n",
" Navigate to https://www.saucedemo.com\n",
" afterEach: |\n",
" Take screenshot\n",
"\n",
"tests:\n",
" - id: \"login_test_001\"\n",
" name: \"Valid Login Test\"\n",
" instructions: |\n",
" 1. Enter standard_user in username field\n",
" 2. Enter secret_sauce in password field\n",
" 3. Click login button\n",
" 4. Verify page contains text \"Products\"\n",
"\"\"\"\n",
"(Path(\"tests/login_test.yml\")).write_text(yaml_content)\n"
]
},
{
"cell_type": "markdown",
"metadata": {
"id": "sUjy-7OdI7p_"
},
"source": [
"# Run Tests"
]
},
{
"cell_type": "code",
"execution_count": 40,
"metadata": {
],
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
"provenance": []
},
"id": "Agkiwh7xJtY3",
"outputId": "759aa77c-fd65-49b8-b239-550588059478"
},
"outputs": [
{
"name": "stderr",
"output_type": "stream",
"text": [
"Exception ignored in: <function BrowserSession.__del__ at 0x7e2f540fb100>\n",
"Traceback (most recent call last):\n",
" File \"/usr/local/lib/python3.11/dist-packages/browser_use/browser/session.py\", line 497, in __del__\n",
" status = f'🪓 killing pid={self.browser_pid}...' if (self.browser_pid and owns_browser) else '☠️'\n",
" ^^^^^^^^^^^^^^^^\n",
" File \"/usr/local/lib/python3.11/dist-packages/pydantic/main.py\", line 991, in __getattr__\n",
" raise AttributeError(f'{type(self).__name__!r} object has no attribute {item!r}')\n",
"AttributeError: 'BrowserSession' object has no attribute 'browser_pid'\n"
]
"kernelspec": {
"display_name": "Python 3",
"name": "python3"
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"✅ Report saved to data/reports/report_20250630_092319.json\n"
]
"language_info": {
"name": "python"
}
],
"source": [
"import asyncio, yaml, json, base64, shutil\n",
"from agent import browser_use, browser\n",
"from pathlib import Path\n",
"from datetime import datetime\n",
"\n",
"# Clean or create data dirs\n",
"for sub in [\"recordings\", \"conversations\", \"screenshots\", \"reports\"]:\n",
" d = Path(\"data\") / sub\n",
" if d.exists(): shutil.rmtree(d)\n",
" d.mkdir(parents=True, exist_ok=True)\n",
"\n",
"async def execute_test_file(path):\n",
" cfg = yaml.safe_load(Path(path).read_text())\n",
" hooks, results = cfg.get(\"hooks\", {}), []\n",
" for t in cfg[\"tests\"]:\n",
" task = f\"{hooks.get('beforeEach','')}\\n{t['instructions']}\\n{hooks.get('afterEach','')}\"\n",
" res = {\"id\": t[\"id\"], \"name\": t[\"name\"]}\n",
" try:\n",
" r = await browser_use(task)\n",
" imgs = r.screenshots()\n",
" if imgs:\n",
" out = Path(\"data/screenshots\") / f\"{t['id']}.png\"\n",
" out.write_bytes(base64.b64decode(imgs[-1]))\n",
" result_obj = r.final_result() if hasattr(r, \"final_result\") else str(r)\n",
" res[\"result\"] = result_obj.dict() if hasattr(result_obj, \"dict\") else str(result_obj)\n",
" except Exception as e:\n",
" res[\"error\"] = str(e)\n",
" results.append(res)\n",
" return results\n",
"\n",
"async def main():\n",
" all_results = []\n",
" for file in Path(\"tests\").glob(\"*.yml\"):\n",
" all_results.extend(await execute_test_file(str(file)))\n",
" report = Path(\"data/reports\") / f\"report_{datetime.now():%Y%m%d_%H%M%S}.json\"\n",
" report.write_text(json.dumps(all_results, indent=2))\n",
" await browser.close()\n",
" print(f\"✅ Report saved to {report}\")\n",
"\n",
"await main()\n"
]
}
],
"metadata": {
"colab": {
"provenance": []
},
"kernelspec": {
"display_name": "Python 3",
"name": "python3"
},
"language_info": {
"name": "python"
}
},
"nbformat": 4,
"nbformat_minor": 0
}
"nbformat": 4,
"nbformat_minor": 0
}
+142
View File
@@ -0,0 +1,142 @@
#!/usr/bin/env python3
"""
Fixed example demonstrating proper guardrail usage with PraisonAI Agents.
This addresses the issues reported in issue #875.
"""
from praisonaiagents import Agent, Task, GuardrailResult, PraisonAIAgents, TaskOutput
from typing import Tuple, Any
import trafilatura
# Example 1: Using GuardrailResult return type (now supported!)
def validate_length_guardrailresult(output: TaskOutput) -> GuardrailResult:
"""Ensure output is between 100-500 characters using GuardrailResult"""
# Extract the raw text from the TaskOutput object
text = output.raw if hasattr(output, 'raw') else str(output)
length = len(text)
if 100 <= length <= 500:
return GuardrailResult(
success=True,
result=output, # Pass through the original output
error=""
)
else:
return GuardrailResult(
success=False,
result=None,
error=f"Output must be 100-500 chars, got {length}"
)
# Example 2: Using Tuple[bool, Any] return type (original method)
def validate_length_tuple(output: TaskOutput) -> Tuple[bool, Any]:
"""Ensure output is between 100-500 characters using tuple"""
text = output.raw if hasattr(output, 'raw') else str(output)
length = len(text)
if 100 <= length <= 500:
return True, output
else:
return False, f"Output must be 100-500 chars, got {length}"
# Tool function
def get_url_context(url):
"""Fetch and extract content from a URL"""
downloaded = trafilatura.fetch_url(url)
if not downloaded:
return "Sorry, I couldn't fetch the content from that URL."
extracted = trafilatura.extract(
downloaded,
include_comments=False,
include_links=True,
output_format='json',
with_metadata=True,
url=url
)
if not extracted:
return "Sorry, I couldn't extract readable content from that page."
return extracted # returns JSON string
# Create agent with FIXED tools parameter (must be a list!)
agent = Agent(
name="Content Summarizer",
role="Content Analysis Expert",
goal="Summarize web content concisely",
instructions="You are a helpful assistant that summarizes web content",
llm="gemini/gemini-2.5-flash-lite-preview-06-17",
self_reflect=False,
verbose=True,
tools=[get_url_context] # FIX: tools must be a list, not a single function
)
# Create task with GuardrailResult guardrail
task_with_guardrailresult = Task(
name="summarise article with GuardrailResult",
description="get the context of this url: https://blog.google/technology/ai/dolphingemma/ and produce a summary below 500 characters",
agent=agent,
guardrail=validate_length_guardrailresult, # Using GuardrailResult
expected_output="summary of the article below 500 characters",
max_retries=3 # Will retry up to 3 times if guardrail fails
)
# Alternative: Create task with tuple guardrail
task_with_tuple = Task(
name="summarise article with tuple",
description="get the context of this url: https://blog.google/technology/ai/dolphingemma/ and produce a summary below 500 characters",
agent=agent,
guardrail=validate_length_tuple, # Using Tuple[bool, Any]
expected_output="summary of the article below 500 characters",
max_retries=3
)
# Example with string-based LLM guardrail
task_with_llm_guardrail = Task(
name="summarise with LLM guardrail",
description="get the context of this url: https://blog.google/technology/ai/dolphingemma/ and produce a summary",
agent=agent,
guardrail="Ensure the summary is professional, factual, and between 100-500 characters",
expected_output="professional summary of the article"
)
# Run with GuardrailResult example
print("=== Running with GuardrailResult guardrail ===")
agents_gr = PraisonAIAgents(
agents=[agent],
tasks=[task_with_guardrailresult]
)
# Uncomment to run:
# result_gr = agents_gr.start()
# Run with Tuple example
print("\n=== Running with Tuple[bool, Any] guardrail ===")
agents_tuple = PraisonAIAgents(
agents=[agent],
tasks=[task_with_tuple]
)
# Uncomment to run:
# result_tuple = agents_tuple.start()
# Run with LLM guardrail example
print("\n=== Running with LLM-based guardrail ===")
agents_llm = PraisonAIAgents(
agents=[agent],
tasks=[task_with_llm_guardrail]
)
# Uncomment to run:
# result_llm = agents_llm.start()
print("""
Key fixes applied:
1. GuardrailResult is now accepted as a valid return type annotation
2. tools parameter must be a list: tools=[get_url_context] not tools=get_url_context
3. Both GuardrailResult and Tuple[bool, Any] return types are supported
4. String-based LLM guardrails are also supported
The guardrail will automatically retry (up to max_retries times) if validation fails.
""")
@@ -0,0 +1,259 @@
# Multi-Provider/Multi-Model Support
PraisonAI now supports intelligent multi-provider and multi-model capabilities, allowing agents to automatically select the most appropriate AI model based on task requirements, cost considerations, and performance needs.
## 🚀 Key Features
### 1. **Automatic Model Selection**
- Analyzes task complexity and selects the best model
- Considers cost vs. performance trade-offs
- Supports fallback mechanisms for reliability
### 2. **Cost Optimization**
- Routes simple tasks to cheaper models (e.g., GPT-4o-mini, Gemini Flash)
- Reserves expensive models for complex tasks
- Tracks usage and provides cost estimates
### 3. **Multi-Provider Support**
- Works with OpenAI, Anthropic, Google, Groq, and more
- Seamless switching between providers
- Provider preference settings
### 4. **Flexible Routing Strategies**
- `auto`: Automatic selection based on task analysis
- `cost-optimized`: Prioritize cheaper models
- `performance-optimized`: Prioritize capability
- `manual`: Use specified model
## 📦 Installation
The multi-provider support is included in the standard installation:
```bash
pip install praisonaiagents
```
## 🔧 Basic Usage
### Simple Multi-Model Agent
```python
from praisonaiagents import Task, PraisonAIAgents
from praisonaiagents.agent import RouterAgent
# Create a multi-model agent
agent = RouterAgent(
name="Smart Assistant",
role="Adaptive AI Assistant",
goal="Complete tasks using the most appropriate model",
models=["gpt-4o-mini", "gpt-4o", "claude-3-5-sonnet-20241022"],
routing_strategy="auto" # Automatic model selection
)
# Create tasks
simple_task = Task(
name="calculate",
description="What is 15% of 250?",
agent=agent
)
complex_task = Task(
name="analyze",
description="Write a Python implementation of the A* pathfinding algorithm",
agent=agent
)
# Run tasks - agent will automatically select appropriate models
agents = PraisonAIAgents(
agents=[agent],
tasks=[simple_task, complex_task]
)
results = agents.start()
```
### Cost-Optimized Workflow
```python
from praisonaiagents.llm import ModelRouter
# Create custom router with cost constraints
router = ModelRouter(
cost_threshold=0.005, # Max $0.005 per 1k tokens
preferred_providers=["google", "openai"]
)
# Create cost-conscious agent
analyzer = RouterAgent(
name="Budget Analyzer",
role="Data Analyst",
goal="Analyze data efficiently",
models={
"gemini/gemini-1.5-flash": {},
"gpt-4o-mini": {},
"claude-3-haiku-20240307": {}
},
model_router=router,
routing_strategy="cost-optimized"
)
```
## 🎯 Routing Strategies
### Auto Routing (Default)
Automatically selects models based on:
- Task complexity analysis
- Required capabilities (tools, vision, etc.)
- Context size requirements
- Cost/performance balance
### Cost-Optimized
Prioritizes cheaper models while meeting task requirements:
- Uses lightweight models for simple tasks
- Only escalates to expensive models when necessary
- Ideal for high-volume operations
### Performance-Optimized
Prioritizes model capability:
- Uses the best available model for the task
- Ideal for quality-critical applications
- Less concern for cost
### Manual
Uses the specified model without routing:
- Direct control over model selection
- Useful for testing or specific requirements
## 📊 Model Profiles
The system includes pre-configured profiles for popular models:
| Model | Provider | Best For | Cost/1k tokens |
|-------|----------|----------|----------------|
| gpt-4o-mini | OpenAI | Simple tasks, speed | $0.00075 |
| gemini-1.5-flash | Google | Cost-effective, multimodal | $0.000125 |
| claude-3-haiku | Anthropic | Fast responses | $0.0008 |
| gpt-4o | OpenAI | General purpose | $0.0075 |
| claude-3-5-sonnet-20241022 | Anthropic | Complex reasoning | $0.009 |
| deepseek-chat | DeepSeek | Code & math | $0.0014 |
## 🔍 Task Complexity Analysis
The system analyzes tasks to determine complexity:
- **Simple**: Basic calculations, definitions, yes/no questions
- **Moderate**: Summarization, basic analysis, classification
- **Complex**: Code generation, algorithm implementation
- **Very Complex**: Multi-step reasoning, system design
## 📈 Usage Tracking
Track model usage and costs:
```python
# Get usage report
report = agent.get_usage_report()
print(report)
# Output:
{
'agent_name': 'Smart Assistant',
'routing_strategy': 'auto',
'model_usage': {
'gpt-4o-mini': {'calls': 5, 'tokens': 1500, 'cost': 0.0011},
'gpt-4o': {'calls': 2, 'tokens': 3000, 'cost': 0.0225}
},
'total_cost_estimate': 0.0236,
'total_calls': 7
}
```
## 🛠️ Custom Model Configuration
Add custom model profiles:
```python
from praisonaiagents.llm import ModelProfile, TaskComplexity
custom_model = ModelProfile(
name="custom-model",
provider="custom",
complexity_range=(TaskComplexity.SIMPLE, TaskComplexity.MODERATE),
cost_per_1k_tokens=0.002,
strengths=["domain-specific"],
capabilities=["text"],
context_window=32000
)
router = ModelRouter(models=[custom_model])
```
## 🔗 Integration with AutoAgents
Works seamlessly with AutoAgents:
```python
from praisonaiagents.agents import AutoAgents
# Create auto agents
auto = AutoAgents(
instructions="Analyze market trends and create a report",
max_agents=3
)
# Convert to multi-model agents
for i, agent in enumerate(auto.agents):
auto.agents[i] = RouterAgent(
name=agent.name,
role=agent.role,
goal=agent.goal,
models=["gpt-4o-mini", "gpt-4o", "claude-3-5-sonnet"],
routing_strategy="auto"
)
results = auto.start()
```
## 🌟 Best Practices
1. **Start with Auto Routing**: Let the system learn your needs
2. **Monitor Costs**: Use usage reports to optimize
3. **Set Cost Thresholds**: Prevent unexpected expenses
4. **Use Appropriate Models**: Don't use GPT-4 for simple math
5. **Leverage Fallbacks**: Configure fallback models for reliability
## 🔐 Environment Variables
Set API keys for each provider:
```bash
export OPENAI_API_KEY="sk-..."
export ANTHROPIC_API_KEY="sk-ant-..."
export GEMINI_API_KEY="..."
export GROQ_API_KEY="..."
```
## 🤝 Contributing
The multi-provider system is extensible. To add new models:
1. Add model profile to `ModelRouter.DEFAULT_MODELS`
2. Ensure the provider is supported by LiteLLM
3. Test with various task complexities
## 📚 Examples
See the complete example in `examples/python/agents/multi-provider-agent.py` for:
- Auto-routing examples
- Cost-optimized workflows
- Performance-optimized agents
- Custom routing logic
- Integration patterns
## 🎉 Benefits
- **Cost Savings**: Reduce API costs by 50-80%
- **Better Performance**: Use the right tool for each job
- **Flexibility**: Switch providers without code changes
- **Reliability**: Automatic fallbacks and error handling
- **Transparency**: Track usage and costs
@@ -0,0 +1,73 @@
"""
AutoAgents Hierarchical Generation Example
This example demonstrates hierarchical agent patterns using PraisonAI
for structured task delegation and coordinated workflows.
"""
from praisonaiagents import Agent, Task, PraisonAIAgents
from praisonaiagents.tools import internet_search
print("=== AutoAgents Hierarchical Generation Example ===\n")
# Create manager agent for hierarchical coordination
manager_agent = Agent(
name="Project Manager",
role="Project Coordinator",
goal="Coordinate hierarchical task execution and manage team workflow",
backstory="Experienced manager who excels at breaking down complex projects and coordinating teams",
tools=[internet_search],
allow_delegation=True,
verbose=True
)
# Create specialist agent for execution
research_agent = Agent(
name="Research Specialist",
role="Market Research Analyst",
goal="Conduct thorough market research and analysis",
backstory="Expert researcher skilled in market analysis and competitive intelligence",
tools=[internet_search],
verbose=True
)
# Hierarchical task structure
planning_task = Task(
description="""Plan a comprehensive market research project for electric vehicles:
1. Define research scope and objectives
2. Identify key research areas and methodologies
3. Create task breakdown structure
4. Set deliverable requirements for execution team
Prepare clear instructions for the research specialist.""",
expected_output="Detailed project plan with clear research instructions",
agent=manager_agent
)
execution_task = Task(
description="""Execute the market research plan:
1. Follow the research plan from the project manager
2. Conduct comprehensive market analysis
3. Gather competitive intelligence and market data
4. Prepare detailed findings report
Use the planning guidance to ensure complete coverage.""",
expected_output="Comprehensive market research report with findings",
agent=research_agent,
context=[planning_task]
)
# Run hierarchical workflow
agents_system = PraisonAIAgents(
agents=[manager_agent, research_agent],
tasks=[planning_task, execution_task],
process="sequential",
verbose=True
)
print("Starting hierarchical agent generation...")
result = agents_system.start()
print(f"\nHierarchical Result: {result[:200]}...")
print("\n✅ AutoAgents hierarchical generation complete!")
print("Demonstrated coordinated hierarchical workflow between manager and specialist agents.")
@@ -0,0 +1,285 @@
"""
Multi-Provider/Multi-Model Agent Example
This example demonstrates how to use multiple AI providers/models with intelligent
agent-based selection for cost optimization and performance.
"""
import os
from praisonaiagents import Agent, Task, PraisonAIAgents
from praisonaiagents.agent import RouterAgent
from praisonaiagents.llm.model_router import ModelRouter, ModelProfile, TaskComplexity
# Example 1: Simple Multi-Model Agent with Auto-Routing
def example_auto_routing():
"""Example of automatic model routing based on task complexity"""
print("\n=== Example 1: Auto-Routing Multi-Model Agent ===\n")
# Create a router agent that automatically selects models
research_agent = RouterAgent(
name="Smart Researcher",
role="Adaptive Research Assistant",
goal="Research topics using the most appropriate AI model",
backstory="I analyze task complexity and route to the best model",
models=["gpt-4o-mini", "gpt-4o", "claude-3-5-sonnet-20241022"],
routing_strategy="auto", # Automatic model selection
verbose=True
)
# Create tasks with different complexity levels
simple_task = Task(
name="simple_calculation",
description="Calculate the sum of 1542 and 2873",
expected_output="The numerical result",
agent=research_agent
)
moderate_task = Task(
name="summarize_article",
description="Summarize the key points about renewable energy trends",
expected_output="A concise summary with main points",
agent=research_agent
)
complex_task = Task(
name="analyze_code",
description="Implement a Python function to find the longest palindromic substring in a given string with O(n^2) complexity",
expected_output="Complete Python implementation with explanation",
agent=research_agent
)
# Run the tasks
agents = PraisonAIAgents(
agents=[research_agent],
tasks=[simple_task, moderate_task, complex_task],
process="sequential",
verbose=True
)
results = agents.start()
# Show usage report
print("\n=== Model Usage Report ===")
print(research_agent.get_usage_report())
# Example 2: Cost-Optimized Workflow with Routing
def example_cost_optimized_workflow():
"""Example of cost-optimized workflow with model routing"""
print("\n=== Example 2: Cost-Optimized Workflow ===\n")
# Create custom model router with cost threshold
cost_router = ModelRouter(
cost_threshold=0.01, # Max $0.01 per 1k tokens
preferred_providers=["google", "openai", "anthropic"] # Provider preference
)
# Create specialized agents with different model strategies
analyzer = RouterAgent(
name="Cost-Conscious Analyzer",
role="Data Analyzer",
goal="Analyze data efficiently while minimizing costs",
models={
"gemini/gemini-1.5-flash": {},
"gpt-4o-mini": {},
"deepseek-chat": {}
},
model_router=cost_router,
routing_strategy="cost-optimized",
verbose=True
)
writer = RouterAgent(
name="Quality Writer",
role="Content Writer",
goal="Create high-quality content",
models={
"claude-3-5-sonnet-20241022": {},
"gpt-4o": {}
},
routing_strategy="performance-optimized", # Prefer better models
verbose=True
)
# Create workflow tasks
analysis_task = Task(
name="analyze_data",
description="Analyze this dataset: [1, 2, 3, 4, 5]. Find mean, median, and mode.",
expected_output="Statistical analysis results",
agent=analyzer
)
writing_task = Task(
name="write_report",
description="Write a professional report based on the analysis",
expected_output="Professional report with insights",
agent=writer,
context=[analysis_task] # Use results from analysis
)
# Run workflow
workflow = PraisonAIAgents(
agents=[analyzer, writer],
tasks=[analysis_task, writing_task],
process="sequential",
verbose=True
)
results = workflow.start()
# Show cost comparison
print("\n=== Cost Analysis ===")
print(f"Analyzer costs: {analyzer.get_usage_report()}")
print(f"Writer costs: {writer.get_usage_report()}")
# Example 3: Auto Agent Mode with Multi-Provider Support
def example_auto_agents_multi_provider():
"""Example using AutoAgents with multi-provider support"""
print("\n=== Example 3: AutoAgents with Multi-Provider Support ===\n")
from praisonaiagents.agents import AutoAgents
# Create AutoAgents that will automatically assign appropriate models
auto_agents = AutoAgents(
instructions="Create a market research report on electric vehicles. Include data analysis, competitor analysis, and future projections.",
max_agents=3,
llm="gpt-4o-mini", # Default model for agent generation
verbose=True
)
# After agents are created, upgrade them to multi-model agents
multi_model_agents = []
for agent in auto_agents.agents:
# Convert regular agents to multi-model agents
multi_agent = RouterAgent(
name=agent.name,
role=agent.role,
goal=agent.goal,
backstory=agent.backstory,
tools=agent.tools,
models=["gpt-4o-mini", "gemini/gemini-1.5-flash", "claude-3-haiku-20240307", "gpt-4o"],
routing_strategy="auto",
verbose=True
)
multi_model_agents.append(multi_agent)
# Update the agents in the AutoAgents instance
auto_agents.agents = multi_model_agents
# Run the auto-generated workflow
results = auto_agents.start()
# Show model usage across all agents
print("\n=== Multi-Provider Usage Summary ===")
total_calls = 0
for agent in multi_model_agents:
report = agent.get_usage_report()
print(f"\nAgent: {agent.name}")
for model, stats in report['model_usage'].items():
if stats['calls'] > 0:
print(f" - {model}: {stats['calls']} calls")
total_calls += stats['calls']
print(f"\nTotal API calls: {total_calls}")
# Example 4: Custom Routing Logic
def example_custom_routing():
"""Example with custom routing logic based on specific requirements"""
print("\n=== Example 4: Custom Routing Logic ===\n")
# Create custom model profiles for specific use cases
custom_models = [
ModelProfile(
name="gpt-4o",
provider="openai",
complexity_range=(TaskComplexity.SIMPLE, TaskComplexity.VERY_COMPLEX),
cost_per_1k_tokens=0.0075,
strengths=["code-generation", "debugging"],
capabilities=["text", "function-calling"],
context_window=128000
),
ModelProfile(
name="deepseek-chat",
provider="deepseek",
complexity_range=(TaskComplexity.MODERATE, TaskComplexity.VERY_COMPLEX),
cost_per_1k_tokens=0.001,
strengths=["mathematics", "algorithms"],
capabilities=["text", "function-calling"],
context_window=128000
)
]
# Create router with custom models
custom_router = ModelRouter(models=custom_models)
# Tool for web search (example)
def search_web(query: str) -> str:
"""Search the web for information"""
return f"Search results for: {query}"
# Create specialized agent
coder = RouterAgent(
name="Adaptive Coder",
role="Software Developer",
goal="Write and debug code using the best model for each task",
backstory="I'm an expert coder who knows when to use different AI models",
model_router=custom_router,
routing_strategy="auto",
tools=[search_web],
verbose=True
)
# Create coding tasks
tasks = [
Task(
name="fix_bug",
description="Fix this Python bug: 'list' object has no attribute 'appendx'",
expected_output="Corrected code with explanation",
agent=coder
),
Task(
name="implement_algorithm",
description="Implement Dijkstra's shortest path algorithm in Python",
expected_output="Complete implementation with comments",
agent=coder,
tools=[search_web]
)
]
# Run tasks
agents = PraisonAIAgents(
agents=[coder],
tasks=tasks,
process="sequential",
verbose=True
)
results = agents.start()
print("\n=== Custom Routing Results ===")
print(coder.get_usage_report())
# Run examples
if __name__ == "__main__":
# Make sure to set your API keys
# os.environ["OPENAI_API_KEY"] = "your-openai-key"
# os.environ["ANTHROPIC_API_KEY"] = "your-anthropic-key"
# os.environ["GEMINI_API_KEY"] = "your-gemini-key"
# Run examples
example_auto_routing()
example_cost_optimized_workflow()
example_auto_agents_multi_provider()
example_custom_routing()
print("\n=== All Examples Completed ===")
print("\nKey Features Demonstrated:")
print("1. Automatic model selection based on task complexity")
print("2. Cost-optimized routing for budget-conscious operations")
print("3. Performance-optimized routing for quality-critical tasks")
print("4. Multi-provider support with fallback mechanisms")
print("5. Integration with AutoAgents for automatic workflow generation")
print("6. Custom routing logic for specialized use cases")
@@ -0,0 +1,82 @@
"""
RouterAgent Cost Optimization Example
This example demonstrates how to use RouterAgent for intelligent model selection
based on cost optimization. The router agent automatically selects the most
cost-effective model for different types of tasks while maintaining quality.
Features demonstrated:
- Cost-optimized model routing
- Automatic model selection based on task complexity
- Usage tracking and cost monitoring
- Fallback mechanisms for reliability
"""
from praisonaiagents import Agent
from praisonaiagents.agent.router_agent import RouterAgent
from praisonaiagents.tools import duckduckgo
# Define model options with cost considerations
# Using different models for different complexity levels
cheap_model = "gpt-4o-mini" # For simple tasks
balanced_model = "gpt-4o" # For moderate complexity
premium_model = "claude-3-5-sonnet-20241022" # For complex tasks
# Create RouterAgent with cost optimization strategy
router = RouterAgent(
name="CostOptimizedRouter",
role="Cost-Efficient Task Router",
goal="Route tasks to the most cost-effective model while maintaining quality",
backstory="You are a smart router that optimizes for cost-efficiency by selecting appropriate models based on task complexity.",
# Model routing strategy - simple tasks to cheap models, complex to premium
models=[cheap_model, balanced_model, premium_model],
# Routing logic - this could be enhanced with more sophisticated rules
instructions="""You are a cost-optimization router. For simple tasks like basic questions,
definitions, or simple calculations, use the cheapest model. For moderate complexity tasks
like writing, analysis, or research, use the balanced model. For complex tasks requiring
deep reasoning, creativity, or specialized knowledge, use the premium model.""",
tools=[duckduckgo],
verbose=True
)
# Example tasks of varying complexity levels
# Simple task - should route to cheap model
print("="*60)
print("TESTING SIMPLE TASK (should use cheap model)")
print("="*60)
simple_result = router.start("What is the capital of France?")
print(f"Simple task result: {simple_result}")
# Moderate complexity task - should route to balanced model
print("\n" + "="*60)
print("TESTING MODERATE TASK (should use balanced model)")
print("="*60)
moderate_result = router.start("Write a brief analysis of renewable energy trends in 2024")
print(f"Moderate task result: {moderate_result}")
# Complex task - should route to premium model
print("\n" + "="*60)
print("TESTING COMPLEX TASK (should use premium model)")
print("="*60)
complex_result = router.start("Conduct a comprehensive strategic analysis of the impact of quantum computing on cybersecurity, including potential vulnerabilities and mitigation strategies")
print(f"Complex task result: {complex_result}")
# Research task with tool usage - should route appropriately
print("\n" + "="*60)
print("TESTING RESEARCH TASK WITH TOOLS")
print("="*60)
research_result = router.start("Research and analyze the latest developments in artificial intelligence regulation globally")
print(f"Research task result: {research_result}")
print("\n" + "="*80)
print("COST OPTIMIZATION ROUTER DEMONSTRATION COMPLETED")
print("="*80)
print("The RouterAgent automatically selected appropriate models based on task complexity:")
print("- Simple factual questions → Cheaper model (gpt-4o-mini)")
print("- Moderate analysis tasks → Balanced model (gpt-4o)")
print("- Complex strategic analysis → Premium model (claude-3-5-sonnet)")
print("- This approach optimizes costs while maintaining quality for each task type")
@@ -440,7 +440,7 @@ async def generate_research(
"message": "Report generation queued",
"created_at": datetime.now().isoformat(),
"updated_at": datetime.now().isoformat(),
"config": request.dict()
"config": request.model_dump()
}
# Create config object
@@ -31,7 +31,7 @@ def save_reasoning_chain(chain: ChainOfThought) -> str:
"""Save a chain of thought reasoning to analyze patterns."""
filename = f"reasoning_{chain.problem[:20].replace(' ', '_')}.json"
with open(filename, 'w') as f:
json.dump(chain.dict(), f, indent=2)
json.dump(chain.model_dump(), f, indent=2)
return f"Reasoning chain saved to {filename}"
# Example 1: Chain of Thought Agent
@@ -0,0 +1,46 @@
"""
Self-Reflection Optimization Patterns Example
This example demonstrates self-reflection capabilities using PraisonAI's
built-in reflection features for iterative improvement and quality optimization.
"""
from praisonaiagents import Agent, Task, PraisonAIAgents
from praisonaiagents.tools import internet_search
print("=== Self-Reflection Optimization Patterns Example ===\n")
# Create self-reflecting agent with built-in reflection capabilities
reflection_agent = Agent(
name="Self-Reflecting Agent",
role="Self-Improving Researcher",
goal="Demonstrate self-reflection and iterative improvement patterns",
backstory="Expert researcher with strong self-reflection and continuous improvement capabilities",
tools=[internet_search],
self_reflect=True,
min_reflect=2,
max_reflect=4,
reflect_llm="gpt-4o-mini",
verbose=True
)
# Create task that benefits from self-reflection
reflection_task = Task(
description="""Research and analyze the future of artificial intelligence:
1. Investigate current AI trends and breakthrough technologies
2. Analyze potential impacts on society and industry
3. Provide well-reasoned predictions for the next 5 years
4. Ensure analysis is comprehensive, accurate, and well-structured
Use self-reflection to improve the quality and completeness of your analysis.""",
expected_output="Comprehensive AI future analysis with high quality through self-reflection",
agent=reflection_agent
)
# Run with self-reflection optimization
print("Starting self-reflection optimization demonstration...")
result = reflection_agent.execute_task(reflection_task)
print(f"\nSelf-Reflection Result: {result[:200]}...")
print("\n✅ Self-reflection optimization complete!")
print("Agent demonstrated iterative improvement through built-in self-reflection capabilities.")
@@ -0,0 +1,59 @@
"""
Advanced Callback Systems Example
This example demonstrates callback systems using PraisonAI's built-in
callback functionality for monitoring agent interactions and tool usage.
"""
from praisonaiagents import Agent, Task, PraisonAIAgents
from praisonaiagents.display_callback import register_display_callback
from praisonaiagents.tools import internet_search
print("=== Advanced Callback Systems Example ===\n")
# Simple callback function to monitor interactions
def interaction_callback(data):
"""Monitor agent interactions"""
print(f"📋 Interaction: {data.get('agent_name', 'Unknown')} - {data.get('type', 'unknown')}")
# Simple callback function to monitor tool usage
def tool_callback(data):
"""Monitor tool usage"""
print(f"🔧 Tool Used: {data.get('agent_name', 'Unknown')} used {data.get('tool_name', 'unknown')}")
# Simple callback function to monitor generation
def generation_callback(data):
"""Monitor content generation"""
print(f"✍️ Generation: {data.get('agent_name', 'Unknown')} generating content")
# Register callbacks with PraisonAI's callback system
print("Registering callbacks...")
register_display_callback('interaction', interaction_callback)
register_display_callback('tool_call', tool_callback)
register_display_callback('generating', generation_callback)
print("✅ Callbacks registered\n")
# Create a research agent with callback monitoring
research_agent = Agent(
name="Research Agent",
role="Information Researcher",
goal="Research topics and demonstrate callback monitoring",
backstory="Expert researcher that demonstrates callback functionality",
tools=[internet_search],
verbose=True
)
# Create a simple task
research_task = Task(
description="Research the latest developments in renewable energy technology",
expected_output="Brief summary of renewable energy developments",
agent=research_agent
)
# Execute with callback monitoring
print("Starting research with callback monitoring...")
result = research_agent.execute_task(research_task)
print(f"\nResearch Result: {result[:200]}...")
print("\n✅ Callback monitoring demonstration complete!")
print("Callbacks successfully tracked agent interactions, tool usage, and content generation.")
@@ -0,0 +1,204 @@
"""
Comprehensive Guardrails Example
This example demonstrates both function-based and LLM-based guardrails for validating
agent outputs and ensuring quality, safety, and compliance.
Features demonstrated:
- Function-based validation guardrails
- LLM-based content validation
- Multiple validation criteria
- Retry mechanisms with feedback
- Quality assurance workflows
"""
from praisonaiagents import Agent, Task, PraisonAIAgents
from praisonaiagents.task import TaskOutput
from typing import Tuple, Any
import re
# Define function-based guardrails for different validation types
def email_format_guardrail(task_output: TaskOutput) -> Tuple[bool, Any]:
"""
Validates that the output contains properly formatted email addresses.
"""
output_text = str(task_output.raw)
# Email regex pattern
email_pattern = r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b'
emails_found = re.findall(email_pattern, output_text)
if emails_found:
return True, task_output
else:
return False, "No valid email addresses found in the output. Please include at least one properly formatted email address."
def word_count_guardrail(task_output: TaskOutput) -> Tuple[bool, Any]:
"""
Validates that the output meets minimum word count requirements.
"""
output_text = str(task_output.raw)
word_count = len(output_text.split())
min_words = 50 # Minimum required words
if word_count >= min_words:
return True, task_output
else:
return False, f"Output too short. Found {word_count} words but need at least {min_words} words. Please provide a more detailed response."
def professional_tone_guardrail(task_output: TaskOutput) -> Tuple[bool, Any]:
"""
Validates that the output maintains a professional tone.
"""
output_text = str(task_output.raw).lower()
# Check for unprofessional words/phrases
unprofessional_terms = ['stupid', 'dumb', 'sucks', 'terrible', 'awful', 'hate']
for term in unprofessional_terms:
if term in output_text:
return False, f"Output contains unprofessional language ('{term}'). Please revise to maintain a professional tone."
return True, task_output
def factual_accuracy_guardrail(task_output: TaskOutput) -> Tuple[bool, Any]:
"""
Validates that the output includes specific factual elements.
"""
output_text = str(task_output.raw).lower()
# Check for presence of factual indicators
factual_indicators = ['according to', 'research shows', 'studies indicate', 'data suggests', 'evidence']
has_factual_backing = any(indicator in output_text for indicator in factual_indicators)
if has_factual_backing:
return True, task_output
else:
return False, "Output lacks factual backing. Please include references to research, studies, or data to support your claims."
# Create agents with different guardrail configurations
# Agent with email format validation
email_agent = Agent(
name="EmailAgent",
role="Business Communication Specialist",
goal="Create professional business communications with proper email formatting",
backstory="You are a professional communication specialist who ensures all business correspondence includes proper contact information.",
instructions="Always include properly formatted email addresses in your responses."
)
# Agent with word count validation
content_agent = Agent(
name="ContentAgent",
role="Content Writer",
goal="Create detailed, comprehensive content that meets length requirements",
backstory="You are a professional content writer who creates detailed, well-researched articles.",
instructions="Provide comprehensive, detailed responses with sufficient depth and explanation."
)
# Agent with professional tone validation
business_agent = Agent(
name="BusinessAgent",
role="Professional Advisor",
goal="Provide professional business advice with appropriate tone",
backstory="You are a seasoned business advisor who maintains professionalism in all communications.",
instructions="Maintain a professional, respectful tone in all communications."
)
# Agent with factual accuracy validation
research_agent = Agent(
name="ResearchAgent",
role="Research Analyst",
goal="Provide well-researched, fact-based analysis",
backstory="You are a research analyst who backs all claims with evidence and citations.",
instructions="Support all statements with references to research, studies, or credible data sources."
)
# Create tasks with specific guardrails
# Task with email format guardrail
email_task = Task(
name="email_communication",
description="Write a business proposal for a new software project, including contact information",
expected_output="Professional business proposal with contact email addresses",
agent=email_agent,
guardrail=email_format_guardrail,
max_retries=3
)
# Task with word count guardrail
content_task = Task(
name="detailed_analysis",
description="Write an analysis of cloud computing benefits for small businesses",
expected_output="Comprehensive analysis of cloud computing benefits (minimum 50 words)",
agent=content_agent,
guardrail=word_count_guardrail,
max_retries=3
)
# Task with professional tone guardrail
business_task = Task(
name="professional_advice",
description="Provide advice on handling difficult client relationships",
expected_output="Professional advice maintaining appropriate business tone",
agent=business_agent,
guardrail=professional_tone_guardrail,
max_retries=3
)
# Task with factual accuracy guardrail
research_task = Task(
name="research_report",
description="Analyze the impact of artificial intelligence on job markets",
expected_output="Research-backed analysis with citations and evidence",
agent=research_agent,
guardrail=factual_accuracy_guardrail,
max_retries=3
)
# LLM-based guardrail example
llm_guardrail_agent = Agent(
name="LLMGuardrailAgent",
role="Content Quality Validator",
goal="Validate content quality using LLM-based assessment",
backstory="You are a quality assurance specialist who uses AI to validate content quality.",
instructions="Create marketing content that is engaging, accurate, and compliant with advertising standards."
)
# Task with LLM-based guardrail
llm_guardrail_task = Task(
name="marketing_content",
description="Create a marketing email for a new fitness app targeting busy professionals",
expected_output="Engaging marketing email that is accurate and compliant",
agent=llm_guardrail_agent,
# LLM-based guardrail using natural language validation
guardrail="Validate that this marketing content is: 1) Factually accurate with no false claims, 2) Compliant with advertising standards (no misleading statements), 3) Engaging and professional in tone, 4) Includes a clear call-to-action. If any criteria are not met, explain what needs to be improved.",
max_retries=2
)
# Execute tasks with guardrails
agents = PraisonAIAgents(
agents=[email_agent, content_agent, business_agent, research_agent, llm_guardrail_agent],
tasks=[email_task, content_task, business_task, research_task, llm_guardrail_task],
verbose=True
)
print("="*80)
print("EXECUTING TASKS WITH COMPREHENSIVE GUARDRAILS")
print("="*80)
result = agents.start()
print("\n" + "="*80)
print("GUARDRAILS DEMONSTRATION COMPLETED")
print("="*80)
print("This example demonstrated:")
print("- Function-based validation (email format, word count, tone, factual accuracy)")
print("- LLM-based content validation using natural language criteria")
print("- Automatic retry mechanisms with specific feedback")
print("- Quality assurance workflows for different content types")
print("- Professional validation standards for business communications")
@@ -0,0 +1,65 @@
"""
Production Guardrails Patterns Example
This example demonstrates production guardrail patterns using PraisonAI's
built-in validation and safety mechanisms for secure agent operations.
"""
from praisonaiagents import Agent, Task, PraisonAIAgents
from praisonaiagents.tools import internet_search
print("=== Production Guardrails Patterns Example ===\n")
# Simple guardrail function for content validation
def content_guardrail(response, context):
"""Simple content validation guardrail"""
if not response or len(response.strip()) < 50:
return {
"valid": False,
"reason": "Response too short for production use",
"action": "regenerate"
}
# Check for sensitive patterns (simplified)
sensitive_keywords = ["password", "secret", "private_key", "api_key"]
if any(keyword in response.lower() for keyword in sensitive_keywords):
return {
"valid": False,
"reason": "Potential sensitive information detected",
"action": "review_and_redact"
}
return {"valid": True}
# Create production-ready agent with guardrails
production_agent = Agent(
name="Production Agent",
role="Production Content Generator",
goal="Generate safe, compliant content for production use",
backstory="Production-ready agent with built-in safety and compliance checks",
tools=[internet_search],
guardrail=content_guardrail,
max_retries=3,
verbose=True
)
# Create secure task with validation
secure_task = Task(
description="""Create a professional report about renewable energy trends:
1. Research current renewable energy market developments
2. Include factual data and reliable sources
3. Ensure content is production-ready and compliant
4. Avoid any sensitive or inappropriate information
Focus on creating safe, accurate, professional content.""",
expected_output="Professional renewable energy report meeting production standards",
agent=production_agent
)
# Run with guardrail validation
print("Starting production guardrails demonstration...")
result = production_agent.execute_task(secure_task)
print(f"\nProduction Result: {result[:200]}...")
print("\n✅ Production guardrails validation complete!")
print("Agent demonstrated safe content generation with built-in guardrail validation.")
@@ -0,0 +1,194 @@
"""
Advanced Knowledge Management Example
This example demonstrates sophisticated knowledge management capabilities including
advanced chunking strategies, reranking, and intelligent knowledge retrieval.
Features demonstrated:
- Advanced document chunking strategies
- Vector store integration with ChromaDB
- Knowledge-based question answering
- Intelligent context retrieval
- Document processing and management
"""
from praisonaiagents import Agent
import tempfile
import os
# Create a sample knowledge document for demonstration
sample_document = """
# Artificial Intelligence in Healthcare
## Introduction
Artificial intelligence (AI) is revolutionizing healthcare by enabling more accurate diagnoses,
personalized treatments, and efficient healthcare delivery. This document explores the current
applications and future potential of AI in healthcare.
## Current Applications
### Medical Imaging
AI algorithms, particularly deep learning models, have shown remarkable success in medical imaging:
- Radiology: AI can detect cancers in mammograms with accuracy matching or exceeding radiologists
- Ophthalmology: AI systems can diagnose diabetic retinopathy from retinal photographs
- Pathology: AI assists in analyzing tissue samples for cancer detection
### Drug Discovery
AI is accelerating drug discovery processes:
- Molecular design: AI predicts molecular properties and drug-target interactions
- Clinical trials: AI optimizes patient recruitment and trial design
- Repurposing: AI identifies new uses for existing drugs
### Predictive Analytics
Healthcare providers use AI for:
- Risk stratification: Identifying patients at risk of complications
- Early warning systems: Detecting deteriorating patient conditions
- Resource planning: Optimizing staff schedules and equipment allocation
## Challenges and Considerations
### Data Privacy and Security
Healthcare AI faces significant challenges:
- Patient data protection and HIPAA compliance
- Secure data sharing between institutions
- Consent management for AI applications
### Regulatory Approval
AI medical devices require:
- FDA approval for diagnostic tools
- Clinical validation studies
- Post-market surveillance
### Ethical Considerations
Important ethical aspects include:
- Algorithmic bias and fairness
- Transparency and explainability
- Patient autonomy and informed consent
## Future Directions
### Personalized Medicine
AI will enable:
- Genomic-based treatment selection
- Individualized drug dosing
- Precision diagnostics
### Integrated Healthcare Systems
Future developments include:
- AI-powered electronic health records
- Interoperable AI systems across healthcare networks
- Real-time decision support systems
## Conclusion
AI has tremendous potential to transform healthcare, but successful implementation requires
addressing technical, regulatory, and ethical challenges while maintaining focus on patient safety
and improved outcomes.
"""
# Save the sample document to a temporary file
temp_dir = tempfile.mkdtemp()
doc_path = os.path.join(temp_dir, "ai_healthcare_guide.txt")
with open(doc_path, "w") as f:
f.write(sample_document)
# Create an agent with advanced knowledge management configuration
knowledge_agent = Agent(
name="AdvancedKnowledgeAgent",
role="Healthcare AI Knowledge Expert",
goal="Provide comprehensive answers about AI in healthcare using advanced knowledge retrieval",
backstory="You are an expert in healthcare AI with access to comprehensive knowledge bases and advanced retrieval capabilities.",
# Advanced knowledge configuration
knowledge=[doc_path],
knowledge_config={
"vector_store": {
"provider": "chroma",
"collection_name": "healthcare_ai_knowledge"
},
"chunking": {
"strategy": "semantic", # Advanced semantic chunking
"chunk_size": 500, # Optimal chunk size for retrieval
"chunk_overlap": 50, # Overlap for context preservation
"separators": ["\n\n", "\n", ".", "!", "?"] # Smart separators
},
"retrieval": {
"search_type": "similarity",
"k": 5, # Retrieve top 5 most relevant chunks
"score_threshold": 0.7, # Minimum relevance score
"rerank": True # Enable reranking for better results
},
"embedding": {
"provider": "openai",
"model": "text-embedding-3-small", # Efficient embedding model
"dimensions": 1536
}
},
instructions="""You are an expert in healthcare AI. Use the knowledge base to provide
comprehensive, accurate answers. Always cite specific information from the documents
when possible. If the knowledge base doesn't contain sufficient information, clearly
state this and provide general guidance based on your training.""",
verbose=True
)
# Test advanced knowledge retrieval with various question types
# Factual question about specific applications
print("="*70)
print("TESTING: Specific factual question about AI applications")
print("="*70)
factual_result = knowledge_agent.start(
"What are the current applications of AI in medical imaging according to the knowledge base?"
)
print(f"Factual query result:\n{factual_result}\n")
# Complex analytical question requiring synthesis
print("="*70)
print("TESTING: Complex analytical question requiring synthesis")
print("="*70)
analytical_result = knowledge_agent.start(
"What are the main challenges facing AI implementation in healthcare and how do they relate to each other?"
)
print(f"Analytical query result:\n{analytical_result}\n")
# Specific detail extraction
print("="*70)
print("TESTING: Specific detail extraction")
print("="*70)
detail_result = knowledge_agent.start(
"What specific regulatory requirements does the document mention for AI medical devices?"
)
print(f"Detail extraction result:\n{detail_result}\n")
# Future-oriented question requiring inference
print("="*70)
print("TESTING: Future-oriented question requiring inference")
print("="*70)
future_result = knowledge_agent.start(
"Based on the knowledge base, what integration capabilities will future AI healthcare systems have?"
)
print(f"Future-oriented query result:\n{future_result}\n")
# Question about relationships and connections
print("="*70)
print("TESTING: Relationship and connection analysis")
print("="*70)
relationship_result = knowledge_agent.start(
"How do the ethical considerations mentioned relate to the regulatory approval process for AI in healthcare?"
)
print(f"Relationship analysis result:\n{relationship_result}\n")
# Clean up temporary files
import shutil
shutil.rmtree(temp_dir)
print("="*80)
print("ADVANCED KNOWLEDGE MANAGEMENT DEMONSTRATION COMPLETED")
print("="*80)
print("This example demonstrated:")
print("- Semantic chunking for better context preservation")
print("- Advanced retrieval with reranking capabilities")
print("- Multiple query types with intelligent responses")
print("- Knowledge base integration with vector search")
print("- Configurable embedding and retrieval parameters")
@@ -0,0 +1,75 @@
"""
Advanced Graph Memory Integration Example
This example demonstrates graph memory capabilities with PraisonAI agents
for knowledge graph construction and relationship-aware memory retrieval.
"""
from praisonaiagents import Agent, Task, PraisonAIAgents
from praisonaiagents.tools import internet_search
print("=== Advanced Graph Memory Integration Example ===\n")
# Configure simple graph memory
memory_config = {
"provider": "mem0",
"config": {
"vector_store": {
"provider": "chroma",
"config": {
"path": ".praison/graph_memory"
}
}
}
}
# Create knowledge builder agent with graph memory
knowledge_agent = Agent(
name="Knowledge Agent",
role="Knowledge Graph Builder",
goal="Build and retrieve from knowledge graphs",
backstory="Expert at creating structured knowledge and finding relationships",
tools=[internet_search],
memory=True,
verbose=True
)
# Task to build knowledge graph about AI companies
build_task = Task(
description="""Research and store information about major AI companies and their relationships:
1. OpenAI - products like GPT-4, ChatGPT, leadership, partnerships
2. Anthropic - Claude AI, safety focus, team
3. Relationships between these companies and their technologies
Focus on capturing connections and relationships between entities.""",
expected_output="Knowledge graph with AI company ecosystem relationships",
agent=knowledge_agent
)
# Task to query the knowledge graph
query_task = Task(
description="""Using the stored knowledge graph, answer:
1. What are the main products of each company?
2. Who are the key leaders in these companies?
3. How do these companies relate to each other?
Use relationship-aware memory retrieval.""",
expected_output="Analysis using graph memory relationships",
agent=knowledge_agent
)
# Run with graph memory integration
agents_system = PraisonAIAgents(
agents=[knowledge_agent],
tasks=[build_task, query_task],
memory=True,
memory_config=memory_config,
verbose=True
)
print("Starting graph memory demonstration...")
result = agents_system.start()
print(f"\nGraph Memory Result: {result[:200]}...")
print("\n✅ Graph memory integration complete!")
print("Agent built knowledge graph and performed relationship-aware queries.")
@@ -0,0 +1,76 @@
"""
Advanced Memory Search and Optimization Example
This example demonstrates memory search optimization using PraisonAI's
built-in memory capabilities with semantic search and quality filtering.
"""
from praisonaiagents import Agent, Task, PraisonAIAgents
from praisonaiagents.tools import internet_search
print("=== Advanced Memory Search and Optimization Example ===\n")
# Configure optimized memory settings
memory_config = {
"provider": "mem0",
"config": {
"vector_store": {
"provider": "chroma",
"config": {
"path": ".praison/optimized_memory"
}
}
}
}
# Create research agent with optimized memory
research_agent = Agent(
name="Memory Research Agent",
role="Information Researcher with Memory Optimization",
goal="Research and demonstrate optimized memory search patterns",
backstory="Expert at storing and retrieving information efficiently with memory optimization",
tools=[internet_search],
memory=True,
verbose=True
)
# Task to build searchable memory
build_memory_task = Task(
description="""Research and store information about these topics for later search:
1. Artificial intelligence market trends and growth data
2. Machine learning technologies and applications
3. Tech industry partnerships and investments
4. Cloud computing adoption statistics
Store detailed information that can be searched later.""",
expected_output="Comprehensive research stored in optimized memory",
agent=research_agent
)
# Task to demonstrate memory search optimization
search_task = Task(
description="""Using the stored memory, perform optimized searches for:
1. "AI market growth trends" - find relevant stored information
2. "technology partnerships" - retrieve partnership data
3. "cloud adoption statistics" - search for cloud computing info
Demonstrate efficient memory retrieval and search optimization.""",
expected_output="Results from optimized memory searches",
agent=research_agent
)
# Run with memory optimization
agents_system = PraisonAIAgents(
agents=[research_agent],
tasks=[build_memory_task, search_task],
memory=True,
memory_config=memory_config,
verbose=True
)
print("Starting memory search optimization demonstration...")
result = agents_system.start()
print(f"\nMemory Search Result: {result[:200]}...")
print("\n✅ Memory search optimization complete!")
print("Agent demonstrated efficient memory storage and optimized search retrieval.")
@@ -0,0 +1,153 @@
#!/usr/bin/env python3
"""
Example: Using Gemini Embedding Models with PraisonAI Memory
This example demonstrates how to configure PraisonAI to use Google's Gemini
embedding models for memory storage and retrieval, as requested in issue #870.
Supported Gemini embedding models:
- gemini-embedding-exp-03-07 (experimental)
- text-embedding-004 (stable)
Prerequisites:
1. Install praisonaiagents with memory support:
pip install "praisonaiagents[memory]"
2. Set your Google API key:
export GOOGLE_API_KEY='your-api-key'
3. Ensure litellm is installed (comes with praisonaiagents)
"""
import os
from praisonaiagents import Agent, Task, PraisonAIAgents
def main():
"""Demonstrate Gemini embedding usage"""
# Verify Google API key
if not os.environ.get("GOOGLE_API_KEY"):
print("Error: GOOGLE_API_KEY environment variable not set")
print("Please set it with: export GOOGLE_API_KEY='your-api-key'")
return
print("=== Gemini Embedding Example ===\n")
# Create agents for different tasks
researcher = Agent(
name="Research Agent",
role="Information Researcher",
goal="Research and store important information about topics",
backstory="Expert at analyzing and documenting information with semantic understanding",
llm="gpt-4o-mini" # Using GPT for chat, Gemini for embeddings
)
retriever = Agent(
name="Retrieval Agent",
role="Information Retriever",
goal="Retrieve relevant information from memory using semantic search",
backstory="Specialist in finding and presenting stored information",
llm="gpt-4o-mini"
)
# Task 1: Store information with semantic meaning
store_task = Task(
description="""
Research and store the following information:
1. Gemini is Google's family of multimodal AI models
2. Gemini models support text, image, video, and audio understanding
3. The embedding models provide semantic representations of text
4. Gemini Ultra is the largest and most capable model
5. Gemini was announced in December 2023
Store each fact with its semantic meaning preserved.
""",
expected_output="Confirmation that all facts have been stored in memory",
agent=researcher
)
# Task 2: Semantic retrieval test
retrieve_task = Task(
description="""
Using semantic search, find information about:
1. Google's AI capabilities
2. Multimodal understanding features
3. When was Google's latest AI announced
Note: Use semantic similarity, not exact keyword matching.
""",
expected_output="Retrieved information based on semantic meaning",
agent=retriever
)
# Configure with Gemini embeddings
# Option 1: Using stable text-embedding-004
agents = PraisonAIAgents(
agents=[researcher, retriever],
tasks=[store_task, retrieve_task],
verbose=True,
memory=True,
embedder={
"provider": "gemini",
"config": {
"model": "text-embedding-004",
# Optional: specify task type for better results
# "task_type": "RETRIEVAL_DOCUMENT"
}
}
)
# Alternative: Using experimental model with advanced features
# agents = PraisonAIAgents(
# agents=[researcher, retriever],
# tasks=[store_task, retrieve_task],
# verbose=True,
# memory=True,
# embedder={
# "provider": "gemini",
# "config": {
# "model": "gemini-embedding-exp-03-07",
# # Task types: SEMANTIC_SIMILARITY, CLASSIFICATION, CLUSTERING,
# # RETRIEVAL_DOCUMENT, RETRIEVAL_QUERY, QUESTION_ANSWERING,
# # FACT_VERIFICATION
# "task_type": "RETRIEVAL_DOCUMENT"
# }
# }
# )
# Execute the workflow
print("Starting agents with Gemini embeddings...\n")
result = agents.start()
# Demonstrate direct memory access with Gemini embeddings
print("\n=== Direct Memory Search Example ===\n")
if agents.shared_memory:
# Search queries that test semantic understanding
queries = [
"What are Google's latest AI innovations?",
"Explain multimodal AI capabilities",
"Large language model features"
]
for query in queries:
print(f"\nSearching for: '{query}'")
results = agents.shared_memory.search_long_term(query, limit=3)
if results:
print(f"Found {len(results)} semantically similar results:")
for i, result in enumerate(results, 1):
print(f"\n{i}. Relevance Score: {result.get('score', 0):.3f}")
print(f" Content: {result.get('text', '')[:150]}...")
else:
print("No results found")
print("\n=== Example Complete ===")
print("\nKey Points:")
print("1. Gemini embeddings provide semantic understanding of text")
print("2. Results are based on meaning, not just keyword matching")
print("3. Different task types can optimize embedding quality")
print("4. Compatible with LiteLLM for easy integration")
if __name__ == "__main__":
main()
+165
View File
@@ -0,0 +1,165 @@
# Gemini Embedding Support in PraisonAI
This document describes how to use Google's Gemini embedding models with PraisonAI's memory system.
## Overview
PraisonAI now supports configurable embedding models through LiteLLM, including Google's Gemini embedding models. This allows you to use models like `text-embedding-004` or `gemini-embedding-exp-03-07` for semantic search and memory storage.
## Configuration
### Default Configuration (OpenAI)
By default, PraisonAI uses OpenAI's `text-embedding-3-small` model:
```python
from praisonaiagents import PraisonAIAgents
agents = PraisonAIAgents(
agents=[...],
tasks=[...],
memory=True # Uses default OpenAI embeddings
)
```
### Using Gemini Embeddings
To use Gemini embeddings, configure the `embedder` parameter:
```python
from praisonaiagents import PraisonAIAgents
agents = PraisonAIAgents(
agents=[...],
tasks=[...],
memory=True,
embedder={
"provider": "gemini",
"config": {
"model": "text-embedding-004" # or "gemini-embedding-exp-03-07"
}
}
)
```
## Prerequisites
1. **API Key**: Set your Google API key:
```bash
export GOOGLE_API_KEY='your-api-key'
```
2. **Dependencies**: Ensure you have the memory dependencies installed:
```bash
pip install "praisonaiagents[memory]"
```
## Supported Models
### Gemini Embedding Models
- `text-embedding-004` - Stable embedding model
- `gemini-embedding-exp-03-07` - Experimental model with advanced features
### Task Types (Optional)
Gemini models support task-specific optimizations:
- `SEMANTIC_SIMILARITY`
- `CLASSIFICATION`
- `CLUSTERING`
- `RETRIEVAL_DOCUMENT`
- `RETRIEVAL_QUERY`
- `QUESTION_ANSWERING`
- `FACT_VERIFICATION`
Example with task type:
```python
embedder={
"provider": "gemini",
"config": {
"model": "text-embedding-004",
"task_type": "RETRIEVAL_DOCUMENT"
}
}
```
## Complete Example
```python
import os
from praisonaiagents import Agent, Task, PraisonAIAgents
# Ensure Google API key is set
os.environ["GOOGLE_API_KEY"] = "your-api-key"
# Create agents
researcher = Agent(
name="Researcher",
role="Information Researcher",
goal="Store information with semantic understanding",
llm="gpt-4o-mini"
)
# Create task
task = Task(
description="Research and store information about AI advancements",
expected_output="Confirmation of stored information",
agent=researcher
)
# Initialize with Gemini embeddings
agents = PraisonAIAgents(
agents=[researcher],
tasks=[task],
memory=True,
embedder={
"provider": "gemini",
"config": {
"model": "text-embedding-004"
}
}
)
# Run the agents
agents.start()
```
## Backward Compatibility
The implementation maintains full backward compatibility:
1. **No embedder config**: Uses default OpenAI `text-embedding-3-small`
2. **Existing configs**: Continue to work without modification
3. **Other providers**: Supports any embedding model available through LiteLLM
## Memory Configuration with Mem0
When using Mem0 as the memory provider, the embedder configuration is passed through:
```python
memory_config = {
"provider": "mem0",
"config": {
"api_key": "your-mem0-key",
"embedder": {
"provider": "gemini",
"config": {
"model": "text-embedding-004"
}
}
}
}
```
## Troubleshooting
1. **API Key Error**: Ensure `GOOGLE_API_KEY` is set in your environment
2. **Model Not Found**: Verify the model name is correct
3. **LiteLLM Error**: Ensure litellm is installed and up to date
4. **Embedding Size Mismatch**: When switching models, you may need to clear the vector database as different models produce different embedding dimensions
## See Also
- [Gemini API Documentation](https://ai.google.dev/gemini-api/docs/embeddings)
- [LiteLLM Embedding Models](https://docs.litellm.ai/docs/embedding/supported_embedding)
- Example: `/examples/python/memory/gemini-embedding-example.py`
@@ -0,0 +1,215 @@
# Gemini Internal Tools Examples
This directory contains examples demonstrating how to use Google Gemini's built-in internal tools through PraisonAI. These tools are natively supported by Gemini models and provide powerful capabilities without requiring external tool implementations.
## Available Internal Tools
### 1. Google Search (`googleSearch`)
- **Purpose**: Real-time web search with automatic result grounding
- **Use Cases**: Current events, latest information, fact-checking
- **Example**: `gemini-google-search.py`
### 2. URL Context (`urlContext`)
- **Purpose**: Fetch and analyze content from specific URLs
- **Use Cases**: Document analysis, web content summarization, research
- **Example**: `gemini-url-context.py`
### 3. Code Execution (`codeExecution`)
- **Purpose**: Execute Python code snippets within the conversation
- **Use Cases**: Calculations, data analysis, code validation
- **Example**: `gemini-code-execution.py`
## Prerequisites
1. **Gemini API Key**: Set your Google AI Studio API key
```bash
export GEMINI_API_KEY="your-api-key-here"
```
2. **Compatible Model**: Use Gemini models that support internal tools:
- `gemini/gemini-2.0-flash` (recommended)
- `gemini/gemini-2.0-flash-thinking-exp`
- Other Gemini 2.0+ models
3. **PraisonAI Installation**: Ensure you have the latest version
```bash
pip install "praisonaiagents[llm]"
```
## Examples
### Individual Tool Examples
1. **Google Search Example** (`gemini-google-search.py`)
```python
agent = Agent(
instructions="Research assistant with web search",
llm="gemini/gemini-2.0-flash",
tools=[{"googleSearch": {}}]
)
response = agent.start("What's the latest news about AI?")
```
2. **URL Context Example** (`gemini-url-context.py`)
```python
agent = Agent(
instructions="Content analyzer",
llm="gemini/gemini-2.0-flash",
tools=[{"urlContext": {}}]
)
response = agent.start("Summarize this: https://example.com/article")
```
3. **Code Execution Example** (`gemini-code-execution.py`)
```python
agent = Agent(
instructions="Data analyst with code execution",
llm="gemini/gemini-2.0-flash",
tools=[{"codeExecution": {}}]
)
response = agent.start("Calculate compound interest for $10,000 at 5% for 10 years")
```
### Combined Tools Example
**All Tools Together** (`gemini-internal-tools-combined.py`)
```python
agent = Agent(
instructions="Advanced research assistant",
llm="gemini/gemini-2.0-flash",
tools=[
{"googleSearch": {}},
{"urlContext": {}},
{"codeExecution": {}}
]
)
# Now the agent can search, analyze URLs, and execute code as needed
```
## How It Works
### PraisonAI Integration
PraisonAI's tool system is designed to pass tools directly to LiteLLM without modification:
1. **Tool Definition**: Define tools using the special internal tool syntax
```python
tools=[{"googleSearch": {}}] # Special format for internal tools
```
2. **Pass-Through**: PraisonAI passes these tools directly to LiteLLM
3. **Execution**: LiteLLM sends them to Gemini as internal tool configurations
4. **Results**: Gemini executes the tools natively and returns integrated responses
### Mixing with External Tools
You can combine internal tools with custom external tools:
```python
def custom_calculator(expression: str) -> str:
"""Custom calculator function - safely evaluates basic math expressions"""
import ast
import operator
# Define safe operations
ops = {
ast.Add: operator.add,
ast.Sub: operator.sub,
ast.Mult: operator.mul,
ast.Div: operator.truediv,
ast.Pow: operator.pow,
ast.USub: operator.neg,
}
def eval_expr(node):
if isinstance(node, ast.Constant): # Numbers
return node.value
elif isinstance(node, ast.BinOp): # Binary operations
return ops[type(node.op)](eval_expr(node.left), eval_expr(node.right))
elif isinstance(node, ast.UnaryOp): # Unary operations
return ops[type(node.op)](eval_expr(node.operand))
else:
raise TypeError(f"Unsupported operation: {type(node)}")
try:
return str(eval_expr(ast.parse(expression, mode='eval').body))
except Exception as e:
return f"Error: {e}"
agent = Agent(
tools=[
{"googleSearch": {}}, # Internal tool
{"codeExecution": {}}, # Internal tool
custom_calculator # External tool
]
)
```
## Running the Examples
1. **Set API Key**:
```bash
export GEMINI_API_KEY="your-api-key"
```
2. **Run Individual Examples**:
```bash
python gemini-google-search.py
python gemini-url-context.py
python gemini-code-execution.py
```
3. **Run Combined Example**:
```bash
python gemini-internal-tools-combined.py
```
## Benefits Over External Tools
### Advantages of Internal Tools
1. **Native Integration**: No need to implement or maintain external tool code
2. **Optimized Performance**: Tools are executed within Gemini's environment
3. **Automatic Grounding**: Search results are automatically integrated into responses
4. **No Rate Limits**: No separate API calls or rate limiting concerns
5. **Security**: Code execution is sandboxed within Gemini's environment
### When to Use External Tools
- When you need specific custom logic
- When you require integration with private APIs
- When you need persistent data storage
- When you want full control over tool behavior
## Troubleshooting
### Common Issues
1. **API Key Error**: Ensure `GEMINI_API_KEY` is set correctly
2. **Model Not Supported**: Use Gemini 2.0+ models for internal tools
3. **Tool Not Available**: Some tools may have regional or quota restrictions
### Debug Mode
Enable verbose logging to see tool usage:
```python
agent = Agent(
# ... config
verbose=True # Shows detailed tool usage
)
```
## References
- [Gemini API: Google Search Grounding](https://ai.google.dev/gemini-api/docs/google-search)
- [Gemini API: URL Context](https://ai.google.dev/gemini-api/docs/url-context)
- [Gemini API: Code Execution](https://ai.google.dev/gemini-api/docs/code-execution)
- [LiteLLM Gemini Provider Documentation](https://docs.litellm.ai/docs/providers/gemini)
## Contributing
If you create additional examples or find improvements, please contribute back to the PraisonAI project!
@@ -1,18 +1,12 @@
import asyncio
from praisonaiagents import Agent
async def main():
agent = Agent(
instructions="You are a helpful assistant",
llm="gemini/gemini-1.5-flash-8b",
self_reflect=True,
verbose=True
)
agent = Agent(
instructions="You are a helpful assistant",
llm="gemini/gemini-1.5-flash-8b",
self_reflect=True,
verbose=True
)
# Use achat instead of start/chat for async operation
response = await agent.achat("Why sky is Blue in 1000 words?")
print(response)
if __name__ == "__main__":
# Run the async main function
asyncio.run(main())
# Use achat instead of start/chat for async operation
response = asyncio.run(agent.achat("Why sky is Blue in 1000 words?"))
@@ -0,0 +1,99 @@
"""
Gemini Code Execution Tool Example
This example demonstrates how to use Gemini's built-in Code Execution functionality
through PraisonAI. The Code Execution tool allows the model to run Python code
snippets directly within the conversation for dynamic computation and validation.
Prerequisites:
- Set GEMINI_API_KEY environment variable
- Use a Gemini model that supports internal tools (gemini-2.0-flash, etc.)
Features:
- Direct Python code execution within the model
- Real-time calculations and data processing
- Code validation and testing
- Mathematical computations and visualizations
"""
from praisonaiagents import Agent
# Ensure you have your Gemini API key set
# import os; os.environ["GEMINI_API_KEY"] = "your-api-key-here"
def main():
# Create agent with Code Execution internal tool
agent = Agent(
instructions="""You are a data analyst and programmer that can execute Python code.
Use the Code Execution tool to perform calculations, analyze data, create visualizations,
and validate logic. Always show your work by executing relevant code snippets.""",
llm="gemini/gemini-2.0-flash",
# Enable Code Execution internal tool
tools=[{"codeExecution": {}}],
verbose=True
)
# Example tasks that benefit from code execution
tasks = [
"Calculate the compound interest on $10,000 invested at 5% annual rate for 10 years, compounded monthly. Show the calculation step by step.",
"Generate the first 20 Fibonacci numbers and find their sum. Also calculate the ratio of consecutive Fibonacci numbers.",
"Create a simple linear regression model for the data points: (1,2), (2,4), (3,6), (4,8), (5,10). Calculate the R-squared value.",
"Solve the quadratic equation 2x² + 5x - 3 = 0 using the quadratic formula. Verify the solutions by substituting back.",
"Calculate the statistical summary (mean, median, mode, standard deviation) for this dataset: [12, 15, 18, 12, 20, 25, 18, 12, 22, 30, 15, 18]",
"Write a function to check if a number is prime, then find all prime numbers between 1 and 100.",
"Calculate the area under the curve y = x² from x = 0 to x = 5 using numerical integration (trapezoidal rule)."
]
print("=== Gemini Code Execution Tool Demo ===\n")
for i, task in enumerate(tasks, 1):
print(f"Task {i}: {task}")
print("-" * 80)
try:
response = agent.start(task)
print(f"Response: {response}")
print("\n" + "="*100 + "\n")
except Exception as e:
print(f"Error: {e}")
print("\n" + "="*100 + "\n")
def demonstrate_direct_usage():
"""
Example of how the Code Execution tool works at the LiteLLM level
(for reference - this requires direct LiteLLM usage)
"""
print("=== Direct LiteLLM Code Execution Usage (Reference) ===")
print("""
# This is how it works at the LiteLLM level:
import litellm
import os
os.environ["GEMINI_API_KEY"] = "your-api-key"
response = litellm.completion(
model="gemini/gemini-2.0-flash",
messages=[{
"role": "user",
"content": "Calculate the factorial of 10 and show the code"
}],
tools=[{"codeExecution": {}}],
)
print(response.choices[0].message.content)
""")
if __name__ == "__main__":
main()
demonstrate_direct_usage()
@@ -0,0 +1,63 @@
"""
Gemini Google Search Tool Example
This example demonstrates how to use Gemini's built-in Google Search functionality
through PraisonAI. The Google Search tool allows the model to perform real-time
web searches and ground responses in current information.
Prerequisites:
- Set GEMINI_API_KEY environment variable
- Use a Gemini model that supports internal tools (gemini-2.0-flash, etc.)
Features:
- Real-time web search capabilities
- Automatic grounding of responses in search results
- No need to implement external search tools
"""
from praisonaiagents import Agent
# Ensure you have your Gemini API key set
# import os; os.environ["GEMINI_API_KEY"] = "your-api-key-here"
def main():
# Create agent with Google Search internal tool
agent = Agent(
instructions="""You are a research assistant that can search the web for current information.
Use the Google Search tool to find up-to-date information when answering questions.
Always cite your sources and indicate when information comes from search results.""",
llm="gemini/gemini-2.0-flash",
# Enable Google Search internal tool
tools=[{"googleSearch": {}}],
verbose=True
)
# Example queries that benefit from real-time search
queries = [
"What's the latest news about AI developments this week?",
"What's the current weather in San Francisco?",
"What are the latest stock prices for major tech companies?",
"What are the most recent breakthroughs in quantum computing?",
"Who won the latest major sports championships?"
]
print("=== Gemini Google Search Tool Demo ===\n")
for i, query in enumerate(queries, 1):
print(f"Query {i}: {query}")
print("-" * 50)
try:
response = agent.start(query)
print(f"Response: {response}")
print("\n" + "="*80 + "\n")
except Exception as e:
print(f"Error: {e}")
print("\n" + "="*80 + "\n")
if __name__ == "__main__":
main()
@@ -0,0 +1,187 @@
"""
Gemini Internal Tools Combined Example
This example demonstrates how to use all three of Gemini's built-in internal tools
together through PraisonAI:
1. Google Search - for real-time web information
2. URL Context - for analyzing web content
3. Code Execution - for computations and analysis
Prerequisites:
- Set GEMINI_API_KEY environment variable
- Use a Gemini model that supports internal tools (gemini-2.0-flash, etc.)
Features:
- Multi-tool agent combining search, content analysis, and computation
- Real-world research and analysis workflows
- Comprehensive data gathering and processing capabilities
"""
from praisonaiagents import Agent
# Ensure you have your Gemini API key set
# import os; os.environ["GEMINI_API_KEY"] = "your-api-key-here"
def main():
# Create agent with all Gemini internal tools enabled
agent = Agent(
instructions="""You are an advanced research and analysis assistant with access to:
1. Google Search - for finding current information
2. URL Context - for analyzing web content
3. Code Execution - for calculations and data processing
Use these tools strategically to provide comprehensive, well-researched, and
analytically sound responses. Always show your work and cite sources.""",
llm="gemini/gemini-2.0-flash",
# Enable all three Gemini internal tools
tools=[
{"googleSearch": {}},
{"urlContext": {}},
{"codeExecution": {}}
],
verbose=True
)
# Complex tasks that benefit from multiple internal tools
complex_tasks = [
"""Research the current market cap of the top 5 technology companies,
then calculate their combined market value and determine what percentage
each company represents of the total. Present the results in a formatted table.""",
"""Find recent research papers about large language models from arXiv,
analyze the abstracts of 2-3 papers, and calculate the average number
of authors across these papers.""",
"""Look up the current population of the 10 largest cities in the world,
then calculate statistical measures (mean, median, standard deviation)
and create a simple visualization of the data distribution.""",
"""Research the latest Python release notes, analyze the key new features,
and write example code demonstrating 2-3 of the most important new features.""",
"""Find information about recent climate data, analyze a specific dataset
or report, and perform calculations to show trends or comparisons over time."""
]
print("=== Gemini Combined Internal Tools Demo ===\n")
for i, task in enumerate(complex_tasks, 1):
print(f"Complex Task {i}:")
print(task)
print("-" * 100)
try:
response = agent.start(task)
print(f"Response: {response}")
print("\n" + "="*120 + "\n")
except Exception as e:
print(f"Error: {e}")
print("\n" + "="*120 + "\n")
def demonstrate_tool_synergy():
"""
Example showing how the tools work together for a research workflow
"""
print("=== Tool Synergy Example ===")
agent = Agent(
instructions="""You are a market research analyst. Use all available tools
to gather information, analyze content, and perform calculations as needed.""",
llm="gemini/gemini-2.0-flash",
tools=[
{"googleSearch": {}},
{"urlContext": {}},
{"codeExecution": {}}
],
verbose=True
)
# Multi-step research task
research_query = """
Conduct a mini market analysis on electric vehicle adoption:
1. Search for recent statistics on global EV sales
2. Find and analyze a recent report or article about EV market trends
3. Calculate the year-over-year growth rate if you can find data for multiple years
4. Provide a summary with key insights and numerical analysis
"""
print("Research Query:", research_query)
print("-" * 80)
try:
response = agent.start(research_query)
print(f"Research Results: {response}")
except Exception as e:
print(f"Error: {e}")
def show_configuration_options():
"""
Show different ways to configure the internal tools
"""
print("\n=== Configuration Examples ===")
print("""
# Basic usage (all tools with default settings)
tools=[
{"googleSearch": {}},
{"urlContext": {}},
{"codeExecution": {}}
]
# Can also be used individually
tools=[{"googleSearch": {}}] # Only Google Search
tools=[{"urlContext": {}}] # Only URL Context
tools=[{"codeExecution": {}}] # Only Code Execution
# Mix with custom tools
from praisonaiagents import Agent
def custom_calculator(expression: str) -> str:
'''Custom calculator function - safely evaluates basic math expressions'''
import ast
import operator
# Define safe operations
ops = {
ast.Add: operator.add,
ast.Sub: operator.sub,
ast.Mult: operator.mul,
ast.Div: operator.truediv,
ast.Pow: operator.pow,
ast.USub: operator.neg,
}
def eval_expr(node):
if isinstance(node, ast.Constant): # Numbers
return node.value
elif isinstance(node, ast.BinOp): # Binary operations
return ops[type(node.op)](eval_expr(node.left), eval_expr(node.right))
elif isinstance(node, ast.UnaryOp): # Unary operations
return ops[type(node.op)](eval_expr(node.operand))
else:
raise TypeError(f"Unsupported operation: {type(node)}")
try:
return str(eval_expr(ast.parse(expression, mode='eval').body))
except Exception as e:
return f"Error: {e}"
agent = Agent(
tools=[
{"googleSearch": {}}, # Gemini internal tool
{"codeExecution": {}}, # Gemini internal tool
custom_calculator # Custom external tool
]
)
""")
if __name__ == "__main__":
main()
print("\n" + "="*50)
demonstrate_tool_synergy()
show_configuration_options()
@@ -0,0 +1,115 @@
"""
Gemini URL Context Tool Example
This example demonstrates how to use Gemini's built-in URL Context functionality
through PraisonAI. The URL Context tool allows the model to fetch and analyze
content from specific URLs in real-time.
Prerequisites:
- Set GEMINI_API_KEY environment variable
- Use a Gemini model that supports internal tools (gemini-2.0-flash, etc.)
Features:
- Real-time URL content fetching
- Automatic content analysis and summarization
- Support for various web content types
- Access to URL metadata through response.model_extra
"""
from praisonaiagents import Agent
# Ensure you have your Gemini API key set
# import os; os.environ["GEMINI_API_KEY"] = "your-api-key-here"
def main():
# Create agent with URL Context internal tool
agent = Agent(
instructions="""You are a content analyst that can fetch and analyze web content.
Use the URL Context tool to access and analyze content from provided URLs.
Provide detailed summaries, extract key information, and answer questions about the content.""",
llm="gemini/gemini-2.0-flash",
# Enable URL Context internal tool
tools=[{"urlContext": {}}],
verbose=True
)
# Example URLs for analysis
url_tasks = [
{
"url": "https://ai.google.dev/gemini-api/docs/models",
"task": "Summarize the key information about Gemini models and their capabilities"
},
{
"url": "https://github.com/MervinPraison/PraisonAI",
"task": "Analyze this GitHub repository and explain what PraisonAI does"
},
{
"url": "https://docs.python.org/3/tutorial/introduction.html",
"task": "Extract the main learning objectives from this Python tutorial"
},
{
"url": "https://arxiv.org/abs/2301.00234",
"task": "Provide a summary of this research paper's abstract and key contributions"
}
]
print("=== Gemini URL Context Tool Demo ===\n")
for i, item in enumerate(url_tasks, 1):
print(f"Analysis {i}: {item['url']}")
print(f"Task: {item['task']}")
print("-" * 50)
# Combine URL and task in the query
query = f"{item['task']}: {item['url']}"
try:
response = agent.start(query)
print(f"Analysis: {response}")
# Note: URL metadata access requires direct LiteLLM response
# In PraisonAI, this would need to be accessed through the LLM response
print(f"\nAnalyzed URL: {item['url']}")
print("\n" + "="*80 + "\n")
except Exception as e:
print(f"Error: {e}")
print("\n" + "="*80 + "\n")
def demonstrate_direct_usage():
"""
Example of how the URL Context tool works at the LiteLLM level
(for reference - this requires direct LiteLLM usage)
"""
print("=== Direct LiteLLM URL Context Usage (Reference) ===")
print("""
# This is how it works at the LiteLLM level:
import litellm
import os
os.environ["GEMINI_API_KEY"] = "your-api-key"
response = litellm.completion(
model="gemini/gemini-2.0-flash",
messages=[{
"role": "user",
"content": "Summarize this document: https://ai.google.dev/gemini-api/docs/models"
}],
tools=[{"urlContext": {}}],
)
# Access URL metadata
if hasattr(response, 'model_extra') and 'vertex_ai_url_context_metadata' in response.model_extra:
url_context_metadata = response.model_extra['vertex_ai_url_context_metadata']
urlMetadata = url_context_metadata[0]['urlMetadata'][0]
print(f"Retrieved URL: {urlMetadata['retrievedUrl']}")
print(f"Retrieval Status: {urlMetadata['urlRetrievalStatus']}")
""")
if __name__ == "__main__":
main()
demonstrate_direct_usage()
@@ -0,0 +1,13 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a competitor intelligence AI agent. "
"Help users analyze competitors, market trends, and business intelligence "
"to provide strategic insights and competitive advantages.",
llm="xai/grok-4"
)
response = agent.start(
"I need to analyze my main competitor's pricing strategy and market positioning. "
"Can you help me gather insights?"
)
@@ -0,0 +1,13 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a finance AI agent. "
"Help users with financial analysis, investment strategies, "
"budget planning, and market trend analysis for informed decision making.",
llm="xai/grok-4"
)
response = agent.start(
"I need to analyze my investment portfolio and create a diversification strategy. "
"Can you help me optimize my financial planning?"
)
@@ -0,0 +1,13 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a game design AI agent. "
"Help users with game concept development, mechanics design, "
"and creative storytelling for various gaming platforms and genres.",
llm="xai/grok-4"
)
response = agent.start(
"I want to create a mobile puzzle game concept. "
"Can you help me design engaging gameplay mechanics and story elements?"
)
@@ -0,0 +1,13 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a legal AI agent. "
"Help users with legal document analysis, contract review, "
"and legal advice while ensuring compliance and risk assessment.",
llm="xai/grok-4"
)
response = agent.start(
"I need to review a software licensing agreement. "
"Can you help me identify potential risks and key terms?"
)
@@ -0,0 +1,13 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a recruitment AI agent. "
"Help users with candidate screening, job posting optimization, "
"and recruitment process automation to find the best talent.",
llm="xai/grok-4"
)
response = agent.start(
"I need to create a job posting for a senior software engineer position. "
"Can you help me write an attractive and comprehensive job description?"
)
@@ -0,0 +1,13 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are an AI services agency agent. "
"Help users with comprehensive business services including marketing, "
"content creation, strategy, and operational optimization using CrewAI framework.",
llm="xai/grok-4"
)
response = agent.start(
"I need help with my startup's marketing strategy and content creation. "
"Can you provide a comprehensive plan?"
)
@@ -0,0 +1,13 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are an AI teaching agent. "
"Help users with educational content creation, lesson planning, "
"and personalized learning experiences across various subjects.",
llm="xai/grok-4"
)
response = agent.start(
"I need to create a lesson plan for teaching Python programming to beginners. "
"Can you help me design an engaging curriculum?"
)
@@ -0,0 +1,30 @@
"""
Basic example of using Grok model in PraisonAI
"""
from praisonaiagents import Agent
# Initialize Agent with Grok model
agent = Agent(
instructions="You are a helpful assistant",
llm="xai/grok-beta",
)
# Example conversation
response = agent.start("Hello! Can you explain quantum computing in simple terms?")
# Example with mathematical reasoning
math_task = """
Solve this problem step by step:
If a train travels at 60 mph for 2.5 hours, how far does it travel?
"""
response = agent.start(math_task)
# Example with creative writing
creative_task = """
Write a short story about a robot learning to paint.
Make it engaging and about 100 words.
"""
response = agent.start(creative_task)
@@ -0,0 +1,13 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a multi MCP AI agent. "
"Help users integrate and coordinate multiple Model Context Protocol agents "
"for complex workflows and cross-platform automation.",
llm="xai/grok-4"
)
response = agent.start(
"I need to coordinate multiple tools and services for my project. "
"Can you help me set up an integrated workflow using MCP agents?"
)
@@ -0,0 +1,13 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a multimodal coding AI agent. "
"Help users with code generation, review, and debugging "
"by analyzing both text and visual inputs like screenshots and diagrams.",
llm="xai/grok-4"
)
response = agent.start(
"I have a screenshot of a UI design and need to generate the corresponding HTML/CSS code. "
"Can you help me create the frontend code?"
)
@@ -0,0 +1,13 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a multimodal design AI agent. "
"Help users with graphic design, UI/UX design, and visual content creation "
"by analyzing text descriptions and visual references.",
llm="xai/grok-4"
)
response = agent.start(
"I need to create a logo for my tech startup. "
"Can you help me design a modern and professional logo concept?"
)
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a crisis management and communication AI agent. "
"Help users develop crisis communication strategies, "
"manage reputation during difficult situations, and create effective messaging. "
"Provide guidance on crisis response planning, stakeholder communication, "
"media relations during crises, and reputation recovery strategies.",
llm="groq/llama3.1-8b-instant"
)
response = agent.start("Hello! I'm your crisis management and communication assistant. "
"How can I help you develop effective crisis communication "
"strategies and manage difficult situations today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a digital marketing and SEO optimization AI agent. "
"Help users develop digital marketing strategies, optimize for search engines, "
"and improve online visibility. Provide guidance on SEO best practices, "
"content marketing, social media strategies, PPC campaigns, "
"and analytics interpretation for better ROI.",
llm="groq/llama3.1-8b-instant"
)
response = agent.start("Hello! I'm your digital marketing and SEO optimization assistant. "
"How can I help you improve your online presence, "
"optimize for search engines, or develop marketing strategies today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are an environmental impact assessment AI agent. "
"Help users evaluate environmental impacts of projects, "
"develop sustainability strategies, and ensure compliance with environmental regulations. "
"Provide guidance on environmental risk assessment, sustainability reporting, "
"carbon footprint analysis, and green technology implementation.",
llm="groq/llama3.1-8b-instant"
)
response = agent.start("Hello! I'm your environmental impact assessment assistant. "
"How can I help you evaluate environmental impacts, "
"develop sustainability strategies, or ensure compliance today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are an event planning and management AI agent. "
"Help users plan, organize, and manage successful events of all types. "
"Provide guidance on venue selection, budgeting, vendor coordination, "
"timeline management, marketing strategies, and post-event analysis. "
"Assist with both virtual and in-person event planning.",
llm="groq/llama3.1-8b-instant"
)
response = agent.start("Hello! I'm your event planning and management assistant. "
"How can I help you plan and organize your next successful "
"event today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a healthcare and medical research AI agent. "
"Help users with medical research, healthcare insights, "
"and understanding complex medical information. Provide guidance on medical literature review, "
"healthcare trends analysis, patient care strategies, "
"and medical technology advancements for better healthcare outcomes.",
llm="groq/llama3.1-8b-instant"
)
response = agent.start("Hello! I'm your healthcare and medical research assistant. "
"How can I help you with medical research, healthcare insights, "
"or understanding complex medical information today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a language learning and translation AI agent. "
"Help users learn new languages, translate content accurately, "
"and understand cultural nuances. Provide guidance on language learning strategies, "
"grammar explanations, vocabulary building, pronunciation, "
"and cultural context for effective communication.",
llm="groq/llama3.1-8b-instant"
)
response = agent.start("Hello! I'm your language learning and translation assistant. "
"How can I help you learn new languages, translate content, "
"or understand cultural nuances today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a patent research and innovation AI agent. "
"Help users conduct patent research, analyze intellectual property, "
"identify innovation opportunities, and navigate patent filing processes. "
"Provide guidance on patent searches, prior art analysis, "
"innovation strategy, IP protection, and technology landscape assessment.",
llm="groq/llama3.1-8b-instant"
)
response = agent.start("Hello! I'm your patent research and innovation assistant. "
"How can I help you with patent research, innovation strategy, "
"and intellectual property protection today?")
@@ -0,0 +1,19 @@
"""
Personal Branding & PR Agent using Groq
Specialized agent for managing personal brand and public relations
"""
from praisonaiagents import Agent
agent = Agent(
instructions="You are a personal branding and PR AI agent. "
"Help users build their personal brand, manage public relations, "
"create compelling content strategies, and enhance their online presence. "
"Provide guidance on social media management, content creation, "
"media relations, reputation management, and brand positioning.",
llm="groq/llama3.1-8b-instant"
)
response = agent.start("Hello! I'm your personal branding and PR assistant. "
"How can I help you build your personal brand and "
"manage your public relations today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a product development and innovation AI agent. "
"Help users develop new products, improve existing ones, "
"and drive innovation in their business. Provide guidance on product strategy, "
"market research, user experience design, prototyping, "
"and innovation management for successful product launches.",
llm="groq/llama3.1-8b-instant"
)
response = agent.start("Hello! I'm your product development and innovation assistant. "
"How can I help you develop new products, improve existing ones, "
"or drive innovation in your business today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a supply chain optimization AI agent. "
"Help users optimize their supply chain operations, "
"improve logistics efficiency, reduce costs, and enhance inventory management. "
"Provide guidance on demand forecasting, supplier management, "
"warehouse optimization, transportation planning, and risk mitigation.",
llm="groq/llama3.1-8b-instant"
)
response = agent.start("Hello! I'm your supply chain optimization assistant. "
"How can I help you optimize your supply chain operations "
"and improve your logistics efficiency today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are an aerospace engineering AI agent. "
"Help users understand aerospace engineering concepts, "
"aircraft design, and space technology. Provide guidance on "
"aerodynamics, propulsion systems, materials science, "
"and satellite technology.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your aerospace engineering assistant. "
"How can I help you with aerospace engineering "
"and space technology today?")
@@ -0,0 +1,11 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are an audio tour creation AI agent. "
"Help users create engaging audio tours, guided experiences, and interactive storytelling content. "
"Provide guidance on script writing, audio production, location-based content, and immersive experiences.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your audio tour creation assistant. "
"How can I help you create captivating audio experiences today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a bioinformatics and genomics AI agent. "
"Help users understand bioinformatics, genomics, "
"and computational biology. Provide guidance on "
"DNA sequence analysis, protein structure prediction, "
"genetic algorithms, and biological data mining.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your bioinformatics and genomics assistant. "
"How can I help you with bioinformatics "
"and genomics research today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a blockchain and DeFi AI agent. "
"Help users understand blockchain technology, "
"decentralized finance, and cryptocurrency. Provide guidance on "
"smart contract development, DeFi protocols, "
"tokenomics, and blockchain security.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your blockchain and DeFi assistant. "
"How can I help you explore blockchain technology "
"and decentralized finance today?")
@@ -0,0 +1,11 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a chess AI agent. "
"Help users improve their chess skills, analyze positions, develop strategies, and understand chess theory. "
"Provide move analysis, opening guidance, endgame techniques, and tactical training.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your chess assistant. "
"How can I help you improve your chess game today?")
@@ -0,0 +1,11 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a creative writing AI agent. "
"Help users develop compelling stories, characters, dialogue, and narrative structures. "
"Provide guidance on plot development, world-building, genre-specific techniques, and creative inspiration.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your creative writing assistant. "
"How can I help you bring your stories and ideas to life today?")
@@ -0,0 +1,11 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a cybersecurity AI agent. "
"Help users with security assessments, threat analysis, vulnerability management, and security best practices. "
"Provide guidance on penetration testing, secure coding, incident response, and compliance frameworks.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your cybersecurity assistant. "
"How can I help you secure your systems and protect against threats today?")
@@ -0,0 +1,11 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a data visualization AI agent. "
"Help users create compelling charts, graphs, dashboards, and interactive visualizations. "
"Provide guidance on matplotlib, seaborn, plotly, Tableau, and best practices for effective data storytelling.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your data visualization assistant. "
"How can I help you create stunning visual representations of your data today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are an energy and renewables AI agent. "
"Help users understand renewable energy technologies, "
"energy systems, and sustainability. Provide guidance on "
"solar power, wind energy, energy storage, "
"and green technology solutions.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your energy and renewables assistant. "
"How can I help you with renewable energy "
"and sustainability today?")
@@ -0,0 +1,11 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a 3D PyGame development AI agent. "
"Help users create 3D games, simulations, and interactive experiences using PyGame. "
"Provide code examples, debugging assistance, and best practices for 3D graphics programming.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your 3D PyGame development assistant. "
"How can I help you create amazing 3D games today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are an IoT and smart cities AI agent. "
"Help users understand Internet of Things, smart city "
"technologies, and connected systems. Provide guidance on "
"IoT device integration, smart infrastructure, "
"sensor networks, and urban technology solutions.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your IoT and smart cities assistant. "
"How can I help you with IoT and smart city "
"technologies today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a metaverse and virtual reality AI agent. "
"Help users understand metaverse concepts, VR development, "
"and immersive technologies. Provide guidance on "
"3D modeling, VR programming, digital twins, "
"and virtual world design.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your metaverse and VR assistant. "
"How can I help you explore the metaverse "
"and virtual reality today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a nanotechnology and materials AI agent. "
"Help users understand nanotechnology, advanced materials, "
"and material science. Provide guidance on "
"nanomaterials, material properties, fabrication techniques, "
"and applications in various industries.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your nanotechnology and materials assistant. "
"How can I help you with nanotechnology "
"and materials science today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a quantum AI agent. "
"Help users understand quantum artificial intelligence, "
"quantum machine learning, and quantum neural networks. "
"Provide guidance on quantum algorithms for AI, "
"quantum optimization, and hybrid quantum-classical systems.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your quantum AI assistant. "
"How can I help you explore quantum artificial "
"intelligence today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a quantum computing AI agent. "
"Help users understand quantum computing concepts, "
"quantum algorithms, and quantum programming. Provide guidance on "
"quantum circuit design, quantum error correction, "
"quantum machine learning, and quantum cryptography.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your quantum computing assistant. "
"How can I help you explore quantum computing "
"and quantum programming today?")
@@ -0,0 +1,14 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a robotics and automation AI agent. "
"Help users understand robotics, automation systems, "
"and industrial applications. Provide guidance on "
"robot programming, automation workflows, "
"sensor integration, and robotic process automation.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your robotics and automation assistant. "
"How can I help you with robotics and "
"automation systems today?")
@@ -0,0 +1,11 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a self-evolving AI agent. "
"Continuously learn from interactions, adapt your responses, and improve your capabilities. "
"Help users with complex problem-solving, creative tasks, and provide insights that evolve based on context and user needs.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your self-evolving AI assistant. "
"How can I help you today, and how can I learn and improve from our interaction?")
@@ -0,0 +1,11 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a startup insight AI agent. "
"Help entrepreneurs and startup teams with market analysis, business strategy, funding advice, product development, and growth tactics. "
"Provide actionable insights for building successful startups.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your startup insight assistant. "
"How can I help you build and grow your startup today?")
@@ -0,0 +1,11 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a system architecture AI agent. "
"Help users design scalable, efficient, and robust system architectures. "
"Provide guidance on microservices, cloud infrastructure, database design, API design, and system integration patterns.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your system architecture assistant. "
"How can I help you design and optimize your system architecture today?")
@@ -0,0 +1,11 @@
from praisonaiagents import Agent
agent = Agent(
instructions="You are a web scraping AI agent. "
"Help users create efficient and ethical web scrapers, data extraction tools, and automation scripts. "
"Provide guidance on BeautifulSoup, Selenium, Scrapy, and best practices for responsible web scraping.",
llm="openrouter/moonshotai/kimi-k2"
)
response = agent.start("Hello! I'm your web scraping assistant. "
"How can I help you extract and analyze web data today?")
@@ -0,0 +1,282 @@
"""
Advanced Workflow Patterns Example
This example demonstrates sophisticated workflow orchestration patterns including
conditional routing, decision trees, loops, and complex multi-agent coordination.
Features demonstrated:
- Advanced workflow patterns with conditional routing
- Decision trees and dynamic task routing
- Loop-based processing with exit conditions
- Complex agent coordination and handoffs
- State-based workflow management
"""
from praisonaiagents import Agent, Task, PraisonAIAgents
from praisonaiagents.tools import duckduckgo
import random
# Create specialized agents for complex workflow
intake_agent = Agent(
name="IntakeAgent",
role="Request Intake Specialist",
goal="Classify and route incoming requests to appropriate specialists",
backstory="You are an intake specialist who categorizes requests and determines the appropriate workflow path.",
instructions="Analyze incoming requests and categorize them by type (research, analysis, technical, creative) and complexity (low, medium, high)."
)
research_agent = Agent(
name="ResearchAgent",
role="Research Specialist",
goal="Conduct comprehensive research on assigned topics",
backstory="You are a research specialist who gathers and analyzes information from various sources.",
tools=[duckduckgo],
instructions="Conduct thorough research using available tools. Provide detailed findings with sources."
)
analysis_agent = Agent(
name="AnalysisAgent",
role="Data Analysis Expert",
goal="Analyze data and provide insights",
backstory="You are a data analysis expert who identifies patterns, trends, and actionable insights.",
instructions="Analyze provided data to identify key insights, trends, and recommendations."
)
technical_agent = Agent(
name="TechnicalAgent",
role="Technical Solutions Specialist",
goal="Provide technical solutions and implementations",
backstory="You are a technical specialist who designs and implements technical solutions.",
instructions="Provide detailed technical solutions with implementation guidance and best practices."
)
creative_agent = Agent(
name="CreativeAgent",
role="Creative Content Specialist",
goal="Create engaging and creative content",
backstory="You are a creative specialist who develops engaging content and innovative solutions.",
instructions="Create original, engaging content that meets the specified requirements."
)
review_agent = Agent(
name="ReviewAgent",
role="Quality Review Specialist",
goal="Review work quality and provide feedback",
backstory="You are a quality reviewer who ensures all work meets high standards.",
instructions="Review completed work for quality, accuracy, and completeness. Provide specific feedback."
)
coordinator_agent = Agent(
name="CoordinatorAgent",
role="Workflow Coordinator",
goal="Coordinate complex workflows and manage handoffs",
backstory="You are a workflow coordinator who manages complex multi-agent processes.",
instructions="Coordinate workflow execution, manage handoffs between agents, and ensure successful completion."
)
# Define workflow state management
workflow_state = {
"request_count": 0,
"completed_tasks": [],
"active_workflows": [],
"quality_threshold": 0.8
}
# Simulate incoming requests of different types
sample_requests = [
{
"id": "REQ001",
"type": "market_research",
"description": "Research emerging trends in sustainable technology",
"complexity": "medium",
"priority": "high"
},
{
"id": "REQ002",
"type": "technical_solution",
"description": "Design architecture for distributed AI system",
"complexity": "high",
"priority": "medium"
},
{
"id": "REQ003",
"type": "creative_content",
"description": "Create marketing campaign for new product launch",
"complexity": "medium",
"priority": "high"
},
{
"id": "REQ004",
"type": "data_analysis",
"description": "Analyze customer behavior patterns from sales data",
"complexity": "low",
"priority": "low"
}
]
# Process each request through the advanced workflow
for request in sample_requests:
print(f"\n{'='*80}")
print(f"PROCESSING REQUEST {request['id']}: {request['type'].upper()}")
print(f"Description: {request['description']}")
print(f"Complexity: {request['complexity']} | Priority: {request['priority']}")
print(f"{'='*80}")
workflow_state["request_count"] += 1
workflow_state["active_workflows"].append(request["id"])
# Step 1: Intake and Classification
intake_task = Task(
name=f"intake_{request['id']}",
description=f"Classify and route this request: {request['description']} (Type: {request['type']}, Complexity: {request['complexity']})",
expected_output="Request classification with recommended workflow path and agent assignment",
agent=intake_agent
)
# Step 2: Route to appropriate specialist based on type
if request['type'] == 'market_research':
specialist_agent = research_agent
specialist_task_desc = f"Conduct comprehensive market research on: {request['description']}"
elif request['type'] == 'technical_solution':
specialist_agent = technical_agent
specialist_task_desc = f"Design technical solution for: {request['description']}"
elif request['type'] == 'creative_content':
specialist_agent = creative_agent
specialist_task_desc = f"Create creative content for: {request['description']}"
elif request['type'] == 'data_analysis':
specialist_agent = analysis_agent
specialist_task_desc = f"Analyze data for: {request['description']}"
else:
specialist_agent = research_agent # Default fallback
specialist_task_desc = f"Process request: {request['description']}"
specialist_task = Task(
name=f"specialist_{request['id']}",
description=specialist_task_desc,
expected_output="Comprehensive response addressing the request requirements",
agent=specialist_agent,
context=[intake_task]
)
# Step 3: Quality Review with conditional routing
review_task = Task(
name=f"review_{request['id']}",
description=f"Review the quality of work completed for request {request['id']}. Assess if it meets standards or needs revision.",
expected_output="Quality assessment with pass/fail decision and specific feedback",
agent=review_agent,
context=[intake_task, specialist_task],
# Conditional routing based on quality
task_type="decision",
condition={
"approved": [f"finalization_{request['id']}"], # If approved, finalize
"needs_revision": [f"revision_{request['id']}"], # If needs work, revise
"needs_escalation": [f"escalation_{request['id']}"] # If complex, escalate
}
)
# Step 4a: Revision Task (conditional)
revision_task = Task(
name=f"revision_{request['id']}",
description=f"Revise the work based on quality review feedback for request {request['id']}",
expected_output="Revised work addressing quality review feedback",
agent=specialist_agent,
context=[specialist_task, review_task]
)
# Step 4b: Escalation Task (conditional)
escalation_task = Task(
name=f"escalation_{request['id']}",
description=f"Handle escalated request {request['id']} requiring additional expertise or coordination",
expected_output="Escalated response with enhanced solution approach",
agent=coordinator_agent,
context=[intake_task, specialist_task, review_task]
)
# Step 5: Finalization and Coordination
finalization_task = Task(
name=f"finalization_{request['id']}",
description=f"Finalize and package the completed work for request {request['id']}",
expected_output="Final deliverable package with summary and next steps",
agent=coordinator_agent,
context=[intake_task, specialist_task, review_task]
)
# Execute the workflow with conditional routing
workflow_agents = PraisonAIAgents(
agents=[
intake_agent, specialist_agent, review_agent,
coordinator_agent
],
tasks=[
intake_task, specialist_task, review_task,
revision_task, # Conditional
escalation_task, # Conditional
finalization_task
],
process="workflow", # Use workflow process for conditional execution
verbose=True
)
print(f"Executing advanced workflow for {request['id']}...")
result = workflow_agents.start()
# Update workflow state
workflow_state["completed_tasks"].append(request["id"])
workflow_state["active_workflows"].remove(request["id"])
# Simulate quality scoring
quality_score = random.uniform(0.75, 0.95)
print(f"✅ Request {request['id']} completed with quality score: {quality_score:.2f}")
print(f"📋 Result preview: {str(result)[:150]}...")
# Loop control - if we have processed enough requests, demonstrate loop exit
if workflow_state["request_count"] >= 2:
print(f"\n🔄 WORKFLOW LOOP CONTROL: Processed {workflow_state['request_count']} requests")
print("Checking if we should continue or exit based on workflow state...")
# Example loop control logic
if workflow_state["request_count"] >= 4:
print("📊 Batch processing complete. Exiting workflow loop.")
break
else:
print("📈 Continuing with next request in the batch...")
# Advanced Workflow Analytics and Summary
print(f"\n{'='*80}")
print("ADVANCED WORKFLOW PATTERNS SUMMARY")
print(f"{'='*80}")
print(f"📊 Workflow Statistics:")
print(f"- Total Requests Processed: {workflow_state['request_count']}")
print(f"- Completed Tasks: {len(workflow_state['completed_tasks'])}")
print(f"- Active Workflows: {len(workflow_state['active_workflows'])}")
print(f"\n🔀 Workflow Patterns Demonstrated:")
print("- Dynamic request classification and routing")
print("- Conditional task execution based on quality review")
print("- Multi-path workflows (approval/revision/escalation)")
print("- State-based workflow management")
print("- Loop control with exit conditions")
print("- Agent specialization and handoffs")
print(f"\n⚙️ Advanced Features Used:")
print("- Workflow process type with conditional routing")
print("- Decision tasks with multiple outcome paths")
print("- Context passing between related tasks")
print("- State management across workflow iterations")
print("- Quality-based routing decisions")
print("- Scalable agent coordination patterns")
print(f"\n🎯 Production Benefits:")
print("- Handles complex, multi-step business processes")
print("- Automatically routes work to appropriate specialists")
print("- Implements quality gates and revision loops")
print("- Scales to handle multiple concurrent workflows")
print("- Provides visibility into workflow state and progress")
print("- Enables sophisticated business process automation")
print(f"\n{'='*80}")
print("ADVANCED WORKFLOW DEMONSTRATION COMPLETED")
print(f"{'='*80}")
@@ -0,0 +1,99 @@
"""
Hierarchical Process Example
This example demonstrates how to create a hierarchical process where a manager agent
delegates tasks to specialized worker agents. The manager orchestrates the workflow
and each specialized agent performs their specific tasks.
Features demonstrated:
- Hierarchical agent structure with manager and workers
- Task delegation and coordination
- Specialized agent roles with different capabilities
- Process-level orchestration
"""
from praisonaiagents import Agent, Task, PraisonAIAgents
from praisonaiagents.tools import duckduckgo
# Create Manager Agent
manager = Agent(
name="Manager",
role="Project Manager",
goal="Coordinate research and analysis tasks among specialized agents",
backstory="You are an experienced project manager who delegates tasks to specialized team members and synthesizes their work into a comprehensive report.",
instructions="You coordinate tasks between researchers and analysts. Delegate research to the researcher, data analysis to the analyst, and writing to the writer."
)
# Create Specialized Worker Agents
researcher = Agent(
name="Researcher",
role="Senior Researcher",
goal="Conduct thorough research on given topics",
backstory="You are a senior researcher with expertise in gathering and analyzing information from various sources.",
tools=[duckduckgo],
instructions="Conduct comprehensive research using web search. Provide detailed, factual information with sources."
)
data_analyst = Agent(
name="DataAnalyst",
role="Data Analyst",
goal="Analyze trends and patterns in research data",
backstory="You are a skilled data analyst who can identify trends, patterns, and insights from research data.",
instructions="Analyze research data to identify key trends, patterns, and actionable insights. Provide data-driven conclusions."
)
writer = Agent(
name="Writer",
role="Technical Writer",
goal="Create well-structured reports and documentation",
backstory="You are a technical writer who excels at creating clear, comprehensive reports from research and analysis.",
instructions="Create well-structured, professional reports that synthesize research findings and analysis into actionable insights."
)
# Define Hierarchical Tasks
research_task = Task(
name="research_task",
description="Research the latest trends in artificial intelligence and machine learning for business applications",
expected_output="Comprehensive research report with current AI/ML trends, key players, and business applications",
agent=researcher
)
analysis_task = Task(
name="analysis_task",
description="Analyze the research findings to identify the most promising AI/ML trends for business adoption",
expected_output="Analysis report highlighting top 5 AI/ML trends with business impact assessment",
agent=data_analyst,
context=[research_task] # Depends on research task
)
writing_task = Task(
name="writing_task",
description="Create a final executive summary combining research and analysis into actionable business recommendations",
expected_output="Executive summary with strategic recommendations for AI/ML adoption",
agent=writer,
context=[research_task, analysis_task] # Depends on both previous tasks
)
coordination_task = Task(
name="coordination_task",
description="Oversee the entire research project, ensure quality and coordination between all team members",
expected_output="Project oversight report confirming successful completion and quality of all deliverables",
agent=manager,
context=[research_task, analysis_task, writing_task] # Oversees all tasks
)
# Create Hierarchical Process
agents = PraisonAIAgents(
agents=[manager, researcher, data_analyst, writer],
tasks=[research_task, analysis_task, writing_task, coordination_task],
process="hierarchical", # Hierarchical process type
verbose=True
)
# Execute Hierarchical Process
result = agents.start()
print("\n" + "="*50)
print("HIERARCHICAL PROCESS COMPLETED")
print("="*50)
print(f"Final Result:\n{result}")
@@ -0,0 +1,31 @@
"""
Basic example of using Anthropic Claude with PraisonAI
"""
from praisonaiagents import Agent
# Initialize Agent with Anthropic Claude
agent = Agent(
instructions="You are a helpful assistant",
llm="anthropic/claude-3-5-sonnet-20241022",
)
# Example conversation
response = agent.start("Hello! Can you help me with a writing task?")
# Example with creative writing
writing_task = """
Write a short story about a time traveler who discovers
they can only travel to moments of great historical significance.
Make it engaging and about 200 words.
"""
response = agent.start(writing_task)
# Example with reasoning
reasoning_task = """
Explain the concept of quantum entanglement in simple terms,
and then discuss its potential applications in quantum computing.
"""
response = agent.start(reasoning_task)
Loaded 100 of 209 files, more files were not shown because too many files have changed in this diff. Show more