Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commit59c02e7

Browse files
committed
feat: add task MCP tools
1 parent7f56212 commit59c02e7

File tree

3 files changed

+628
-3
lines changed

3 files changed

+628
-3
lines changed

‎coderd/database/dbfake/dbfake.go‎

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ import (
2424
"github.com/coder/coder/v2/coderd/rbac"
2525
"github.com/coder/coder/v2/coderd/telemetry"
2626
"github.com/coder/coder/v2/coderd/wspubsub"
27+
"github.com/coder/coder/v2/codersdk"
2728
"github.com/coder/coder/v2/provisionersdk"
2829
sdkproto"github.com/coder/coder/v2/provisionersdk/proto"
2930
)
@@ -55,6 +56,7 @@ type WorkspaceBuildBuilder struct {
5556
params []database.WorkspaceBuildParameter
5657
agentTokenstring
5758
dispoworkspaceBuildDisposition
59+
taskAppID uuid.UUID
5860
}
5961

6062
typeworkspaceBuildDispositionstruct {
@@ -117,6 +119,23 @@ func (b WorkspaceBuildBuilder) WithAgent(mutations ...func([]*sdkproto.Agent) []
117119
returnb
118120
}
119121

122+
func (bWorkspaceBuildBuilder)WithTask()WorkspaceBuildBuilder {
123+
//nolint: revive // returns modified struct
124+
b.taskAppID=uuid.New()
125+
returnb.Params(database.WorkspaceBuildParameter{
126+
Name:codersdk.AITaskPromptParameterName,
127+
Value:"list me",
128+
}).WithAgent(func(a []*sdkproto.Agent) []*sdkproto.Agent {
129+
a[0].Apps= []*sdkproto.App{
130+
{
131+
Id:b.taskAppID.String(),
132+
Slug:"vcode",
133+
},
134+
}
135+
returna
136+
})
137+
}
138+
120139
func (bWorkspaceBuildBuilder)Starting()WorkspaceBuildBuilder {
121140
//nolint: revive // returns modified struct
122141
b.dispo.starting=true
@@ -134,6 +153,14 @@ func (b WorkspaceBuildBuilder) Do() WorkspaceResponse {
134153
b.seed.ID=uuid.New()
135154
b.seed.JobID=jobID
136155

156+
ifb.taskAppID!=uuid.Nil {
157+
b.seed.HasAITask= sql.NullBool{
158+
Bool:true,
159+
Valid:true,
160+
}
161+
b.seed.AITaskSidebarAppID= uuid.NullUUID{UUID:b.taskAppID,Valid:true}
162+
}
163+
137164
resp:=WorkspaceResponse{
138165
AgentToken:b.agentToken,
139166
}

‎codersdk/toolsdk/toolsdk.go‎

Lines changed: 245 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,10 @@ const (
5050
ToolNameWorkspaceEditFile="coder_workspace_edit_file"
5151
ToolNameWorkspaceEditFiles="coder_workspace_edit_files"
5252
ToolNameWorkspacePortForward="coder_workspace_port_forward"
53+
ToolNameCreateTask="coder_create_task"
54+
ToolNameDeleteTask="coder_delete_task"
55+
ToolNameListTasks="coder_list_tasks"
56+
ToolNameGetTaskStatus="coder_get_task_status"
5357
)
5458

5559
funcNewDeps(client*codersdk.Client,opts...func(*Deps)) (Deps,error) {
@@ -223,6 +227,10 @@ var All = []GenericTool{
223227
WorkspaceEditFile.Generic(),
224228
WorkspaceEditFiles.Generic(),
225229
WorkspacePortForward.Generic(),
230+
CreateTask.Generic(),
231+
DeleteTask.Generic(),
232+
ListTasks.Generic(),
233+
GetTaskStatus.Generic(),
226234
}
227235

228236
typeReportTaskArgsstruct {
@@ -344,7 +352,7 @@ is provisioned correctly and the agent can connect to the control plane.
344352
Properties:map[string]any{
345353
"user":map[string]any{
346354
"type":"string",
347-
"description":"Username or ID of the user tocreatethe workspace for. Use the `me` keyword to createa workspace for the authenticated user.",
355+
"description":userDescription("create a workspace"),
348356
},
349357
"template_version_id":map[string]any{
350358
"type":"string",
@@ -1393,8 +1401,6 @@ type WorkspaceLSResponse struct {
13931401
Contents []WorkspaceLSFile`json:"contents"`
13941402
}
13951403

1396-
constworkspaceDescription="The workspace name in the format [owner/]workspace[.agent]. If an owner is not specified, the authenticated user is used."
1397-
13981404
varWorkspaceLS=Tool[WorkspaceLSArgs,WorkspaceLSResponse]{
13991405
Tool: aisdk.Tool{
14001406
Name:ToolNameWorkspaceLS,
@@ -1750,6 +1756,232 @@ var WorkspacePortForward = Tool[WorkspacePortForwardArgs, WorkspacePortForwardRe
17501756
},
17511757
}
17521758

1759+
typeCreateTaskArgsstruct {
1760+
Promptstring`json:"prompt"`
1761+
TemplateVersionIDstring`json:"template_version_id"`
1762+
TemplateVersionPresetIDstring`json:"template_version_preset_id"`
1763+
Userstring`json:"user"`
1764+
}
1765+
1766+
varCreateTask=Tool[CreateTaskArgs, codersdk.Task]{
1767+
Tool: aisdk.Tool{
1768+
Name:ToolNameCreateTask,
1769+
Description:`Create a task.`,
1770+
Schema: aisdk.Schema{
1771+
Properties:map[string]any{
1772+
"prompt":map[string]any{
1773+
"type":"string",
1774+
"description":"Prompt for the task.",
1775+
},
1776+
"template_version_id":map[string]any{
1777+
"type":"string",
1778+
"description":"ID of the template version to create the task from.",
1779+
},
1780+
"template_preset_version_id":map[string]any{
1781+
"type":"string",
1782+
"description":"Optional ID of the template version preset to create the task from.",
1783+
},
1784+
"user":map[string]any{
1785+
"type":"string",
1786+
"description":userDescription("create a task"),
1787+
},
1788+
},
1789+
Required: []string{"prompt","template_version_id"},
1790+
},
1791+
},
1792+
UserClientOptional:true,
1793+
Handler:func(ctx context.Context,depsDeps,argsCreateTaskArgs) (codersdk.Task,error) {
1794+
ifargs.Prompt=="" {
1795+
return codersdk.Task{},xerrors.New("a prompt is required")
1796+
}
1797+
1798+
tvID,err:=uuid.Parse(args.TemplateVersionID)
1799+
iferr!=nil {
1800+
return codersdk.Task{},xerrors.New("template_version_id must be a valid UUID")
1801+
}
1802+
1803+
vartvPresetID uuid.UUID
1804+
ifargs.TemplateVersionPresetID!="" {
1805+
tvPresetID,err=uuid.Parse(args.TemplateVersionPresetID)
1806+
iferr!=nil {
1807+
return codersdk.Task{},xerrors.New("template_version_preset_id must be a valid UUID")
1808+
}
1809+
}
1810+
1811+
ifargs.User=="" {
1812+
args.User=codersdk.Me
1813+
}
1814+
1815+
expClient:=codersdk.NewExperimentalClient(deps.coderClient)
1816+
task,err:=expClient.CreateTask(ctx,args.User, codersdk.CreateTaskRequest{
1817+
Prompt:args.Prompt,
1818+
TemplateVersionID:tvID,
1819+
TemplateVersionPresetID:tvPresetID,
1820+
})
1821+
iferr!=nil {
1822+
return codersdk.Task{},xerrors.Errorf("create task: %w",err)
1823+
}
1824+
1825+
returntask,nil
1826+
},
1827+
}
1828+
1829+
typeDeleteTaskArgsstruct {
1830+
TaskIDstring`json:"task_id"`
1831+
}
1832+
1833+
varDeleteTask=Tool[DeleteTaskArgs, codersdk.Response]{
1834+
Tool: aisdk.Tool{
1835+
Name:ToolNameDeleteTask,
1836+
Description:`Delete a task.`,
1837+
Schema: aisdk.Schema{
1838+
Properties:map[string]any{
1839+
"task_id":map[string]any{
1840+
"type":"string",
1841+
"description":taskIDDescription("delete"),
1842+
},
1843+
},
1844+
Required: []string{"task_id"},
1845+
},
1846+
},
1847+
UserClientOptional:true,
1848+
Handler:func(ctx context.Context,depsDeps,argsDeleteTaskArgs) (codersdk.Response,error) {
1849+
ifargs.TaskID=="" {
1850+
return codersdk.Response{},xerrors.New("task_id is required")
1851+
}
1852+
1853+
expClient:=codersdk.NewExperimentalClient(deps.coderClient)
1854+
1855+
varownerstring
1856+
id,err:=uuid.Parse(args.TaskID)
1857+
iferr==nil {
1858+
task,err:=expClient.TaskByID(ctx,id)
1859+
iferr!=nil {
1860+
return codersdk.Response{},xerrors.Errorf("get task %q: %w",args.TaskID,err)
1861+
}
1862+
owner=task.OwnerName
1863+
}else {
1864+
workspaceName:=NormalizeWorkspaceInput(args.TaskID)
1865+
ws,err:=namedWorkspace(ctx,deps.coderClient,workspaceName)
1866+
iferr!=nil {
1867+
return codersdk.Response{},xerrors.Errorf("get task workspace %q: %w",args.TaskID,err)
1868+
}
1869+
owner=ws.OwnerName
1870+
id=ws.ID
1871+
}
1872+
1873+
err=expClient.DeleteTask(ctx,owner,id)
1874+
iferr!=nil {
1875+
return codersdk.Response{},xerrors.Errorf("delete task: %w",err)
1876+
}
1877+
1878+
return codersdk.Response{
1879+
Message:"Task deleted successfully",
1880+
},nil
1881+
},
1882+
}
1883+
1884+
typeListTasksArgsstruct {
1885+
Statusstring`json:"status"`
1886+
Userstring`json:"user"`
1887+
}
1888+
1889+
typeListTasksResponsestruct {
1890+
Tasks []codersdk.Task`json:"tasks"`
1891+
}
1892+
1893+
varListTasks=Tool[ListTasksArgs,ListTasksResponse]{
1894+
Tool: aisdk.Tool{
1895+
Name:ToolNameListTasks,
1896+
Description:`List tasks.`,
1897+
Schema: aisdk.Schema{
1898+
Properties:map[string]any{
1899+
"status":map[string]any{
1900+
"type":"string",
1901+
"description":"Optional filter by task status.",
1902+
},
1903+
"user":map[string]any{
1904+
"type":"string",
1905+
"description":userDescription("list tasks"),
1906+
},
1907+
},
1908+
Required: []string{},
1909+
},
1910+
},
1911+
UserClientOptional:true,
1912+
Handler:func(ctx context.Context,depsDeps,argsListTasksArgs) (ListTasksResponse,error) {
1913+
ifargs.User=="" {
1914+
args.User=codersdk.Me
1915+
}
1916+
1917+
expClient:=codersdk.NewExperimentalClient(deps.coderClient)
1918+
tasks,err:=expClient.Tasks(ctx,&codersdk.TasksFilter{
1919+
Owner:args.User,
1920+
Status:args.Status,
1921+
})
1922+
iferr!=nil {
1923+
returnListTasksResponse{},xerrors.Errorf("list tasks: %w",err)
1924+
}
1925+
1926+
returnListTasksResponse{
1927+
Tasks:tasks,
1928+
},nil
1929+
},
1930+
}
1931+
1932+
typeGetTaskStatusArgsstruct {
1933+
TaskIDstring`json:"task_id"`
1934+
}
1935+
1936+
typeGetTaskStatusResponsestruct {
1937+
Status codersdk.WorkspaceStatus`json:"status"`
1938+
State*codersdk.TaskStateEntry`json:"state"`
1939+
}
1940+
1941+
varGetTaskStatus=Tool[GetTaskStatusArgs,GetTaskStatusResponse]{
1942+
Tool: aisdk.Tool{
1943+
Name:ToolNameGetTaskStatus,
1944+
Description:`Get the status of a task.`,
1945+
Schema: aisdk.Schema{
1946+
Properties:map[string]any{
1947+
"task_id":map[string]any{
1948+
"type":"string",
1949+
"description":taskIDDescription("get"),
1950+
},
1951+
},
1952+
Required: []string{"task_id"},
1953+
},
1954+
},
1955+
UserClientOptional:true,
1956+
Handler:func(ctx context.Context,depsDeps,argsGetTaskStatusArgs) (GetTaskStatusResponse,error) {
1957+
ifargs.TaskID=="" {
1958+
returnGetTaskStatusResponse{},xerrors.New("task_id is required")
1959+
}
1960+
1961+
expClient:=codersdk.NewExperimentalClient(deps.coderClient)
1962+
1963+
id,err:=uuid.Parse(args.TaskID)
1964+
iferr!=nil {
1965+
workspaceName:=NormalizeWorkspaceInput(args.TaskID)
1966+
ws,err:=namedWorkspace(ctx,deps.coderClient,workspaceName)
1967+
iferr!=nil {
1968+
returnGetTaskStatusResponse{},xerrors.Errorf("get task workspace %q: %w",args.TaskID,err)
1969+
}
1970+
id=ws.ID
1971+
}
1972+
1973+
task,err:=expClient.TaskByID(ctx,id)
1974+
iferr!=nil {
1975+
returnGetTaskStatusResponse{},xerrors.Errorf("get task %q: %w",args.TaskID,err)
1976+
}
1977+
1978+
returnGetTaskStatusResponse{
1979+
Status:task.Status,
1980+
State:task.CurrentState,
1981+
},nil
1982+
},
1983+
}
1984+
17531985
// NormalizeWorkspaceInput converts workspace name input to standard format.
17541986
// Handles the following input formats:
17551987
// - workspace → workspace
@@ -1810,3 +2042,13 @@ func newAgentConn(ctx context.Context, client *codersdk.Client, workspace string
18102042
}
18112043
returnconn,nil
18122044
}
2045+
2046+
constworkspaceDescription="The workspace name in the format [owner/]workspace[.agent]. If an owner is not specified, the authenticated user is used."
2047+
2048+
functaskIDDescription(actionstring)string {
2049+
returnfmt.Sprintf("ID or workspace identifier in the format [owner/]workspace[.agent] for the task to %s. If an owner is not specified, the authenticated user is used.",action)
2050+
}
2051+
2052+
funcuserDescription(actionstring)string {
2053+
returnfmt.Sprintf("Username or ID of the user for which to %s. Omit or use the `me` keyword to %s for the authenticated user.",action,action)
2054+
}

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp