zhengchen.tao/ezbookkeeping
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

the query_transactions_tool_handler mcp tool supports filtering multiple categories or accounts with the same name, and filtering sub-accounts / secondary categories by their parent account / category name

Browse Source
This commit is contained in:
MaysWind
2025-09-20 17:45:21 +08:00
parent d50ce0140f
commit 4def7ed60c
3 changed files with 88 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/mcp/query_transactions_tool_handler.go
+6 -8
View File
@@ -126,13 +126,12 @@ func (h *mcpQueryTransactionsToolHandler) Handle(c *core.WebContext, callToolReq
return nil, nil, err
}
accountsMap := services.GetAccountService().GetVisibleAccountNameMapByList(allAccounts)
filterAccountIds := make([]int64, 0)
if queryTransactionsRequest.AccountName != "" {
if account, exists := accountsMap[queryTransactionsRequest.AccountName]; exists {
filterAccountIds = append(filterAccountIds, account.AccountId)
} else {
filterAccountIds = services.GetAccountService().GetAccountOrSubAccountIdsByAccountName(allAccounts, queryTransactionsRequest.AccountName)
if len(filterAccountIds) < 1 {
return nil, nil, errs.ErrAccountNotFound
}
}
@@ -144,13 +143,12 @@ func (h *mcpQueryTransactionsToolHandler) Handle(c *core.WebContext, callToolReq
return nil, nil, err
}
categoriesMap := services.GetTransactionCategoryService().GetVisibleCategoryNameMapByList(allCategories)
filterCategoryIds := make([]int64, 0)
if queryTransactionsRequest.SecondaryCategoryName != "" {
if category, exists := categoriesMap[queryTransactionsRequest.SecondaryCategoryName]; exists {
filterCategoryIds = append(filterCategoryIds, category.CategoryId)
} else {
filterCategoryIds = services.GetTransactionCategoryService().GetCategoryOrSubCategoryIdsByCategoryName(allCategories, queryTransactionsRequest.SecondaryCategoryName)
if len(filterCategoryIds) < 1 {
return nil, nil, errs.ErrTransactionCategoryNotFound
}
}