diff --git a/hive.yaml b/hive.yaml index 77bf082..13e7dce 100644 --- a/hive.yaml +++ b/hive.yaml @@ -24,6 +24,10 @@ sandbox: - name: foo value: bar prompt: + context: foo + ideas: + - foo + - bar enable_evolution: false provider: gcp: diff --git a/src/hive_cli/config.py b/src/hive_cli/config.py index cd79f64..dfc155b 100644 --- a/src/hive_cli/config.py +++ b/src/hive_cli/config.py @@ -69,7 +69,18 @@ class SandboxConfig(BaseModel): class PromptConfig(BaseModel): - enable_evolution: bool = False + context: Optional[str] = Field( + default=None, + description="Some useful experiment-specific context to provide to the Hive.", + ) + ideas: Optional[list[str]] = Field( + default=None, + description="A list of ideas which will be randomly sampled to inject into the Hive.", + ) + enable_evolution: bool = Field( + default=False, + description="Whether to enable evolution for the experiment. Default to False.", + ) class RepoConfig(BaseModel): diff --git a/src/hive_cli/platform/k8s.py b/src/hive_cli/platform/k8s.py index 769a47d..303a125 100644 --- a/src/hive_cli/platform/k8s.py +++ b/src/hive_cli/platform/k8s.py @@ -279,6 +279,8 @@ def construct_experiment(name: str, namespace: str, config: HiveConfig) -> dict: if config.prompt: result["spec"]["prompt"] = { "enableEvolution": config.prompt.enable_evolution, + "context": config.prompt.context, + "ideas": config.prompt.ideas, } return result