mirror of
https://github.com/block/buzz.git
synced 2026-10-30 05:51:46 +00:00
f437e43d09
Signed-off-by: npub12gtutshhh76rx0jx697f32f9tffd4hhp3hx58fp4x6u4uemkm7sqf8f757 <5217c5c2f7bfb4333e46d17c98a9255a52dadee18dcd43a43536b95e6776dfa0@sprout-oss.stage.blox.sqprod.co> Signed-off-by: Tyler Longwell <tlongwell@block.xyz> Co-authored-by: npub12gtutshhh76rx0jx697f32f9tffd4hhp3hx58fp4x6u4uemkm7sqf8f757 <5217c5c2f7bfb4333e46d17c98a9255a52dadee18dcd43a43536b95e6776dfa0@sprout-oss.stage.blox.sqprod.co> Co-authored-by: npub1qyvc0c5kl4gqv2fd97fsk46tu378sqgy35vc83rvgfwne90sel7s0ed67d <011987e296fd5006292d2f930b574be47c7801048d1983c46c425d3c95f0cffd@sprout-oss.stage.blox.sqprod.co> Co-authored-by: Tyler Longwell <tlongwell@block.xyz>
60 lines
2.0 KiB
Python
60 lines
2.0 KiB
Python
"""Shared fixtures for testbed provisioning tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
from harbor_buzz_orchestra.manifest import ExperimentManifest
|
|
|
|
PROMPT_REF = {"path": "personas/x.md", "sha256": "0" * 64}
|
|
|
|
|
|
@pytest.fixture()
|
|
def manifest() -> ExperimentManifest:
|
|
"""A minimal two-class manifest: one orchestrator, two workers."""
|
|
return ExperimentManifest.load(
|
|
{
|
|
"condition": "O-test",
|
|
"roster": [
|
|
{
|
|
"id": "worker-glm",
|
|
"kind": "worker",
|
|
"role": "implementer",
|
|
"count": 2,
|
|
"endpoint": "databricks/glm",
|
|
"model_revision": "glm-5.2",
|
|
"prompt": PROMPT_REF,
|
|
"generation": {
|
|
"max_output_tokens": 4096,
|
|
"context_window_tokens": 128000,
|
|
},
|
|
},
|
|
{
|
|
"id": "orch-opus",
|
|
"kind": "orchestrator",
|
|
"role": "lead",
|
|
"count": 1,
|
|
"endpoint": "databricks/opus",
|
|
"model_revision": "opus-4.8",
|
|
"prompt": PROMPT_REF,
|
|
"generation": {
|
|
"max_output_tokens": 8192,
|
|
"context_window_tokens": 200000,
|
|
},
|
|
},
|
|
],
|
|
"prices": {
|
|
"databricks/glm": {
|
|
"input_per_million_usd": 0.5,
|
|
"cached_input_per_million_usd": 0.1,
|
|
"output_per_million_usd": 1.5,
|
|
},
|
|
"databricks/opus": {
|
|
"input_per_million_usd": 15.0,
|
|
"cached_input_per_million_usd": 1.5,
|
|
"output_per_million_usd": 75.0,
|
|
},
|
|
},
|
|
"trial_budget": {"timeout_seconds": 3600},
|
|
}
|
|
)
|