intentkit 0.6.8.dev2__py3-none-any.whl → 0.6.9.dev1__py3-none-any.whl

intentkit/__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-dev1"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

intentkit/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.dist-info → intentkit-0.6.9.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.8.dev2
+Version: 0.6.9.dev1
 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.dist-info → intentkit-0.6.9.dev1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-intentkit/__init__.py,sha256=VZCUEI9HK5GtcCKtsmbH7h_BJ3cbr0m9LmfqarooGMM,383
+intentkit/__init__.py,sha256=DvWnR7DSV3CNFs0DauTHc22pPwTwOyKkJ_tKCoVp1YE,383
 intentkit/abstracts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 intentkit/abstracts/agent.py,sha256=108gb5W8Q1Sy4G55F2_ZFv2-_CnY76qrBtpIr0Oxxqk,1489
 intentkit/abstracts/api.py,sha256=ZUc24vaQvQVbbjznx7bV0lbbQxdQPfEV8ZxM2R6wZWo,166
@@ -35,7 +35,7 @@ intentkit/models/generator.py,sha256=lyZu9U9rZUGkqd_QT5SAhay9DY358JJY8EhDSpN8I1M
 intentkit/models/llm.py,sha256=f_F11cjeXACu5qDB1JaCltWqS_uVb8GqOSmPTq1LYo8,25123
 intentkit/models/redis.py,sha256=UoN8jqLREO1VO9_w6m-JhldpP19iEHj4TiGVCMutQW4,3702
 intentkit/models/skill.py,sha256=h_2wtKEbYE29TLsMdaSnjfOv6vXY6GwMU_abw-ONX28,16374
-intentkit/models/user.py,sha256=KJLxIuRqZztzG4_QL9eDWhlwWLOGdA8iMGlX5KA7Fyc,7512
+intentkit/models/user.py,sha256=P7l6LOsZmXZ5tDPTczTbqDtDB_MKc_9_ddZkAB2npPk,9288
 intentkit/skills/__init__.py,sha256=WkjmKB4xvy36zyXMroPMf_DTPgQloNS3L73nVnBmuQI,303
 intentkit/skills/base.py,sha256=nPHX8kZ-r5ffbCGH6Tnh0KwGWlF1D4zMmnjsFf9iScA,6677
 intentkit/skills/skills.toml,sha256=iFMn_UmdFFCk4HTHolpSv6Ve6O-iRckJtxVcZ5EjxQM,2634
@@ -395,7 +395,7 @@ intentkit/utils/random.py,sha256=DymMxu9g0kuQLgJUqalvgksnIeLdS-v0aRk5nQU0mLI,452
 intentkit/utils/s3.py,sha256=9trQNkKQ5VgxWsewVsV8Y0q_pXzGRvsCYP8xauyUYkg,8549
 intentkit/utils/slack_alert.py,sha256=s7UpRgyzLW7Pbmt8cKzTJgMA9bm4EP-1rQ5KXayHu6E,2264
 intentkit/utils/tx.py,sha256=2yLLGuhvfBEY5n_GJ8wmIWLCzn0FsYKv5kRNzw_sLUI,1454
-intentkit-0.6.8.dev2.dist-info/METADATA,sha256=EK0jvD1SSHNx8CNnaqWrEhS14tNW_cri7WTPpDVdR6U,6328
-intentkit-0.6.8.dev2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-intentkit-0.6.8.dev2.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
-intentkit-0.6.8.dev2.dist-info/RECORD,,
+intentkit-0.6.9.dev1.dist-info/METADATA,sha256=U8uY2M0KeyU_AwEKh2FLvPkOi92jRiTvnNwjnZGZAFs,6342
+intentkit-0.6.9.dev1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+intentkit-0.6.9.dev1.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
+intentkit-0.6.9.dev1.dist-info/RECORD,,