Feature/typedb integration

[ascender1729]

Add TypeDB Integration

Description

This PR adds TypeDB integration to the Swarms project, providing a robust graph database wrapper for knowledge representation and relationship modeling. The implementation includes:

1. TypeDBWrapper Class:

   • Connection management with configurable settings
   • Schema definition and management
   • CRUD operations (Create, Read, Update, Delete)
   • Transaction handling
   • Error handling and logging
   • Context manager support

2. Features:

   • Type-safe schema definitions
   • Efficient query execution
   • Automatic connection management
   • Comprehensive error handling
   • Detailed logging
   • Example usage

3. Example Implementation:

   • Knowledge graph schema
   • Data insertion and querying
   • Relationship modeling
   • Data deletion and verification

Issue

Fixes #794 (Integration with the TypeDB graph database)

Dependencies

Added to requirements.txt:

• typedb-client>=2.25.0
• typedb-protocol>=2.25.0
• typedb-driver>=2.25.0

Testing

Added comprehensive test suite in tests/utils/test_typedb_wrapper.py:

• Connection management tests
• Schema definition tests
• CRUD operation tests
• Error handling tests
• Context manager tests

All tests can be run using:

pytest tests/utils/test_typedb_wrapper.py -v

Tag Maintainer

@kyegomez (General / Misc)

Twitter Handle

@the_complex_one

Example Usage

from swarms.utils.typedb_wrapper import TypeDBWrapper, TypeDBConfig

# Initialize wrapper
config = TypeDBConfig(
    uri="localhost:1729",
    database="swarms_example"
)

# Use context manager for automatic cleanup
with TypeDBWrapper(config) as db:
    # Define schema
    schema = """
    define
    person sub entity,
        owns name: string,
        owns age: long;
    """
    db.define_schema(schema)
    
    # Insert data
    query = """
    insert
    $p isa person, has name "John Doe", has age 30;
    """
    db.insert_data(query)
    
    # Query data
    results = db.query_data("match $p isa person; get;")
    print(results)

Benefits

1. Enhanced knowledge representation
2. Efficient relationship modeling
3. Type-safe schema definitions
4. Robust error handling
5. Easy integration with existing code
6. Comprehensive documentation and examples

---

📚 Documentation preview 📚: https://swarms--803.org.readthedocs.build/en/803/

[greptile-apps]

PR Summary

Your free trial has ended. If you'd like to continue receiving code reviews, you can add a payment method here: https://app.greptile.com/review/github.

_{💡 (4/5) You can add custom instructions or style guidelines for the bot here!}

_{12 file(s) reviewed, no comment(s)
Edit PR Review Bot Settings | Greptile}