jcfs
7faef25f3a
- Add AgentStatusReady status for when agent finishes a prompt - Add AgentReady event published when prompt completes - PromptAgent accepts RUNNING/READY states, marks READY when done - Add /complete endpoint to explicitly complete a task - E2E test polls for READY status instead of file creation - E2E test validates multi-turn conversation flow
68 lines
2.7 KiB
Go
68 lines
2.7 KiB
Go
package v1
|
|
|
|
import "time"
|
|
|
|
// AgentStatus represents the status of an agent instance
|
|
type AgentStatus string
|
|
|
|
const (
|
|
AgentStatusPending AgentStatus = "PENDING"
|
|
AgentStatusStarting AgentStatus = "STARTING"
|
|
AgentStatusRunning AgentStatus = "RUNNING" // Agent is processing a prompt
|
|
AgentStatusReady AgentStatus = "READY" // Agent finished prompt, waiting for next
|
|
AgentStatusCompleted AgentStatus = "COMPLETED"
|
|
AgentStatusFailed AgentStatus = "FAILED"
|
|
AgentStatusStopped AgentStatus = "STOPPED"
|
|
)
|
|
|
|
// ResourceLimits defines container resource limits
|
|
type ResourceLimits struct {
|
|
CPULimit string `json:"cpu_limit"`
|
|
MemoryLimit string `json:"memory_limit"`
|
|
DiskLimit string `json:"disk_limit"`
|
|
}
|
|
|
|
// AgentInstance represents a running or completed agent container
|
|
type AgentInstance struct {
|
|
ID string `json:"id"`
|
|
TaskID string `json:"task_id"`
|
|
AgentType string `json:"agent_type"`
|
|
ContainerID *string `json:"container_id,omitempty"`
|
|
ContainerName *string `json:"container_name,omitempty"`
|
|
Status AgentStatus `json:"status"`
|
|
ImageName string `json:"image_name"`
|
|
StartedAt *time.Time `json:"started_at,omitempty"`
|
|
StoppedAt *time.Time `json:"stopped_at,omitempty"`
|
|
ExitCode *int `json:"exit_code,omitempty"`
|
|
ErrorMessage *string `json:"error_message,omitempty"`
|
|
ResourceLimits ResourceLimits `json:"resource_limits"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// AgentType represents a registered agent type
|
|
type AgentType struct {
|
|
ID string `json:"id"`
|
|
Name string `json:"name"`
|
|
Description string `json:"description"`
|
|
DockerImage string `json:"docker_image"`
|
|
DockerTag string `json:"docker_tag"`
|
|
DefaultResources ResourceLimits `json:"default_resources"`
|
|
EnvironmentVars map[string]string `json:"environment_vars,omitempty"`
|
|
Capabilities []string `json:"capabilities,omitempty"`
|
|
Enabled bool `json:"enabled"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// AgentLog represents a log entry from an agent
|
|
type AgentLog struct {
|
|
ID int64 `json:"id"`
|
|
AgentInstanceID string `json:"agent_instance_id"`
|
|
LogLevel string `json:"log_level"`
|
|
Message string `json:"message"`
|
|
Metadata map[string]interface{} `json:"metadata,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
}
|
|
|