intentkit 0.6.8.dev2__tar.gz → 0.6.9__tar.gz

{intentkit-0.6.8.dev2 → intentkit-0.6.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.8.dev2
+Version: 0.6.9
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestal-network/intentkit
 Project-URL: Repository, https://github.com/crestal-network/intentkit
@@ -64,8 +64,8 @@ Requires-Dist: langchain-xai>=0.2.1
 Requires-Dist: langchain<0.4.0,>=0.3.25
 Requires-Dist: langgraph-checkpoint-postgres<2.0.23,>=2.0.16
 Requires-Dist: langgraph-checkpoint>=2.0.18
-Requires-Dist: langgraph-prebuilt>=0.5.1
-Requires-Dist: langgraph>=0.4.3
+Requires-Dist: langgraph-prebuilt<0.6.0,>=0.5.1
+Requires-Dist: langgraph<0.6.0,>=0.5.1
 Requires-Dist: langmem>=0.0.27
 Requires-Dist: mypy-boto3-s3<2.0.0,>=1.37.24
 Requires-Dist: openai>=1.59.6

{intentkit-0.6.8.dev2 → intentkit-0.6.9}/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "0.6.8-dev2"
+__version__ = "0.6.9"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

{intentkit-0.6.8.dev2 → intentkit-0.6.9}/models/user.py

@@ -1,7 +1,7 @@
 import logging
 from datetime import datetime, timezone
 from decimal import Decimal
-from typing import Annotated, Optional
+from typing import Annotated, Optional, Type, TypeVar
 
 from intentkit.models.base import Base
 from intentkit.models.credit import CreditAccount
@@ -14,6 +14,47 @@ from sqlalchemy.ext.asyncio import AsyncSession
 logger = logging.getLogger(__name__)
 
 
+# TypeVar for User model constraint
+UserModelType = TypeVar("UserModelType", bound="User")
+UserTableType = TypeVar("UserTableType", bound="UserTable")
+
+
+class UserRegistry:
+    """Registry for extended model classes."""
+
+    def __init__(self):
+        self._user_table_class: Optional[Type[UserTableType]] = None
+        self._user_model_class: Optional[Type[UserModelType]] = None
+
+    def register_user_table(self, user_table_class: Type[UserTableType]) -> None:
+        """Register extended UserTable class.
+
+        Args:
+            user_table_class: A class that inherits from UserTable
+        """
+        self._user_table_class = user_table_class
+
+    def get_user_table_class(self) -> Type[UserTableType]:
+        """Get registered UserTable class or default."""
+        return self._user_table_class or UserTable
+
+    def register_user_model(self, user_model_class: Type[UserModelType]) -> None:
+        """Register extended UserModel class.
+
+        Args:
+            user_model_class: A class that inherits from User
+        """
+        self._user_model_class = user_model_class
+
+    def get_user_model_class(self) -> Type[UserModelType]:
+        """Get registered UserModel class or default."""
+        return self._user_model_class or User
+
+
+# Global registry instance
+user_model_registry = UserRegistry()
+
+
 class UserTable(Base):
     """User database table model."""
 
@@ -135,13 +176,15 @@ class UserUpdate(BaseModel):
         Returns:
             Updated or newly created User model
         """
+        user_model_class = user_model_registry.get_user_model_class()
+        user_table_class = user_model_registry.get_user_table_class()
         async with get_session() as db:
-            db_user = await db.get(UserTable, id)
+            db_user = await db.get(user_table_class, id)
             old_nft_count = 0  # Default for new users
 
             if not db_user:
                 # Create new user if it doesn't exist
-                db_user = UserTable(id=id)
+                db_user = user_table_class(id=id)
                 db.add(db_user)
             else:
                 old_nft_count = db_user.nft_count
@@ -158,7 +201,7 @@ class UserUpdate(BaseModel):
             await db.commit()
             await db.refresh(db_user)
 
-            return User.model_validate(db_user)
+            return user_model_class.model_validate(db_user)
 
     async def put(self, id: str) -> "User":
         """Replace all fields of a user in the database with the provided values.
@@ -171,13 +214,15 @@ class UserUpdate(BaseModel):
         Returns:
             Updated or newly created User model
         """
+        user_model_class = user_model_registry.get_user_model_class()
+        user_table_class = user_model_registry.get_user_table_class()
         async with get_session() as db:
-            db_user = await db.get(UserTable, id)
+            db_user = await db.get(user_table_class, id)
             old_nft_count = 0  # Default for new users
 
             if not db_user:
                 # Create new user if it doesn't exist
-                db_user = UserTable(id=id)
+                db_user = user_table_class(id=id)
                 db.add(db_user)
             else:
                 old_nft_count = db_user.nft_count
@@ -193,7 +238,7 @@ class UserUpdate(BaseModel):
             await db.commit()
             await db.refresh(db_user)
 
-            return User.model_validate(db_user)
+            return user_model_class.model_validate(db_user)
 
 
 class User(UserUpdate):
@@ -236,7 +281,10 @@ class User(UserUpdate):
         Returns:
             User model or None if not found
         """
-        result = await session.execute(select(UserTable).where(UserTable.id == user_id))
+        user_table_class = user_model_registry.get_user_table_class()
+        result = await session.execute(
+            select(user_table_class).where(user_table_class.id == user_id)
+        )
         user = result.scalars().first()
         if user is None:
             return None

{intentkit-0.6.8.dev2 → intentkit-0.6.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "intentkit"
-version = "0.6.8-dev2"
+version = "0.6.9"
 description = "Intent-based AI Agent Platform - Core Package"
 authors = [{ name = "hyacinthus", email = "hyacinthus@gmail.com" }]
 requires-python = ">=3.12"
@@ -33,10 +33,10 @@ dependencies = [
     "langchain-openai>=0.3.8",
     "langchain-text-splitters>=0.3.8",
     "langchain-xai>=0.2.1",
-    "langgraph>=0.4.3",
+    "langgraph (>=0.5.1,<0.6.0)",
     "langgraph-checkpoint>=2.0.18",
     "langgraph-checkpoint-postgres>=2.0.16,<2.0.23",
-    "langgraph-prebuilt>=0.5.1",
+    "langgraph-prebuilt (>=0.5.1,<0.6.0)",
     "langmem>=0.0.27",
     "mypy-boto3-s3>=1.37.24,<2.0.0",
     "openai>=1.59.6",