mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-08-28 05:25:04 +00:00
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
"""Return a one-line summary of every registered job for LLM consumption."""
|
|
|
|
from ..base_step import BaseStep
|
|
from ...components import R
|
|
|
|
|
|
def _format_params(parameters: dict) -> str:
|
|
props = (parameters or {}).get("properties") or {}
|
|
if not props:
|
|
return "no args"
|
|
required = set((parameters or {}).get("required") or [])
|
|
parts = []
|
|
for name, schema in props.items():
|
|
ptype = schema.get("type", "any")
|
|
if name in required:
|
|
parts.append(f"{name}:{ptype}*")
|
|
elif "default" in schema:
|
|
parts.append(f"{name}:{ptype}={schema['default']}")
|
|
else:
|
|
parts.append(f"{name}:{ptype}")
|
|
return ", ".join(parts)
|
|
|
|
|
|
@R.register("help_step")
|
|
class HelpStep(BaseStep):
|
|
"""List all registered jobs (excluding self and non-servable) as compact one-liners for an LLM."""
|
|
|
|
async def execute(self):
|
|
assert self.context is not None
|
|
|
|
lines = []
|
|
if self.app_context is not None:
|
|
for name, job in self.app_context.jobs.items():
|
|
if name == "help" or not getattr(job, "enable_serve", True):
|
|
continue
|
|
lines.append(f"🛠️ `{name}` — {job.description} 📥 {_format_params(job.parameters)}")
|
|
|
|
self.logger.info(f"[{self.name}] returning {len(lines)} jobs")
|
|
self.context.response.answer = "\n".join(lines)
|
|
self.context.response.metadata["job_count"] = len(lines)
|
|
return self.context.response
|