diff --git a/pricing/pricing.py b/pricing/pricing.py index 7dce329..257ca94 100644 --- a/pricing/pricing.py +++ b/pricing/pricing.py @@ -37,7 +37,7 @@ async def get_remote_pricing(sor, charge, data): return d async def pricing_program_charging(sor, pricing_program_id, data): - pp = PricingProgram(pricing_program_id) + pp = PricingProgram(pricing_program_id, sor) await pp.init() pp_items = pp.get_items() specs = pt.get_spec() diff --git a/wwwroot/test_pricing_program.dspy b/wwwroot/test_pricing_program.dspy index 73b23be..a853295 100644 --- a/wwwroot/test_pricing_program.dspy +++ b/wwwroot/test_pricing_program.dspy @@ -5,6 +5,7 @@ async with db.sqlorContext(dbname) as sor: 'input_tokens': 123456, 'completions_tokens': 332234 } - x = pricing_program_charging(sor, 'vq5OpFe5JTA255sdRNWtD', data) - print(x) + x = await pricing_program_charging(sor, 'vq5OpFe5JTA255sdRNWtD', data) + return x +return f'Except:{db.e_except},{format_exc()}'