Skip to content

Commit 435fbb0

Browse files
committed
update agentic capability.
1 parent 31957f0 commit 435fbb0

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

src/agentic_capability_generator.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
def main(cfg: DictConfig) -> None:
3030
"""Run the multi-agent debate-based capability generation system."""
3131
areas_tag = cfg.pipeline_tags.areas_tag
32+
resume_tag = getattr(cfg.pipeline_tags, "resume_capabilities_tag", None)
3233
domain_name = cfg.global_cfg.domain
3334
exp_id = cfg.exp_cfg.exp_id
3435
num_capabilities_per_area = cfg.capability_generation.num_capabilities_per_area
@@ -68,18 +69,26 @@ def main(cfg: DictConfig) -> None:
6869
)
6970
return
7071

72+
if resume_tag:
73+
msg = f"Resuming capability generation from tag: {resume_tag}"
74+
log.info(msg)
75+
span.update(
76+
metadata={"resume_tag_found": msg, "resume_tag": resume_tag}
77+
)
78+
7179
span.update_trace(
7280
metadata={
7381
"domain": domain_name,
7482
"exp_id": exp_id,
7583
"areas_tag": areas_tag,
84+
"resume_tag": resume_tag,
7685
"num_capabilities_per_area": num_capabilities_per_area,
7786
"config": config_yaml,
7887
},
7988
tags=["agentic_capability_generation", exp_id],
8089
)
8190

82-
asyncio.run(generate_capabilities(cfg, areas_tag, lf))
91+
asyncio.run(generate_capabilities(cfg, areas_tag, lf, resume_tag))
8392

8493
msg = (
8594
"Multi-agent debate-based capability generation completed successfully"

0 commit comments

Comments
 (0)