refactor: pass ModelInfo instead of model name string through provider interface
This commit is contained in:
parent
8e3ed7010f
commit
d15444bdb0
8 changed files with 83 additions and 43 deletions
|
|
@ -6,6 +6,7 @@ import abc
|
|||
from pathlib import Path
|
||||
|
||||
from bulkgen.config import TargetConfig
|
||||
from bulkgen.providers.models import ModelInfo
|
||||
|
||||
|
||||
class Provider(abc.ABC):
|
||||
|
|
@ -17,7 +18,7 @@ class Provider(abc.ABC):
|
|||
target_name: str,
|
||||
target_config: TargetConfig,
|
||||
resolved_prompt: str,
|
||||
resolved_model: str,
|
||||
resolved_model: ModelInfo,
|
||||
project_dir: Path,
|
||||
) -> None:
|
||||
"""Generate the target artifact and write it to *project_dir / target_name*.
|
||||
|
|
@ -26,6 +27,6 @@ class Provider(abc.ABC):
|
|||
target_name: Output filename (relative to project_dir).
|
||||
target_config: The parsed target configuration.
|
||||
resolved_prompt: The fully resolved prompt text.
|
||||
resolved_model: The resolved model name.
|
||||
resolved_model: The resolved model information.
|
||||
project_dir: The project working directory.
|
||||
"""
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue