Skip to content

Conversation

@TropicalDog17
Copy link

Description

This PR implements Google Agent Development Kit (ADK) integration for the Flare AI Kit, exposing all SDK functionality as ADK tools. The implementation enables natural language interaction with Flare blockchain capabilities through Gemini models.

Closes #98

Changes

  • Core ADK Infrastructure

    • Created @tool decorator for registering Python functions as ADK tools
    • Implemented automatic schema generation from type hints
    • Set up global tool registry
  • Tool Implementations

    • Ecosystem tools: FTSO prices, blockchain interaction
    • TEE tools: Attestation token management
    • Wallet tools: Turnkey integration, transaction management
    • Social tools: Telegram and X integration
  • Agent Enhancement

    • Added comprehensive system instruction
    • Configured agent with Gemini model
    • Implemented automatic tool registration
  • Testing

    • Unit tests for tool decorator and individual tools
    • Integration tests for ADK functionality
    • Mock prompt response tests
    • Error handling test cases

Dependencies Added

  • google-adk>=1.9.0 (core ADK)
  • dulwich>=0.24.1 (RAG support)
  • qdrant-client>=1.15.1 (RAG support)
  • pyjwt>=2.10.1 (TEE functionality)
  • pyopenssl>=25.1.0 (TEE functionality)

Testing Done

  • ✅ Tool decorator tests pass
  • ✅ Individual tool unit tests pass
  • ✅ Integration tests pass
  • ✅ Mock prompt tests pass
  • ✅ Error handling tests pass

Documentation

  • Added implementation details in code docstrings
  • Added type hints for schema generation
  • Added example usage in tests

Breaking Changes

None. This is an additive feature that enhances existing functionality.

Checklist

  • Code follows project style guidelines
  • Tests added and passing
  • Documentation updated
  • Dependencies updated
  • No breaking changes

…tionality

- Add ecosystem tools for FTSO price queries and blockchain interaction
- Add social tools for Telegram and X integration
- Add TEE tools for attestation and secure operations
- Add wallet tools for Turnkey integration and transaction management
- Implement type-safe tool definitions with proper schemas
- Add comprehensive error handling and logging
- Import all tool modules to register tools automatically
- Add detailed system instruction for Flare ecosystem interaction
- Configure agent with proper name and Gemini model
- Set up tool registry with all available tools
- Add unit tests for tool decorator and individual tools
- Add integration tests for full ADK functionality
- Add mock prompt response tests for agent orchestration
- Test error handling and edge cases
- Implement proper test mocking for external dependencies
- Add google-adk>=1.9.0 for core ADK functionality
- Add dulwich and qdrant-client for RAG support
- Add pyjwt and pyopenssl for TEE functionality
- Update dependency lock file
- Add tests for sync and async function decoration
- Test tool registration in registry
- Test schema generation from type hints
- Test error handling cases
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Advanced] Expose all SDK functionality as Google ADK tools

1 participant