kfickel/hokusai
1
0
Fork 0
Code Issues 1 Pull requests 1 Projects Releases Packages Wiki Activity Actions

refactor: pass ModelInfo instead of model name string through provider interface

Browse source
This commit is contained in:
Konstantin Fickel 2026-02-15 08:12:04 +01:00
parent 8e3ed7010f
commit d15444bdb0
Signed by: kfickel
GPG key ID: A793722F9933C1A5
8 changed files with 83 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/test_builder.py
View file

@ -18,6 +18,7 @@ from bulkgen.builder import (
)
from bulkgen.config import ProjectConfig, TargetConfig, TargetType
from bulkgen.providers import Provider
from bulkgen.providers.models import ModelInfo
from bulkgen.state import load_state
WriteConfig = Callable[[dict[str, object]], ProjectConfig]
@ -32,7 +33,7 @@ class FakeProvider(Provider):
target_name: str,
target_config: TargetConfig,
resolved_prompt: str,
resolved_model: str,
resolved_model: ModelInfo,
project_dir: Path,
) -> None:
output = project_dir / target_name
@ -48,7 +49,7 @@ class FailingProvider(Provider):
target_name: str,
target_config: TargetConfig,
resolved_prompt: str,
resolved_model: str,
resolved_model: ModelInfo,
project_dir: Path,
) -> None:
msg = f"Simulated failure for {target_name}"
@ -247,7 +248,7 @@ class TestRunBuild:
target_name: str,
target_config: TargetConfig,
resolved_prompt: str,
resolved_model: str,
resolved_model: ModelInfo,
project_dir: Path,
) -> None:
if target_name == "fail.txt":