88 lines
3.0 KiB
C
88 lines
3.0 KiB
C
#ifndef CST_SUBAGENT_H
|
|
#define CST_SUBAGENT_H
|
|
|
|
/*
|
|
* Sub-agents: the task(description, prompt, agent?) tool runs one turn of a
|
|
* child agent with its own context and returns its final report and worker_id.
|
|
* Pass worker_id on subsequent calls to continue the persisted context.
|
|
*
|
|
* Roles: built-in general / explore / plan, plus Markdown files in
|
|
* <ws>/.agent/agents/ and <config>/agents/ (the workspace wins, both win
|
|
* over built-ins). Frontmatter: name, description, model, tools (allowlist),
|
|
* profile, max_steps, readonly; the body is the role's instructions.
|
|
*
|
|
* Transcripts: <ws>/.agent/sessions/<parent-id>/sub-<n>.jsonl, with a
|
|
* sub-<n>.json summary next to each (not listed as sessions).
|
|
*/
|
|
|
|
#include <stddef.h>
|
|
#include "cst_agent.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#define CST_ROLES_MAX 32
|
|
|
|
typedef struct {
|
|
char name[64];
|
|
char desc[256];
|
|
char source[16]; /* "builtin" | "workspace" | "user" */
|
|
char model[128]; /* empty = the parent's */
|
|
char profile[64];
|
|
char tools[1024]; /* ",a,b," allowlist; empty = all but task */
|
|
int max_steps; /* 0 = default */
|
|
int read_only;
|
|
char *prompt; /* role instructions (malloc'd) */
|
|
} cst_agent_role_t;
|
|
|
|
/* Roles sorted built-ins first, then by name. Free with cst_roles_free. */
|
|
int cst_roles_list(const char *workspace, cst_agent_role_t *out, int max);
|
|
void cst_roles_free(cst_agent_role_t *v, int n);
|
|
/* System prompt section listing the roles (malloc'd), or NULL. */
|
|
char *cst_subagent_index(const char *workspace);
|
|
/* Register the task tool (owner "builtin:task") for workspace. */
|
|
int cst_subagent_register_tool(const char *workspace);
|
|
|
|
typedef struct {
|
|
char file[64]; /* "sub-3" */
|
|
char role[64];
|
|
char desc[160];
|
|
char status[32]; /* "done" | "error" | "cancelled" | "running" */
|
|
char provider[32], model[128];
|
|
int requests, max_steps;
|
|
long prompt_tokens, cached_tokens, cache_write_tokens, output_tokens;
|
|
int priced;
|
|
int cache_known;
|
|
double cost;
|
|
char currency[8];
|
|
int steps, tools;
|
|
double ms;
|
|
long long mtime;
|
|
} cst_subagent_rec_t;
|
|
|
|
void *cst_worker_budget_new(void);
|
|
void cst_worker_budget_free(void *budget);
|
|
void cst_worker_budget_reset(void *budget);
|
|
char *cst_workers_budget_json(const cst_agent_config_t *cfg);
|
|
/* Validated JSON configuration, stored in <workspace>/.agent/workers.json. */
|
|
char *cst_workers_config_get(const char *workspace);
|
|
int cst_workers_config_set(const char *workspace, const char *json,
|
|
char *err, size_t err_sz);
|
|
void cst_workers_config_apply(cst_agent_config_t *cfg);
|
|
int cst_worker_cancel(const char *workspace, const char *parent_id,
|
|
const char *worker_id);
|
|
|
|
/* This parent session's sub-agent runs, newest first. */
|
|
int cst_subagent_list(const char *workspace, const char *parent_id,
|
|
cst_subagent_rec_t *out, int max);
|
|
/* Full path of a transcript (sub-<n>.jsonl). */
|
|
int cst_subagent_path(const char *workspace, const char *parent_id,
|
|
const char *file, char *out, size_t out_sz);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|