Method Handler

runtime/ai/developer_agent.go:49–112

(ctx context.Context, args *DeveloperAgentArgs)

}

func (t *DeveloperAgent) Handler(ctx context.Context, args *DeveloperAgentArgs) (*DeveloperAgentResult, error) {
	// Generate the prompts
	systemPrompt, err := t.systemPrompt()
	if err != nil {
		return nil, err
	}
	userPrompt, err := t.userPrompt(ctx, args)
	if err != nil {
		return nil, err
	}

	// Pre-invoke some tool calls
	s := GetSession(ctx)
	_, err = s.CallTool(ctx, RoleAssistant, ListFilesName, nil, &ListFilesArgs{})
	if err != nil {
		return nil, err
	}
	_, err = s.CallTool(ctx, RoleAssistant, ProjectStatusName, nil, &ProjectStatusArgs{})
	if err != nil {
		return nil, err
	}
	if args.CurrentFilePath != "" {
		_, _ = s.CallTool(ctx, RoleAssistant, ReadFileName, nil, &ReadFileArgs{
			Path: args.CurrentFilePath,
		})
		if ctx.Err() != nil { // Ignore tool error since the file may not exist
			return nil, ctx.Err()
		}
	}

	// Build initial completion messages
	messages := []*aiv1.CompletionMessage{NewTextCompletionMessage(RoleSystem, systemPrompt)}
	messages = append(messages, s.NewCompletionMessages(s.MessagesWithResults(FilterByType(MessageTypeCall), FilterByTool(DeveloperAgentName)))...)
	messages = append(messages, NewTextCompletionMessage(RoleUser, userPrompt))
	messages = append(messages, s.NewCompletionMessages(s.MessagesWithResults(FilterByParent(s.ParentID)))...)

	// Run an LLM tool call loop
	var response string
	err = s.Complete(ctx, "Developer loop", &response, &CompleteOptions{
		Messages: messages,
		Tools: []string{
			ProjectStatusName,
			ListFilesName,
			SearchFilesName,
			ReadFileName,
			ListBucketsName,
			ListBucketObjectsName,
			ListTablesName,
			ShowTableName,
			QuerySQLName,
			DevelopFileName,
			NavigateName,
		},
		MaxIterations: 20,
		UnwrapCall:    true,
	})
	if err != nil {
		return nil, err