mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-11 03:12:55 +00:00
feat(pricing): rates for the default DocsGPT model
$0.15 input, $0.50 output and $0.03 cached input per 1M tokens, so cost budgets see usage of the default model instead of recording it at $0.
This commit is contained in:
1 parent
4bd259fc09
commit
3a30f5cce9
2 files changed
+13
-1
No files matched your search
@@ -7,3 +7,6 @@ models:
|
||||
supports_tools: true
|
||||
attachments: [image]
|
||||
context_window: 1048576
|
||||
input_cost_per_million: 0.15
|
||||
output_cost_per_million: 0.5
|
||||
cached_input_cost_per_million: 0.03
|
||||
+10
-1
@@ -118,7 +118,7 @@ class TestCatalogFields:
|
||||
self._load(tmp_path, "provider: openai\nmodels:\n - id: m\n input_cost_per_million: -1\n")
|
||||
|
||||
def test_hosted_builtin_models_are_priced(self):
|
||||
hosted = {"anthropic", "deepseek", "google", "groq", "novita", "openai", "openrouter"}
|
||||
hosted = {"anthropic", "deepseek", "docsgpt", "google", "groq", "novita", "openai", "openrouter"}
|
||||
catalogs = [
|
||||
c for c in load_model_yamls([BUILTIN_MODELS_DIR]) if c.source_path.stem in hosted
|
||||
]
|
||||
@@ -128,3 +128,12 @@ class TestCatalogFields:
|
||||
caps = model.capabilities
|
||||
assert caps.input_cost_per_million is not None, model.id
|
||||
assert caps.output_cost_per_million is not None, model.id
|
||||
|
||||
def test_default_docsgpt_model_rates(self):
|
||||
(model,) = [
|
||||
m for c in load_model_yamls([BUILTIN_MODELS_DIR]) for m in c.models if m.id == "docsgpt-local"
|
||||
]
|
||||
caps = model.capabilities
|
||||
assert (caps.input_cost_per_million, caps.output_cost_per_million) == (0.15, 0.5)
|
||||
assert caps.cached_input_cost_per_million == 0.03
|
||||
assert caps.cache_write_cost_per_million is None
|
||||
Reference in new issue
Block a user