|
|
|
@ -30,11 +30,11 @@ def create_test_graph_runtime_state() -> GraphRuntimeState:
|
|
|
|
llm_usage = LLMUsage(
|
|
|
|
llm_usage = LLMUsage(
|
|
|
|
prompt_tokens=150,
|
|
|
|
prompt_tokens=150,
|
|
|
|
prompt_unit_price=Decimal("0.001"),
|
|
|
|
prompt_unit_price=Decimal("0.001"),
|
|
|
|
prompt_price_unit=Decimal("1000"),
|
|
|
|
prompt_price_unit=Decimal(1000),
|
|
|
|
prompt_price=Decimal("0.15"),
|
|
|
|
prompt_price=Decimal("0.15"),
|
|
|
|
completion_tokens=75,
|
|
|
|
completion_tokens=75,
|
|
|
|
completion_unit_price=Decimal("0.002"),
|
|
|
|
completion_unit_price=Decimal("0.002"),
|
|
|
|
completion_price_unit=Decimal("1000"),
|
|
|
|
completion_price_unit=Decimal(1000),
|
|
|
|
completion_price=Decimal("0.15"),
|
|
|
|
completion_price=Decimal("0.15"),
|
|
|
|
total_tokens=225,
|
|
|
|
total_tokens=225,
|
|
|
|
total_price=Decimal("0.30"),
|
|
|
|
total_price=Decimal("0.30"),
|
|
|
|
@ -121,11 +121,11 @@ def test_llm_usage_round_trip():
|
|
|
|
llm_usage = LLMUsage(
|
|
|
|
llm_usage = LLMUsage(
|
|
|
|
prompt_tokens=100,
|
|
|
|
prompt_tokens=100,
|
|
|
|
prompt_unit_price=Decimal("0.0015"),
|
|
|
|
prompt_unit_price=Decimal("0.0015"),
|
|
|
|
prompt_price_unit=Decimal("1000"),
|
|
|
|
prompt_price_unit=Decimal(1000),
|
|
|
|
prompt_price=Decimal("0.15"),
|
|
|
|
prompt_price=Decimal("0.15"),
|
|
|
|
completion_tokens=50,
|
|
|
|
completion_tokens=50,
|
|
|
|
completion_unit_price=Decimal("0.003"),
|
|
|
|
completion_unit_price=Decimal("0.003"),
|
|
|
|
completion_price_unit=Decimal("1000"),
|
|
|
|
completion_price_unit=Decimal(1000),
|
|
|
|
completion_price=Decimal("0.15"),
|
|
|
|
completion_price=Decimal("0.15"),
|
|
|
|
total_tokens=150,
|
|
|
|
total_tokens=150,
|
|
|
|
total_price=Decimal("0.30"),
|
|
|
|
total_price=Decimal("0.30"),
|
|
|
|
|