intentkit 0.7.1.dev3__py3-none-any.whl → 0.7.2.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.7.1-dev3"
+__version__ = "0.7.2-dev1"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

intentkit/core/credit.py

@@ -22,6 +22,7 @@ from intentkit.models.credit import (
     DEFAULT_PLATFORM_ACCOUNT_REFILL,
     DEFAULT_PLATFORM_ACCOUNT_REWARD,
     DEFAULT_PLATFORM_ACCOUNT_SKILL,
+    DEFAULT_PLATFORM_ACCOUNT_WITHDRAW,
     CreditAccount,
     CreditAccountTable,
     CreditDebit,
@@ -212,6 +213,166 @@ async def recharge(
     return user_account
 
 
+async def withdraw(
+    session: AsyncSession,
+    agent_id: str,
+    amount: Decimal,
+    upstream_tx_id: str,
+    note: Optional[str] = None,
+) -> CreditAccount:
+    """
+    Withdraw credits from an agent account to platform account.
+
+    Args:
+        session: Async session to use for database operations
+        agent_id: ID of the agent to withdraw from
+        amount: Amount of credits to withdraw
+        upstream_tx_id: ID of the upstream transaction
+        note: Optional note for the transaction
+
+    Returns:
+        Updated agent credit account
+    """
+    # Check for idempotency - prevent duplicate transactions
+    await CreditEvent.check_upstream_tx_id_exists(
+        session, UpstreamType.API, upstream_tx_id
+    )
+
+    if amount <= Decimal("0"):
+        raise ValueError("Withdraw amount must be positive")
+
+    # Get agent to retrieve user_id from agent.owner
+    agent = await Agent.get(agent_id)
+    if not agent:
+        raise HTTPException(status_code=404, detail="Agent not found")
+
+    if not agent.owner:
+        raise HTTPException(status_code=400, detail="Agent has no owner")
+
+    # Get agent wallet address
+    agent_data = await AgentData.get(agent.id)
+    agent_wallet_address = agent_data.evm_wallet_address if agent_data else None
+
+    user_id = agent.owner
+
+    # Get agent account to check balance
+    agent_account = await CreditAccount.get_in_session(
+        session=session,
+        owner_type=OwnerType.AGENT,
+        owner_id=agent_id,
+    )
+
+    # Check if agent has sufficient permanent credits
+    if agent_account.credits < amount:
+        raise HTTPException(
+            status_code=400,
+            detail=f"Insufficient balance. Available: {agent_account.credits}, Required: {amount}",
+        )
+
+    # 1. Create credit event record first to get event_id
+    event_id = str(XID())
+
+    # 2. Update agent account - deduct credits
+    updated_agent_account = await CreditAccount.deduction_in_session(
+        session=session,
+        owner_type=OwnerType.AGENT,
+        owner_id=agent_id,
+        credit_type=CreditType.PERMANENT,
+        amount=amount,
+        event_id=event_id,
+    )
+
+    # 3. Update platform withdraw account - add credits
+    platform_account = await CreditAccount.income_in_session(
+        session=session,
+        owner_type=OwnerType.PLATFORM,
+        owner_id=DEFAULT_PLATFORM_ACCOUNT_WITHDRAW,
+        amount_details={
+            CreditType.PERMANENT: amount
+        },  # Withdraw adds to platform permanent credits
+        event_id=event_id,
+    )
+
+    # 4. Create credit event record
+    event = CreditEventTable(
+        id=event_id,
+        event_type=EventType.WITHDRAW,
+        user_id=user_id,
+        upstream_type=UpstreamType.API,
+        upstream_tx_id=upstream_tx_id,
+        direction=Direction.EXPENSE,
+        account_id=updated_agent_account.id,
+        total_amount=amount,
+        credit_type=CreditType.PERMANENT,
+        credit_types=[CreditType.PERMANENT],
+        balance_after=updated_agent_account.credits
+        + updated_agent_account.free_credits
+        + updated_agent_account.reward_credits,
+        base_amount=amount,
+        base_original_amount=amount,
+        base_free_amount=Decimal("0"),  # No free credits involved in base amount
+        base_reward_amount=Decimal("0"),  # No reward credits involved in base amount
+        base_permanent_amount=amount,  # All base amount is permanent for withdraw
+        permanent_amount=amount,  # Set permanent_amount since this is a permanent credit
+        free_amount=Decimal("0"),  # No free credits involved
+        reward_amount=Decimal("0"),  # No reward credits involved
+        agent_wallet_address=agent_wallet_address,  # Include agent wallet address
+        note=note,
+    )
+    session.add(event)
+    await session.flush()
+
+    # 5. Create credit transaction records
+    # 5.1 Agent account transaction (debit)
+    agent_tx = CreditTransactionTable(
+        id=str(XID()),
+        account_id=updated_agent_account.id,
+        event_id=event_id,
+        tx_type=TransactionType.WITHDRAW,
+        credit_debit=CreditDebit.DEBIT,
+        change_amount=amount,
+        credit_type=CreditType.PERMANENT,
+        free_amount=Decimal("0"),
+        reward_amount=Decimal("0"),
+        permanent_amount=amount,
+    )
+    session.add(agent_tx)
+
+    # 5.2 Platform withdraw account transaction (credit)
+    platform_tx = CreditTransactionTable(
+        id=str(XID()),
+        account_id=platform_account.id,
+        event_id=event_id,
+        tx_type=TransactionType.WITHDRAW,
+        credit_debit=CreditDebit.CREDIT,
+        change_amount=amount,
+        credit_type=CreditType.PERMANENT,
+        free_amount=Decimal("0"),
+        reward_amount=Decimal("0"),
+        permanent_amount=amount,
+    )
+    session.add(platform_tx)
+
+    # Commit all changes
+    await session.commit()
+
+    # Send Slack notification for withdraw
+    try:
+        send_slack_message(
+            f"💸 **Credit Withdraw**\n"
+            f"• Agent ID: `{agent_id}`\n"
+            f"• User ID: `{user_id}`\n"
+            f"• Amount: `{amount}` credits\n"
+            f"• Transaction ID: `{upstream_tx_id}`\n"
+            f"• New Balance: `{updated_agent_account.credits}` credits\n"
+            f"• Note: {note or 'N/A'}"
+        )
+    except Exception as e:
+        logger.error(f"Failed to send Slack notification for withdraw: {str(e)}")
+
+    return updated_agent_account
+
+
 async def reward(
     session: AsyncSession,
     user_id: str,

intentkit/models/credit.py

@@ -60,6 +60,7 @@ DEFAULT_PLATFORM_ACCOUNT_VOICE = "platform_voice"
 DEFAULT_PLATFORM_ACCOUNT_KNOWLEDGE = "platform_knowledge"
 DEFAULT_PLATFORM_ACCOUNT_FEE = "platform_fee"
 DEFAULT_PLATFORM_ACCOUNT_DEV = "platform_dev"
+DEFAULT_PLATFORM_ACCOUNT_WITHDRAW = "platform_withdraw"
 
 
 class CreditAccountTable(Base):
@@ -735,7 +736,7 @@ class CreditAccount(BaseModel):
                 id=str(XID()),
                 account_id=account.id,
                 event_id=event_id,
-                tx_type=TransactionType.RECHARGE,
+                tx_type=TransactionType.REFILL,
                 credit_debit=CreditDebit.CREDIT,
                 change_amount=free_quota,
                 credit_type=CreditType.FREE,
@@ -865,6 +866,7 @@ class EventType(str, Enum):
     REFUND = "refund"
     ADJUSTMENT = "adjustment"
     REFILL = "refill"
+    WITHDRAW = "withdraw"
     # Sync with RewardType values
     REWARD = "reward"
     EVENT_REWARD = "event_reward"
@@ -1378,6 +1380,7 @@ class TransactionType(str, Enum):
     REFUND = "refund"
     ADJUSTMENT = "adjustment"
     REFILL = "refill"
+    WITHDRAW = "withdraw"
     # Sync with RewardType values
     REWARD = "reward"
     EVENT_REWARD = "event_reward"

{intentkit-0.7.1.dev3.dist-info → intentkit-0.7.2.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.7.1.dev3
+Version: 0.7.2.dev1
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestalnetwork/intentkit
 Project-URL: Repository, https://github.com/crestalnetwork/intentkit

{intentkit-0.7.1.dev3.dist-info → intentkit-0.7.2.dev1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-intentkit/__init__.py,sha256=Vy_-mTwSa2SkcSN21QaZzVvTHhR_hPQEer29UVDb2fw,383
+intentkit/__init__.py,sha256=6Irgq-hTcw_LXvAzBoQzdKrpqnaXVn1ju6AtdJ_Xtpk,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
@@ -17,7 +17,7 @@ intentkit/core/agent.py,sha256=GIKDn1dTenIHWMRxe-ud7hd1cQaHzbTDdypy5IAgPfU,16658
 intentkit/core/api.py,sha256=WfoaHNquujYJIpNPuTR1dSaaxog0S3X2W4lG9Ehmkm4,3284
 intentkit/core/chat.py,sha256=6wdwaWfO5Tika6dXUhgeMVFUMP8Zpn3NeeAN4QF9_PA,1682
 intentkit/core/client.py,sha256=J5K7f08-ucszBKAbn9K3QNOFKIC__7amTbKYii1jFkI,3056
-intentkit/core/credit.py,sha256=Jbl7swi9d-4HyoxgC1iLeOIzfjc__x0-lJfIYWCIgK4,69817
+intentkit/core/credit.py,sha256=vW3I5c_fKe_bmFvOc9_EaMsMuAbqw-XQ_NbJg_okdA8,75284
 intentkit/core/engine.py,sha256=H4ew1jhn2coMYJ3zR9isM1Y-XbnXNMg91SeDoeXdQ4U,36562
 intentkit/core/node.py,sha256=7h9zgDSd928bzUi3m3EZnKkhbwqlbRAQUr_uz7gKB5Y,8880
 intentkit/core/prompt.py,sha256=clZmH9Ryn7cUwi6lQADEH9-AupT4V4NLMaUCzOoFHS4,15899
@@ -29,7 +29,7 @@ intentkit/models/app_setting.py,sha256=iYbW63QD91bt4oEYV3wOXHuRFav2b4VXLwb_StgUQ
 intentkit/models/base.py,sha256=o-zRjVrak-f5Jokdvj8BjLm8gcC3yYiYMCTLegwT2lA,185
 intentkit/models/chat.py,sha256=cDccEHU8nd7Y5uhrHDCuZGwqrRwhqCaeztMiZcemiug,20469
 intentkit/models/conversation.py,sha256=nrbDIw-3GK5BYi_xkI15FLdx4a6SNrFK8wfAGLCsrqk,9032
-intentkit/models/credit.py,sha256=QQzc8oSbCTUfgRT-aLsMYxk6eTTktBCGM-nzD3glsFU,52169
+intentkit/models/credit.py,sha256=JQ_ITxOM98XTkllxGDduKGf9-ZF4R-rYixN3OgNvl-Y,52275
 intentkit/models/db.py,sha256=nZjp6HlfLAWbMJeUkWrUcvlS4DAS8C_kpl5PzlO7eTc,5060
 intentkit/models/db_mig.py,sha256=vT6Tanm-BHC2T7dTztuB1UG494EFBAlHADKsNzR6xaQ,3577
 intentkit/models/generator.py,sha256=lyZu9U9rZUGkqd_QT5SAhay9DY358JJY8EhDSpN8I1M,10298
@@ -418,7 +418,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.7.1.dev3.dist-info/METADATA,sha256=-6o8ouKvstC4DYOIKTEqkzC_YECsMygcIfhd-E_HR_g,6409
-intentkit-0.7.1.dev3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-intentkit-0.7.1.dev3.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
-intentkit-0.7.1.dev3.dist-info/RECORD,,
+intentkit-0.7.2.dev1.dist-info/METADATA,sha256=NFtTG-asUUpnQp9Sd4v_22mjhekVTQcL9jNHZQORiHI,6409
+intentkit-0.7.2.dev1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+intentkit-0.7.2.dev1.dist-info/licenses/LICENSE,sha256=Bln6DhK-LtcO4aXy-PBcdZv2f24MlJFm_qn222biJtE,1071
+intentkit-0.7.2.dev1.dist-info/RECORD,,