Extend AI Toolkit MCP Server

from typing import List, TypedDict, Type from pydantic import BaseModel from .configuration import Scope, Product from .enums import ExtendAPITools from .prompts import ( get_virtual_cards_prompt, get_virtual_card_detail_prompt, cancel_virtual_card_prompt, close_virtual_card_prompt, get_transactions_prompt, get_transaction_detail_prompt, get_credit_cards_prompt, get_expense_categories_prompt, get_expense_category_prompt, get_expense_category_labels_prompt, create_expense_category_prompt, create_expense_category_label_prompt, update_expense_category_prompt, get_credit_card_detail_prompt, update_transaction_expense_data_prompt, create_receipt_attachment_prompt, get_automatch_status_prompt, automatch_receipts_prompt, send_receipt_reminder_prompt, ) from .schemas import ( GetVirtualCards, GetVirtualCardDetail, CancelVirtualCard, CloseVirtualCard, GetCreditCards, GetTransactions, GetTransactionDetail, GetExpenseCategories, GetExpenseCategory, GetExpenseCategoryLabels, CreateExpenseCategory, CreateExpenseCategoryLabel, UpdateExpenseCategory, GetCreditCardDetail, UpdateTransactionExpenseData, GetAutomatchStatusSchema, AutomatchReceiptsSchema, CreateReceiptAttachmentSchema, SendReceiptReminderSchema, ) class ActionDict(TypedDict): read: bool create: bool update: bool delete: bool class Tool(BaseModel): method: ExtendAPITools description: str args_schema: Type[BaseModel] required_scope: List[Scope] @property def name(self) -> str: return self.method.value tools: List[Tool] = [ Tool( method=ExtendAPITools.GET_VIRTUAL_CARDS, description=get_virtual_cards_prompt, args_schema=GetVirtualCards, required_scope=[ Scope( type=Product.VIRTUAL_CARDS, actions={"read": True}) ], ), Tool( method=ExtendAPITools.GET_VIRTUAL_CARD_DETAIL, description=get_virtual_card_detail_prompt, args_schema=GetVirtualCardDetail, required_scope=[ Scope( type=Product.VIRTUAL_CARDS, actions={"read": True}) ], ), Tool( method=ExtendAPITools.CANCEL_VIRTUAL_CARD, description=cancel_virtual_card_prompt, args_schema=CancelVirtualCard, required_scope=[ Scope( type=Product.VIRTUAL_CARDS, actions={ "read": True, "update": True, }) ], ), Tool( method=ExtendAPITools.CLOSE_VIRTUAL_CARD, description=close_virtual_card_prompt, args_schema=CloseVirtualCard, required_scope=[ Scope( type=Product.VIRTUAL_CARDS, actions={ "read": True, "update": True, }) ], ), Tool( method=ExtendAPITools.GET_CREDIT_CARDS, description=get_credit_cards_prompt, args_schema=GetCreditCards, required_scope=[ Scope( type=Product.CREDIT_CARDS, actions={ "read": True, }) ], ), Tool( method=ExtendAPITools.GET_CREDIT_CARD_DETAIL, description=get_credit_card_detail_prompt, args_schema=GetCreditCardDetail, required_scope=[ Scope( type=Product.CREDIT_CARDS, actions={"read": True} ) ], ), Tool( method=ExtendAPITools.GET_TRANSACTIONS, description=get_transactions_prompt, args_schema=GetTransactions, required_scope=[ Scope( type=Product.TRANSACTIONS, actions={ "read": True, }) ], ), Tool( method=ExtendAPITools.GET_TRANSACTION_DETAIL, description=get_transaction_detail_prompt, args_schema=GetTransactionDetail, required_scope=[ Scope( type=Product.TRANSACTIONS, actions={ "read": True, }) ], ), Tool( method=ExtendAPITools.UPDATE_TRANSACTION_EXPENSE_DATA, description=update_transaction_expense_data_prompt, args_schema=UpdateTransactionExpenseData, required_scope=[ Scope( type=Product.TRANSACTIONS, actions={ "read": True, "update": True, } ) ], ), Tool( method=ExtendAPITools.GET_EXPENSE_CATEGORIES, description=get_expense_categories_prompt, args_schema=GetExpenseCategories, required_scope=[ Scope( type=Product.EXPENSE_CATEGORIES, actions={"read": True} ) ], ), Tool( method=ExtendAPITools.GET_EXPENSE_CATEGORY, description=get_expense_category_prompt, args_schema=GetExpenseCategory, required_scope=[ Scope( type=Product.EXPENSE_CATEGORIES, actions={"read": True} ) ], ), Tool( method=ExtendAPITools.GET_EXPENSE_CATEGORY_LABELS, description=get_expense_category_labels_prompt, args_schema=GetExpenseCategoryLabels, required_scope=[ Scope( type=Product.EXPENSE_CATEGORIES, actions={"read": True} ) ], ), Tool( method=ExtendAPITools.CREATE_EXPENSE_CATEGORY, description=create_expense_category_prompt, args_schema=CreateExpenseCategory, required_scope=[ Scope( type=Product.EXPENSE_CATEGORIES, actions={"read": True, "create": True} ) ], ), Tool( method=ExtendAPITools.CREATE_EXPENSE_CATEGORY_LABEL, description=create_expense_category_label_prompt, args_schema=CreateExpenseCategoryLabel, required_scope=[ Scope( type=Product.EXPENSE_CATEGORIES, actions={"read": True, "create": True} ) ], ), Tool( method=ExtendAPITools.UPDATE_EXPENSE_CATEGORY, description=update_expense_category_prompt, args_schema=UpdateExpenseCategory, required_scope=[ Scope( type=Product.EXPENSE_CATEGORIES, actions={"read": True, "update": True} ) ], ), Tool( method=ExtendAPITools.CREATE_RECEIPT_ATTACHMENT, description=create_receipt_attachment_prompt, args_schema=CreateReceiptAttachmentSchema, required_scope=[ Scope( type=Product.RECEIPT_ATTACHMENTS, actions={"read": True, "create": True} ), Scope( type=Product.TRANSACTIONS, actions={"read": True, "update": True} ) ], ), Tool( method=ExtendAPITools.AUTOMATCH_RECEIPTS, description=automatch_receipts_prompt, args_schema=AutomatchReceiptsSchema, required_scope=[ Scope( type=Product.RECEIPT_ATTACHMENTS, actions={"read": True} ), Scope( type=Product.TRANSACTIONS, actions={"read": True, "update": True} ) ], ), Tool( method=ExtendAPITools.GET_AUTOMATCH_STATUS, description=get_automatch_status_prompt, args_schema=GetAutomatchStatusSchema, required_scope=[ Scope( type=Product.RECEIPT_ATTACHMENTS, actions={"read": True} ) ], ), Tool( method=ExtendAPITools.SEND_RECEIPT_REMINDER, description=send_receipt_reminder_prompt, args_schema=SendReceiptReminderSchema, required_scope=[ Scope( type=Product.RECEIPT_ATTACHMENTS, actions={"read": True} ), Scope( type=Product.TRANSACTIONS, actions={"read": True} ) ], ), # Tool( # method=ExtendAPITools.PROPOSE_EXPENSE_CATEGORY_LABEL, # name="propose_transaction_expense_data", # description=propose_transaction_expense_data_prompt, # args_schema=ProposeTransactionExpenseData, # required_scope=[ # Scope( # type=Product.TRANSACTIONS, # actions={"read": True} # ), # Scope( # type=Product.EXPENSE_CATEGORIES, # actions={"read": True} # ) # ], # ), # Tool( # method=ExtendAPITools.CONFIRM_EXPENSE_CATEGORY_LABEL, # name="confirm_transaction_expense_data", # description=confirm_transaction_expense_data_prompt, # args_schema=ConfirmTransactionExpenseData, # required_scope=[ # Scope( # type=Product.TRANSACTIONS, # actions={"read": True, "update": True} # ), # Scope( # type=Product.EXPENSE_CATEGORIES, # actions={"read": True} # ) # ], # ) ]