apunkt/dograh
1
0
Fork 0
mirror of https://github.com/dograh-hq/dograh.git synced 2026-07-04 10:52:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

Implement cost calculator for Tuber (#471)

Browse source 
* Adding cost calculation in tuner for BYOK

* fix

* implement cost calculator for Tuner

* Update api/services/integrations/tuner/completion.py

Co-authored-by: cubic-dev-ai[bot] <191113872+cubic-dev-ai[bot]@users.noreply.github.com>

* feat: expose render_options in node spec

* Update api/services/integrations/registry.py

Co-authored-by: cubic-dev-ai[bot] <191113872+cubic-dev-ai[bot]@users.noreply.github.com>

---------

Co-authored-by: mohamed salem <259547077+mohamedsalem-bot@users.noreply.github.com>
Co-authored-by: cubic-dev-ai[bot] <191113872+cubic-dev-ai[bot]@users.noreply.github.com>
Co-authored-by: Abhishek Kumar <abhishek@a6k.me>
This commit is contained in:
Mohamed-Mamdouh 2026-07-02 08:21:14 +01:00 • committed by GitHub
parent 97803b8121
commit 65d46bc313
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
16 changed files with 479 additions and 188 deletions
Download patch file Download diff file Expand all files Collapse all files

38
api/tests/test_node_specs.py
View file

@ -13,6 +13,7 @@ from __future__ import annotations
import re
import pytest
from pydantic import ValidationError
from api.services.workflow.dto import (
ReactFlowDTO,
@ -22,6 +23,7 @@ from api.services.workflow.dto import (
from api.services.workflow.node_data import BaseNodeData
from api.services.workflow.node_specs import (
NodeSpec,
PropertyRendererOptions,
PropertySpec,
PropertyType,
all_specs,
@ -296,6 +298,13 @@ def test_all_registered_node_models_inherit_base_node_data():
"tuner_agent_id",
"tuner_workspace_id",
"tuner_api_key",
"cost_calculation_enabled",
"cost_llm_input_rate",
"cost_llm_cached_input_rate",
"cost_llm_output_rate",
"cost_tts_rate",
"cost_stt_rate",
"cost_telephony_rate",
],
),
],
@ -305,6 +314,33 @@ def test_node_spec_property_order_stable(spec_name: str, expected_order: list[st
assert [prop.name for prop in spec.properties] == expected_order
def test_tuner_cost_rate_fields_use_typed_renderer_options():
spec = next(spec for spec in all_specs() if spec.name == "tuner")
cost_rate_props = [
prop
for prop in spec.properties
if prop.name.startswith("cost_") and prop.name.endswith("_rate")
]
assert len(cost_rate_props) == 6
assert all(prop.renderer_options is not None for prop in cost_rate_props)
assert all(
prop.renderer_options.layout is not None
and prop.renderer_options.layout.column_span == 6
for prop in cost_rate_props
)
assert all(
prop.renderer_options.number_input is not None
and prop.renderer_options.number_input.fractional is True
for prop in cost_rate_props
)
def test_property_renderer_options_reject_unknown_hints():
with pytest.raises(ValidationError):
PropertyRendererOptions.model_validate({"layout": {"width": "half"}})
# ─────────────────────────────────────────────────────────────────────────
# `to_mcp_dict` projection — the lean view served by the `get_node_type`
# MCP tool. UI-only metadata is dropped so it doesn't poison LLM context;
@ -322,7 +358,7 @@ _UI_ONLY_KEYS = frozenset(
"placeholder",
"display_options",
"editor",
"extra",
"renderer_options",
"label", # PropertyOption display string
}
)