44
55import pytest
66
7- from intentkit .core .credit import expense_message , expense_skill , expense_summarize , skill_cost
7+ from intentkit .core .credit import (
8+ expense_skill ,
9+ expense_summarize ,
10+ skill_cost ,
11+ )
812from intentkit .models .agent import Agent
913from intentkit .models .credit import CreditAccountTable , CreditType , OwnerType
1014
@@ -24,7 +28,9 @@ async def test_skill_cost_self_key_pricing():
2428 mock_skill .author = None
2529
2630 with (
27- patch ("intentkit.core.credit.expense.Skill.get" , new_callable = AsyncMock ) as mock_get ,
31+ patch (
32+ "intentkit.core.credit.expense.Skill.get" , new_callable = AsyncMock
33+ ) as mock_get ,
2834 patch ("intentkit.config.config.config.payment_enabled" , True ),
2935 patch (
3036 "intentkit.models.app_setting.AppSetting.payment" , new_callable = AsyncMock
@@ -49,7 +55,9 @@ async def test_skill_cost_missing_skill():
4955 agent .fee_percentage = Decimal ("0" )
5056 agent .skills = {}
5157
52- with patch ("intentkit.core.credit.expense.Skill.get" , new_callable = AsyncMock ) as mock_get :
58+ with patch (
59+ "intentkit.core.credit.expense.Skill.get" , new_callable = AsyncMock
60+ ) as mock_get :
5361 mock_get .return_value = None
5462 with pytest .raises (ValueError , match = "price of missing" ):
5563 await skill_cost ("missing" , "user-1" , agent )
@@ -119,7 +127,10 @@ def side_effect_refresh(instance):
119127 ),
120128 ):
121129 mock_skill_cost .return_value = skill_cost_info
122- mock_expense .return_value = (mock_user_account , {CreditType .PERMANENT : Decimal ("5.0000" )})
130+ mock_expense .return_value = (
131+ mock_user_account ,
132+ {CreditType .PERMANENT : Decimal ("5.0000" )},
133+ )
123134
124135 result = await expense_skill (
125136 mock_session ,
@@ -232,7 +243,9 @@ async def test_skill_cost_uses_author_fee_recipient():
232243 mock_skill .author = "dev-user"
233244
234245 with (
235- patch ("intentkit.core.credit.expense.Skill.get" , new_callable = AsyncMock ) as mock_get ,
246+ patch (
247+ "intentkit.core.credit.expense.Skill.get" , new_callable = AsyncMock
248+ ) as mock_get ,
236249 patch ("intentkit.config.config.config.payment_enabled" , True ),
237250 patch (
238251 "intentkit.models.app_setting.AppSetting.payment" , new_callable = AsyncMock
0 commit comments