intentkit 0.8.6.dev2__py3-none-any.whl → 0.8.17__py3-none-any.whl

intentkit/models/credit.py

@@ -1,8 +1,8 @@
 import logging
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from decimal import ROUND_HALF_UP, Decimal
 from enum import Enum
-from typing import Annotated, Any, Dict, List, Optional, Tuple
+from typing import Annotated, Any
 
 from epyxid import XID
 from fastapi import HTTPException
@@ -169,7 +169,7 @@ class CreditAccountTable(Base):
         DateTime(timezone=True),
         nullable=False,
         server_default=func.now(),
-        onupdate=lambda: datetime.now(timezone.utc),
+        onupdate=lambda: datetime.now(UTC),
     )
 
 
@@ -221,15 +221,15 @@ class CreditAccount(BaseModel):
         Field(default=Decimal("0"), description="Credits added through top-ups"),
     ]
     income_at: Annotated[
-        Optional[datetime],
+        datetime | None,
         Field(None, description="Timestamp of the last income transaction"),
     ]
     expense_at: Annotated[
-        Optional[datetime],
+        datetime | None,
         Field(None, description="Timestamp of the last expense transaction"),
     ]
     last_event_id: Annotated[
-        Optional[str],
+        str | None,
         Field(None, description="ID of the last event that modified this account"),
     ]
     # Total statistics fields
@@ -316,7 +316,7 @@ class CreditAccount(BaseModel):
         """Round decimal values to 4 decimal places."""
         if isinstance(v, Decimal):
             return v.quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
-        elif isinstance(v, (int, float)):
+        elif isinstance(v, int | float):
             return Decimal(str(v)).quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
         return v
 
@@ -413,7 +413,7 @@ class CreditAccount(BaseModel):
         owner_id: str,
         credit_type: CreditType,
         amount: Decimal,
-        event_id: Optional[str] = None,
+        event_id: str | None = None,
     ) -> "CreditAccount":
         """Deduct credits from an account. Not checking balance"""
         # check first, create if not exists
@@ -424,7 +424,7 @@ class CreditAccount(BaseModel):
         values_dict = {
             credit_type.value: getattr(CreditAccountTable, credit_type.value)
             - quantized_amount,
-            "expense_at": datetime.now(timezone.utc),
+            "expense_at": datetime.now(UTC),
             # Update total expense statistics
             "total_expense": CreditAccountTable.total_expense + quantized_amount,
         }
@@ -470,8 +470,8 @@ class CreditAccount(BaseModel):
         owner_type: OwnerType,
         owner_id: str,
         amount: Decimal,
-        event_id: Optional[str] = None,
-    ) -> Tuple["CreditAccount", Dict[CreditType, Decimal]]:
+        event_id: str | None = None,
+    ) -> tuple["CreditAccount", dict[CreditType, Decimal]]:
         """Expense credits and return account and credit type.
         We are not checking balance here, since a conversation may have
         multiple expenses, we can't interrupt the conversation.
@@ -506,7 +506,7 @@ class CreditAccount(BaseModel):
 
         # Create values dict based on what's in details, defaulting to 0 for missing keys
         values_dict = {
-            "expense_at": datetime.now(timezone.utc),
+            "expense_at": datetime.now(UTC),
         }
         if event_id:
             values_dict["last_event_id"] = event_id
@@ -581,14 +581,14 @@ class CreditAccount(BaseModel):
         session: AsyncSession,
         owner_type: OwnerType,
         owner_id: str,
-        amount_details: Dict[CreditType, Decimal],
-        event_id: Optional[str] = None,
+        amount_details: dict[CreditType, Decimal],
+        event_id: str | None = None,
     ) -> "CreditAccount":
         # check first, create if not exists
         await cls.get_or_create_in_session(session, owner_type, owner_id)
         # income
         values_dict = {
-            "income_at": datetime.now(timezone.utc),
+            "income_at": datetime.now(UTC),
         }
         if event_id:
             values_dict["last_event_id"] = event_id
@@ -646,8 +646,8 @@ class CreditAccount(BaseModel):
         session: AsyncSession,
         owner_type: OwnerType,
         owner_id: str,
-        free_quota: Optional[Decimal] = None,
-        refill_amount: Optional[Decimal] = None,
+        free_quota: Decimal | None = None,
+        refill_amount: Decimal | None = None,
     ) -> "CreditAccount":
         """Get an existing credit account or create a new one if it doesn't exist.
 
@@ -687,7 +687,7 @@ class CreditAccount(BaseModel):
             free_credits=free_quota,
             reward_credits=0.0,
             credits=0.0,
-            income_at=datetime.now(timezone.utc),
+            income_at=datetime.now(UTC),
             expense_at=None,
             last_event_id=event_id if owner_type == OwnerType.USER else None,
             # Initialize new statistics fields
@@ -781,8 +781,8 @@ class CreditAccount(BaseModel):
         cls,
         session: AsyncSession,
         user_id: str,
-        free_quota: Optional[Decimal] = None,
-        refill_amount: Optional[Decimal] = None,
+        free_quota: Decimal | None = None,
+        refill_amount: Decimal | None = None,
         upstream_tx_id: str = "",
         note: str = "",
     ) -> "CreditAccount":
@@ -1147,34 +1147,34 @@ class CreditEvent(BaseModel):
     ]
     event_type: Annotated[EventType, Field(description="Type of the event")]
     user_id: Annotated[
-        Optional[str], Field(None, description="ID of the user if applicable")
+        str | None, Field(None, description="ID of the user if applicable")
     ]
     upstream_type: Annotated[
         UpstreamType, Field(description="Type of upstream transaction")
     ]
     upstream_tx_id: Annotated[str, Field(description="Upstream transaction ID if any")]
     agent_id: Annotated[
-        Optional[str], Field(None, description="ID of the agent if applicable")
+        str | None, Field(None, description="ID of the agent if applicable")
     ]
     agent_wallet_address: Annotated[
-        Optional[str],
+        str | None,
         Field(None, description="Wallet address of the agent if applicable"),
     ]
     start_message_id: Annotated[
-        Optional[str],
+        str | None,
         Field(None, description="ID of the starting message if applicable"),
     ]
     message_id: Annotated[
-        Optional[str], Field(None, description="ID of the message if applicable")
+        str | None, Field(None, description="ID of the message if applicable")
     ]
     model: Annotated[
-        Optional[str], Field(None, description="LLM model used if applicable")
+        str | None, Field(None, description="LLM model used if applicable")
     ]
     skill_call_id: Annotated[
-        Optional[str], Field(None, description="ID of the skill call if applicable")
+        str | None, Field(None, description="ID of the skill call if applicable")
     ]
     skill_name: Annotated[
-        Optional[str], Field(None, description="Name of the skill if applicable")
+        str | None, Field(None, description="Name of the skill if applicable")
     ]
     direction: Annotated[Direction, Field(description="Direction of the credit flow")]
     total_amount: Annotated[
@@ -1186,11 +1186,11 @@ class CreditEvent(BaseModel):
     ]
     credit_type: Annotated[CreditType, Field(description="Type of credits involved")]
     credit_types: Annotated[
-        Optional[List[CreditType]],
+        list[CreditType] | None,
         Field(default=None, description="Array of credit types involved"),
     ]
     balance_after: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(None, description="Account total balance after the transaction"),
     ]
     base_amount: Annotated[
@@ -1198,115 +1198,115 @@ class CreditEvent(BaseModel):
         Field(default=Decimal("0"), description="Base amount of credits involved"),
     ]
     base_discount_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Base discount amount"),
     ]
     base_original_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Base original amount"),
     ]
     base_llm_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Base LLM cost amount"),
     ]
     base_skill_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Base skill cost amount"),
     ]
     base_free_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Base free credit amount"),
     ]
     base_reward_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Base reward credit amount"),
     ]
     base_permanent_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Base permanent credit amount"),
     ]
     fee_platform_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Platform fee amount"),
     ]
     fee_platform_free_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(
             default=Decimal("0"), description="Platform fee amount from free credits"
         ),
     ]
     fee_platform_reward_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(
             default=Decimal("0"), description="Platform fee amount from reward credits"
         ),
     ]
     fee_platform_permanent_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(
             default=Decimal("0"),
             description="Platform fee amount from permanent credits",
         ),
     ]
     fee_dev_account: Annotated[
-        Optional[str], Field(None, description="Developer account ID receiving fee")
+        str | None, Field(None, description="Developer account ID receiving fee")
     ]
     fee_dev_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Developer fee amount"),
     ]
     fee_dev_free_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(
             default=Decimal("0"), description="Developer fee amount from free credits"
         ),
     ]
     fee_dev_reward_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(
             default=Decimal("0"), description="Developer fee amount from reward credits"
         ),
     ]
     fee_dev_permanent_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(
             default=Decimal("0"),
             description="Developer fee amount from permanent credits",
         ),
     ]
     fee_agent_account: Annotated[
-        Optional[str], Field(None, description="Agent account ID receiving fee")
+        str | None, Field(None, description="Agent account ID receiving fee")
     ]
     fee_agent_amount: Annotated[
-        Optional[Decimal], Field(default=Decimal("0"), description="Agent fee amount")
+        Decimal | None, Field(default=Decimal("0"), description="Agent fee amount")
     ]
     fee_agent_free_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Agent fee amount from free credits"),
     ]
     fee_agent_reward_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Agent fee amount from reward credits"),
     ]
     fee_agent_permanent_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(
             default=Decimal("0"), description="Agent fee amount from permanent credits"
         ),
     ]
     free_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Free credit amount involved"),
     ]
     reward_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Reward credit amount involved"),
     ]
     permanent_amount: Annotated[
-        Optional[Decimal],
+        Decimal | None,
         Field(default=Decimal("0"), description="Permanent credit amount involved"),
     ]
-    note: Annotated[Optional[str], Field(None, description="Additional notes")]
+    note: Annotated[str | None, Field(None, description="Additional notes")]
     created_at: Annotated[
         datetime, Field(description="Timestamp when this event was created")
     ]
@@ -1339,13 +1339,13 @@ class CreditEvent(BaseModel):
         "permanent_amount",
     )
     @classmethod
-    def round_decimal(cls, v: Any) -> Optional[Decimal]:
+    def round_decimal(cls, v: Any) -> Decimal | None:
         """Round decimal values to 4 decimal places."""
         if v is None:
             return None
         if isinstance(v, Decimal):
             return v.quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
-        elif isinstance(v, (int, float)):
+        elif isinstance(v, int | float):
             return Decimal(str(v)).quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
         return v
 
@@ -1520,7 +1520,7 @@ class CreditTransaction(BaseModel):
         """Round decimal values to 4 decimal places."""
         if isinstance(v, Decimal):
             return v.quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
-        elif isinstance(v, (int, float)):
+        elif isinstance(v, int | float):
             return Decimal(str(v)).quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
         return v
 
@@ -1585,7 +1585,7 @@ class CreditPriceTable(Base):
         DateTime(timezone=True),
         nullable=False,
         server_default=func.now(),
-        onupdate=lambda: datetime.now(timezone.utc),
+        onupdate=lambda: datetime.now(UTC),
     )
 
 
@@ -1623,7 +1623,7 @@ class CreditPrice(BaseModel):
         """Round decimal values to 4 decimal places."""
         if isinstance(v, Decimal):
             return v.quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
-        elif isinstance(v, (int, float)):
+        elif isinstance(v, int | float):
             return Decimal(str(v)).quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
         return v
 
@@ -1700,13 +1700,11 @@ class CreditPriceLog(BaseModel):
         """Round decimal values to 4 decimal places."""
         if isinstance(v, Decimal):
             return v.quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
-        elif isinstance(v, (int, float)):
+        elif isinstance(v, int | float):
             return Decimal(str(v)).quantize(Decimal("0.0001"), rounding=ROUND_HALF_UP)
         return v
 
-    note: Annotated[
-        Optional[str], Field(None, description="Note about the modification")
-    ]
+    note: Annotated[str | None, Field(None, description="Note about the modification")]
     modified_by: Annotated[
         str, Field(description="ID of the user who made the modification")
     ]

intentkit/models/db.py

@@ -1,5 +1,6 @@
+from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
-from typing import Annotated, AsyncGenerator, Optional
+from typing import Annotated
 from urllib.parse import quote_plus
 
 from intentkit.models.db_mig import safe_migrate
@@ -12,7 +13,7 @@ from pydantic import Field
 from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, create_async_engine
 
 engine = None
-_langgraph_checkpointer: Optional[Checkpointer] = None
+_langgraph_checkpointer: Checkpointer | None = None
 
 
 async def check_connection(conn):
@@ -28,11 +29,11 @@ async def check_connection(conn):
 
 
 async def init_db(
-    host: Optional[str],
-    username: Optional[str],
-    password: Optional[str],
-    dbname: Optional[str],
-    port: Annotated[Optional[str], Field(default="5432", description="Database port")],
+    host: str | None,
+    username: str | None,
+    password: str | None,
+    dbname: str | None,
+    port: Annotated[str | None, Field(default="5432", description="Database port")],
     auto_migrate: Annotated[
         bool, Field(default=True, description="Whether to run migrations automatically")
     ],

intentkit/models/db_mig.py

@@ -1,7 +1,7 @@
 """Database migration utilities."""
 
 import logging
-from typing import Callable
+from collections.abc import Callable
 
 from intentkit.models.base import Base
 from sqlalchemy import Column, MetaData, inspect, text
@@ -40,7 +40,7 @@ async def add_column_if_not_exists(
                         default_value = str(default_value).lower()
                     elif isinstance(default_value, str):
                         default_value = f"'{default_value}'"
-                    elif isinstance(default_value, (list, dict)):
+                    elif isinstance(default_value, list | dict):
                         default_value = "'{}'"
                     column_def += f" DEFAULT {default_value}"
 

intentkit/models/llm.csv

@@ -1,23 +1,26 @@
 id,name,provider,enabled,input_price,output_price,price_level,context_length,output_length,intelligence,speed,supports_image_input,supports_skill_calls,supports_structured_output,has_reasoning,supports_search,supports_temperature,supports_frequency_penalty,supports_presence_penalty,api_base,timeout
-qwen/qwen3-235b-a22b-2507,Qwen3 235B A22B Instruct 2507,gatewayz,TRUE,0.1,0.6,1,262000,262000,4,2,FALSE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
-google/gemini-2.5-flash,Gemini 2.5 Flash,gatewayz,TRUE,0.3,2.5,2,1050000,65000,4,2,TRUE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
-x-ai/grok-4-fast:free,Grok 4 Fast,gatewayz,TRUE,0.2,0.5,1,128000,30000,4,2,TRUE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
-x-ai/grok-code-fast-1,Grok Code Fast 1,gatewayz,TRUE,3,15,5,200000,64000,5,1,TRUE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
-anthropic/claude-sonnet-4,Anthropic Claude Sonnet 4,gatewayz,TRUE,3,15,5,200000,64000,5,1,TRUE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+minimax/minimax-m2:free,MiniMax M2,gatewayz,TRUE,0,0,1,204800,131000,4,2,FALSE,TRUE,FALSE,FALSE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+qwen/qwen3-235b-a22b-2507,Qwen3 235B A22B Instruct 2507,gatewayz,TRUE,0.1,0.6,1,262000,262000,3,3,FALSE,TRUE,FALSE,FALSE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+qwen/qwen3-max,Qwen3 Max,gatewayz,TRUE,1.2,6,4,128000,32000,4,2,FALSE,TRUE,FALSE,FALSE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+google/gemini-2.5-flash,Gemini 2.5 Flash,gatewayz,TRUE,0.3,2.5,2,1050000,65000,3,4,TRUE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+google/gemini-2.5-pro,Gemini 2.5 Pro,gatewayz,TRUE,1.25,10,4,1050000,65000,4,2,TRUE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+anthropic/claude-sonnet-4.5,Anthropic Claude Sonnet 4.5,gatewayz,TRUE,3,15,5,200000,64000,5,1,TRUE,TRUE,FALSE,TRUE,FALSE,TRUE,FALSE,FALSE,https://api.gatewayz.ai/v1,300
+gpt-5-nano,GPT-5 Nano,openai,TRUE,0.05,0.4,1,400000,128000,3,5,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,,180
+gpt-5-mini,GPT-5 Mini,openai,TRUE,0.25,2,2,400000,128000,4,4,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,,180
+gpt-5,GPT-5,openai,TRUE,1.25,10,4,400000,128000,5,3,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,,180
 gpt-4o,GPT-4o,openai,FALSE,2.5,10,4,128000,4096,4,3,TRUE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
 gpt-4o-mini,GPT-4o Mini,openai,FALSE,0.15,0.6,1,128000,4096,3,4,FALSE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
 gpt-4.1-nano,GPT-4.1 Nano,openai,TRUE,0.1,0.4,1,128000,4096,3,5,FALSE,TRUE,TRUE,FALSE,FALSE,TRUE,FALSE,FALSE,,180
 gpt-4.1-mini,GPT-4.1 Mini,openai,TRUE,0.4,1.6,2,128000,4096,4,4,FALSE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
 gpt-4.1,GPT-4.1,openai,TRUE,2,8,4,128000,4096,5,3,TRUE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
-gpt-5-nano,GPT-5 Nano,openai,TRUE,0.05,0.4,1,400000,128000,3,5,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,,180
-gpt-5-mini,GPT-5 Mini,openai,TRUE,0.25,2,2,400000,128000,4,4,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,,180
-gpt-5,GPT-5,openai,TRUE,1.25,10,4,400000,128000,5,3,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,,180
 o3,OpenAI o3,openai,TRUE,2,8,4,200000,100000,5,1,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,,180
 o4-mini,OpenAI o4-mini,openai,TRUE,1.1,4.4,3,128000,4096,4,3,FALSE,TRUE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,,180
-deepseek-chat,Deepseek V3.1,deepseek,TRUE,0.56,1.68,2,128000,4096,4,3,FALSE,TRUE,TRUE,FALSE,FALSE,TRUE,TRUE,TRUE,https://api.deepseek.com,300
-deepseek-reasoner,DeepSeek V3.1 (Thinking Mode),deepseek,TRUE,0.56,1.68,2,128000,32000,5,1,FALSE,TRUE,TRUE,TRUE,FALSE,TRUE,TRUE,TRUE,https://api.deepseek.com,300
-grok-2,Grok 2,xai,TRUE,2,10,4,120000,4096,3,3,FALSE,TRUE,TRUE,FALSE,FALSE,TRUE,TRUE,TRUE,,180
-grok-3,Grok 3,xai,TRUE,3,15,4,131072,4096,5,3,FALSE,TRUE,TRUE,FALSE,TRUE,TRUE,TRUE,TRUE,,180
+deepseek-chat,Deepseek V3.2 Exp,deepseek,TRUE,0.28,0.42,2,128000,4096,4,3,FALSE,TRUE,TRUE,FALSE,FALSE,TRUE,TRUE,TRUE,https://api.deepseek.com,300
+deepseek-reasoner,DeepSeek V3.2 Exp (Thinking Mode),deepseek,TRUE,0.28,0.42,2,128000,32000,5,1,FALSE,TRUE,TRUE,TRUE,FALSE,TRUE,TRUE,TRUE,https://api.deepseek.com,300
+grok-4,Grok 4,xai,TRUE,3,15,4,256000,4096,5,3,TRUE,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,,180
+grok-4-fast-non-reasoning,Grok 4 Fast,xai,TRUE,0.2,0.5,2,2000000,4096,3,4,TRUE,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,,180
+grok-code-fast-1,Grok Code Fast 1,xai,TRUE,0.2,1.5,3,256000,4096,4,4,FALSE,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,,180
+grok-3,Grok 3,xai,TRUE,3,15,4,131072,4096,5,3,FALSE,TRUE,TRUE,FALSE,TRUE,TRUE,FALSE,FALSE,,180
 grok-3-mini,Grok 3 Mini,xai,TRUE,0.3,0.5,2,131072,4096,5,3,FALSE,TRUE,TRUE,TRUE,TRUE,TRUE,FALSE,FALSE,,180
 eternalai,Eternal AI (Llama-3.3-70B),eternal,FALSE,0.25,0.75,2,64000,4096,4,3,FALSE,TRUE,TRUE,FALSE,FALSE,TRUE,TRUE,TRUE,https://api.eternalai.org/v1,300
 reigent,REI Network,reigent,FALSE,0.5,1.5,3,32000,4096,4,3,FALSE,TRUE,TRUE,FALSE,FALSE,FALSE,FALSE,FALSE,https://api.reisearch.box/v1,300

intentkit/models/llm.py

@@ -1,18 +1,18 @@
 import csv
 import json
 import logging
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 from decimal import ROUND_HALF_UP, Decimal
 from enum import Enum
 from pathlib import Path
-from typing import Annotated, Any, Optional
+from typing import Annotated, Any
 
 from intentkit.config.config import config
 from intentkit.models.app_setting import AppSetting
 from intentkit.models.base import Base
 from intentkit.models.db import get_session
 from intentkit.models.redis import get_redis
-from intentkit.utils.error import IntentKitLookUpError
+from intentkit.utils.error import IntentKitAPIError
 from langchain.chat_models.base import BaseChatModel
 from pydantic import BaseModel, ConfigDict, Field
 from sqlalchemy import Boolean, Column, DateTime, Integer, Numeric, String, func, select
@@ -24,13 +24,13 @@ _credit_per_usdc = None
 FOURPLACES = Decimal("0.0001")
 
 
-def _parse_bool(value: Optional[str]) -> bool:
+def _parse_bool(value: str | None) -> bool:
     if value is None:
         return False
     return value.strip().lower() in {"true", "1", "yes"}
 
 
-def _parse_optional_int(value: Optional[str]) -> Optional[int]:
+def _parse_optional_int(value: str | None) -> int | None:
     if value is None:
         return None
     value = value.strip()
@@ -45,12 +45,12 @@ def _load_default_llm_models() -> dict[str, "LLMModelInfo"]:
         logger.warning("Default LLM CSV not found at %s", path)
         return {}
 
-    defaults: dict[str, "LLMModelInfo"] = {}
+    defaults: dict[str, LLMModelInfo] = {}
     with path.open(newline="", encoding="utf-8") as csvfile:
         reader = csv.DictReader(csvfile)
         for row in reader:
             try:
-                timestamp = datetime.now(timezone.utc)
+                timestamp = datetime.now(UTC)
                 model = LLMModelInfo(
                     id=row["id"],
                     name=row["name"],
@@ -154,7 +154,7 @@ class LLMModelInfoTable(Base):
         DateTime(timezone=True),
         nullable=False,
         server_default=func.now(),
-        onupdate=lambda: datetime.now(timezone.utc),
+        onupdate=lambda: datetime.now(UTC),
     )
 
 
@@ -173,7 +173,7 @@ class LLMModelInfo(BaseModel):
     enabled: bool = Field(default=True)
     input_price: Decimal  # Price per 1M input tokens in USD
     output_price: Decimal  # Price per 1M output tokens in USD
-    price_level: Optional[int] = Field(
+    price_level: int | None = Field(
         default=None, ge=1, le=5
     )  # Price level rating from 1-5
     context_length: int  # Maximum context length in tokens
@@ -198,22 +198,20 @@ class LLMModelInfo(BaseModel):
     supports_presence_penalty: bool = (
         True  # Whether the model supports presence_penalty parameter
     )
-    api_base: Optional[str] = (
-        None  # Custom API base URL if not using provider's default
-    )
+    api_base: str | None = None  # Custom API base URL if not using provider's default
     timeout: int = 180  # Default timeout in seconds
     created_at: Annotated[
         datetime,
         Field(
             description="Timestamp when this data was created",
-            default=datetime.now(timezone.utc),
+            default=datetime.now(UTC),
         ),
     ]
     updated_at: Annotated[
         datetime,
         Field(
             description="Timestamp when this data was updated",
-            default=datetime.now(timezone.utc),
+            default=datetime.now(UTC),
         ),
     ]
 
@@ -282,7 +280,11 @@ class LLMModelInfo(BaseModel):
             return model_info
 
         # Not found anywhere
-        raise IntentKitLookUpError(f"Model {model_id} not found")
+        raise IntentKitAPIError(
+            400,
+            "ModelNotFound",
+            f"Model {model_id} not found, maybe deprecated, please change it in the agent configuration.",
+        )
 
     @classmethod
     async def get_all(cls, session: AsyncSession | None = None) -> list["LLMModelInfo"]:
@@ -292,7 +294,7 @@ class LLMModelInfo(BaseModel):
             async with get_session() as db:
                 return await cls.get_all(session=db)
 
-        models: dict[str, "LLMModelInfo"] = {
+        models: dict[str, LLMModelInfo] = {
             model_id: model.model_copy(deep=True)
             for model_id, model in AVAILABLE_MODELS.items()
         }

intentkit/models/redis.py

@@ -1,21 +1,20 @@
 """Redis client module for IntentKit."""
 
 import logging
-from typing import Optional
 
 from redis.asyncio import Redis
 
 logger = logging.getLogger(__name__)
 
 # Global Redis client instance
-_redis_client: Optional[Redis] = None
+_redis_client: Redis | None = None
 
 
 async def init_redis(
     host: str,
     port: int = 6379,
     db: int = 0,
-    password: Optional[str] = None,
+    password: str | None = None,
     ssl: bool = False,
     encoding: str = "utf-8",
     decode_responses: bool = True,