-
Notifications
You must be signed in to change notification settings - Fork 2.2k
feat: Add agent tasks API support #4225
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: master
Are you sure you want to change the base?
Changes from all commits
8d3c4df
efa12d7
4263d58
43387ee
3c4d321
93e009c
738801b
ceabbf4
f8042ad
a2f45a2
5160bbf
ab0cc87
d10a780
a982cae
41b1ac1
dc0817a
8e1ed64
96ff69b
d40f10b
526dbeb
ca7b94c
f60454c
38fdd45
127e602
e939a80
0de7fd7
5669508
4d43376
3637558
139e26e
24294fe
0c9c94e
a39607d
0ee670d
848a6ab
c8a51f9
7053044
77ee6f4
5847795
332e7f9
2f707eb
67b9739
5aeb196
d2b4f56
2860c60
cdc4021
05edbf1
b36299e
195d12f
a69466f
eb71298
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,278 @@ | ||
| // Copyright 2026 The go-github AUTHORS. All rights reserved. | ||
| // | ||
| // Use of this source code is governed by a BSD-style | ||
| // license that can be found in the LICENSE file. | ||
|
|
||
| package github | ||
|
|
||
| import ( | ||
| "context" | ||
| "fmt" | ||
| "time" | ||
| ) | ||
|
|
||
| // AgentTasksService handles communication with the agent tasks | ||
| // methods of the GitHub API. | ||
| // | ||
| // GitHub API docs: https://docs.github.com/rest/agent-tasks/agent-tasks?apiVersion=2022-11-28 | ||
| type AgentTasksService service | ||
|
|
||
| // AgentTask represents a Copilot cloud agent task. | ||
| type AgentTask struct { | ||
|
danyalahmed1995 marked this conversation as resolved.
|
||
| ID string `json:"id"` | ||
| URL *string `json:"url,omitempty"` | ||
| HTMLURL *string `json:"html_url,omitempty"` | ||
| Name *string `json:"name,omitempty"` | ||
| Creator *AgentTaskCreator `json:"creator,omitempty"` | ||
| CreatorType *string `json:"creator_type,omitempty"` | ||
| // UserCollaborators are the user objects of collaborators on this task. | ||
| // | ||
| // Deprecated: This field is deprecated by the GitHub API. | ||
| UserCollaborators []*User `json:"user_collaborators,omitempty"` | ||
| Owner *AgentTaskOwner `json:"owner,omitempty"` | ||
| Repository *AgentTaskRepository `json:"repository,omitempty"` | ||
| State string `json:"state"` | ||
| SessionCount *int `json:"session_count,omitempty"` | ||
| Artifacts []*AgentTaskArtifact `json:"artifacts,omitempty"` | ||
| ArchivedAt *Timestamp `json:"archived_at,omitempty"` | ||
| CreatedAt Timestamp `json:"created_at"` | ||
| UpdatedAt *Timestamp `json:"updated_at,omitempty"` | ||
| Sessions []*AgentTaskSession `json:"sessions,omitempty"` | ||
| } | ||
|
|
||
| // AgentTaskCreator represents an agent task creator. | ||
| type AgentTaskCreator struct { | ||
| ID *int64 `json:"id,omitempty"` | ||
| } | ||
|
|
||
| // AgentTaskOwner represents an agent task owner. | ||
| type AgentTaskOwner struct { | ||
| ID *int64 `json:"id,omitempty"` | ||
| } | ||
|
|
||
| // AgentTaskRepository represents an agent task repository. | ||
| type AgentTaskRepository struct { | ||
| ID *int64 `json:"id,omitempty"` | ||
| } | ||
|
|
||
| // AgentTaskArtifact represents an artifact produced by an agent task. | ||
| type AgentTaskArtifact struct { | ||
| Provider string `json:"provider"` | ||
| Type string `json:"type"` | ||
| Data *AgentTaskArtifactData `json:"data,omitempty"` | ||
| } | ||
|
|
||
| // AgentTaskArtifactData represents data associated with an agent task artifact. | ||
| type AgentTaskArtifactData struct { | ||
| ID *int64 `json:"id,omitempty"` | ||
| GlobalID *string `json:"global_id,omitempty"` | ||
| HeadRef *string `json:"head_ref,omitempty"` | ||
| BaseRef *string `json:"base_ref,omitempty"` | ||
| } | ||
|
|
||
| // AgentTaskSession represents a session associated with an agent task. | ||
| type AgentTaskSession struct { | ||
| ID string `json:"id"` | ||
| Name *string `json:"name,omitempty"` | ||
| User *User `json:"user,omitempty"` | ||
| Owner *AgentTaskOwner `json:"owner,omitempty"` | ||
| Repository *AgentTaskRepository `json:"repository,omitempty"` | ||
| TaskID *string `json:"task_id,omitempty"` | ||
| State string `json:"state"` | ||
| CreatedAt Timestamp `json:"created_at"` | ||
| UpdatedAt *Timestamp `json:"updated_at,omitempty"` | ||
| CompletedAt *Timestamp `json:"completed_at,omitempty"` | ||
| Prompt *string `json:"prompt,omitempty"` | ||
| HeadRef *string `json:"head_ref,omitempty"` | ||
| BaseRef *string `json:"base_ref,omitempty"` | ||
| Model *string `json:"model,omitempty"` | ||
| Error *AgentTaskSessionError `json:"error,omitempty"` | ||
| } | ||
|
|
||
| // AgentTaskSessionError represents error details for a failed agent task session. | ||
| type AgentTaskSessionError struct { | ||
| Message *string `json:"message,omitempty"` | ||
| } | ||
|
|
||
| // AgentTaskList represents a list of agent tasks. | ||
| type AgentTaskList struct { | ||
| Tasks []*AgentTask `json:"tasks"` | ||
| TotalActiveCount *int `json:"total_active_count,omitempty"` | ||
| TotalArchivedCount *int `json:"total_archived_count,omitempty"` | ||
| } | ||
|
danyalahmed1995 marked this conversation as resolved.
|
||
|
|
||
| // AgentTaskListOptions specifies optional parameters to AgentTasksService.List. | ||
| type AgentTaskListOptions struct { | ||
| // Sort specifies the field to sort results by. Possible values are: updated_at, created_at. | ||
| Sort string `url:"sort,omitempty"` | ||
|
|
||
| // Direction specifies the direction to sort results by. Possible values are: asc, desc. | ||
| Direction string `url:"direction,omitempty"` | ||
|
|
||
| // State is a comma-separated list of task states to filter by. | ||
|
danyalahmed1995 marked this conversation as resolved.
|
||
| // Possible values are: queued, in_progress, completed, failed, idle, | ||
| // waiting_for_user, timed_out, cancelled. | ||
| State string `url:"state,omitempty"` | ||
|
|
||
| // IsArchived filters tasks by archived status. | ||
| IsArchived bool `url:"is_archived,omitempty"` | ||
|
|
||
| // Since filters tasks updated at or after this time. | ||
| Since *time.Time `url:"since,omitempty"` | ||
|
|
||
| ListOptions | ||
| } | ||
|
|
||
| // AgentTaskListByRepoOptions specifies optional parameters to AgentTasksService.ListByRepo. | ||
| type AgentTaskListByRepoOptions struct { | ||
| AgentTaskListOptions | ||
|
|
||
| // CreatorID filters tasks by creator user IDs. | ||
| CreatorID []int64 `url:"creator_id,omitempty"` | ||
| } | ||
|
|
||
| // CreateAgentTaskRequest represents the parameters for creating an agent task. | ||
| type CreateAgentTaskRequest struct { | ||
| // Prompt is the user's prompt for the agent. | ||
| Prompt string `json:"prompt"` | ||
|
|
||
| // Model is the model to use for this task. | ||
| Model *string `json:"model,omitempty"` | ||
|
|
||
| // CreatePullRequest indicates whether to create a pull request. | ||
| CreatePullRequest *bool `json:"create_pull_request,omitempty"` | ||
|
|
||
| // HeadRef is the head ref for the new branch or pull request. | ||
| HeadRef *string `json:"head_ref,omitempty"` | ||
|
|
||
| // BaseRef is the base ref for the new branch or pull request. | ||
| BaseRef *string `json:"base_ref,omitempty"` | ||
| } | ||
|
danyalahmed1995 marked this conversation as resolved.
|
||
|
|
||
| // ListByRepo lists tasks for a repository. | ||
| // | ||
| // Note: This endpoint is in public preview and is subject to change. | ||
| // | ||
| // GitHub API docs: https://docs.github.com/rest/agent-tasks/agent-tasks?apiVersion=2022-11-28#list-tasks-for-repository | ||
| // | ||
| //meta:operation GET /agents/repos/{owner}/{repo}/tasks | ||
| func (s *AgentTasksService) ListByRepo(ctx context.Context, owner, repo string, opts *AgentTaskListByRepoOptions) (*AgentTaskList, *Response, error) { | ||
| u := fmt.Sprintf("agents/repos/%v/%v/tasks", owner, repo) | ||
| u, err := addOptions(u, opts) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
|
|
||
| req, err := s.client.NewRequest(ctx, "GET", u, nil) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
|
|
||
| var tasks *AgentTaskList | ||
| resp, err := s.client.Do(req, &tasks) | ||
| if err != nil { | ||
| return nil, resp, err | ||
| } | ||
|
|
||
| return tasks, resp, nil | ||
| } | ||
|
|
||
| // Create starts a new Copilot cloud agent task for a repository. | ||
| // | ||
| // Note: This endpoint is in public preview and is subject to change. | ||
| // | ||
| // GitHub API docs: https://docs.github.com/rest/agent-tasks/agent-tasks?apiVersion=2022-11-28#start-a-task | ||
| // | ||
| //meta:operation POST /agents/repos/{owner}/{repo}/tasks | ||
| func (s *AgentTasksService) Create(ctx context.Context, owner, repo string, req CreateAgentTaskRequest) (*AgentTask, *Response, error) { | ||
| u := fmt.Sprintf("agents/repos/%v/%v/tasks", owner, repo) | ||
|
|
||
| request, err := s.client.NewRequest(ctx, "POST", u, req) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
|
|
||
| var task *AgentTask | ||
| resp, err := s.client.Do(request, &task) | ||
| if err != nil { | ||
| return nil, resp, err | ||
| } | ||
|
|
||
| return task, resp, nil | ||
| } | ||
|
|
||
| // GetByRepoAndID gets a repository task by ID. | ||
| // | ||
| // Note: This endpoint is in public preview and is subject to change. | ||
| // | ||
| // GitHub API docs: https://docs.github.com/rest/agent-tasks/agent-tasks?apiVersion=2022-11-28#get-a-task-by-repo | ||
| // | ||
| //meta:operation GET /agents/repos/{owner}/{repo}/tasks/{task_id} | ||
| func (s *AgentTasksService) GetByRepoAndID(ctx context.Context, owner, repo, taskID string) (*AgentTask, *Response, error) { | ||
|
Contributor
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. The session schema for this endpoint and for error:
type: object
description: Error details for a failed session
properties:
message:
type: stringThe
Contributor
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Added |
||
| u := fmt.Sprintf("agents/repos/%v/%v/tasks/%v", owner, repo, taskID) | ||
|
|
||
| req, err := s.client.NewRequest(ctx, "GET", u, nil) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
|
|
||
| var task *AgentTask | ||
| resp, err := s.client.Do(req, &task) | ||
| if err != nil { | ||
| return nil, resp, err | ||
| } | ||
|
|
||
| return task, resp, nil | ||
| } | ||
|
|
||
| // List lists tasks for the authenticated user. | ||
| // | ||
| // Note: This endpoint is in public preview and is subject to change. | ||
| // | ||
| // GitHub API docs: https://docs.github.com/rest/agent-tasks/agent-tasks?apiVersion=2022-11-28#list-tasks | ||
| // | ||
| //meta:operation GET /agents/tasks | ||
| func (s *AgentTasksService) List(ctx context.Context, opts *AgentTaskListOptions) (*AgentTaskList, *Response, error) { | ||
| u := "agents/tasks" | ||
| u, err := addOptions(u, opts) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
|
|
||
| req, err := s.client.NewRequest(ctx, "GET", u, nil) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
|
|
||
| var tasks *AgentTaskList | ||
| resp, err := s.client.Do(req, &tasks) | ||
| if err != nil { | ||
| return nil, resp, err | ||
| } | ||
|
|
||
| return tasks, resp, nil | ||
| } | ||
|
|
||
| // Get gets a task by ID for the authenticated user. | ||
| // | ||
| // Note: This endpoint is in public preview and is subject to change. | ||
| // | ||
| // GitHub API docs: https://docs.github.com/rest/agent-tasks/agent-tasks?apiVersion=2022-11-28#get-a-task-by-id | ||
| // | ||
| //meta:operation GET /agents/tasks/{task_id} | ||
| func (s *AgentTasksService) Get(ctx context.Context, taskID string) (*AgentTask, *Response, error) { | ||
| u := fmt.Sprintf("agents/tasks/%v", taskID) | ||
|
|
||
| req, err := s.client.NewRequest(ctx, "GET", u, nil) | ||
| if err != nil { | ||
| return nil, nil, err | ||
| } | ||
|
|
||
| var task *AgentTask | ||
| resp, err := s.client.Do(req, &task) | ||
| if err != nil { | ||
| return nil, resp, err | ||
| } | ||
|
|
||
| return task, resp, nil | ||
| } | ||
Uh oh!
There was an error while loading. Please reload this page.