init claude-code
This commit is contained in:
@@ -0,0 +1,161 @@
|
||||
import axios from 'axios'
|
||||
import { z } from 'zod/v4'
|
||||
import { getOauthConfig } from '../../constants/oauth.js'
|
||||
import { getFeatureValue_CACHED_MAY_BE_STALE } from '../../services/analytics/growthbook.js'
|
||||
import { getOrganizationUUID } from '../../services/oauth/client.js'
|
||||
import { isPolicyAllowed } from '../../services/policyLimits/index.js'
|
||||
import type { ToolUseContext } from '../../Tool.js'
|
||||
import { buildTool, type ToolDef } from '../../Tool.js'
|
||||
import {
|
||||
checkAndRefreshOAuthTokenIfNeeded,
|
||||
getClaudeAIOAuthTokens,
|
||||
} from '../../utils/auth.js'
|
||||
import { lazySchema } from '../../utils/lazySchema.js'
|
||||
import { jsonStringify } from '../../utils/slowOperations.js'
|
||||
import { DESCRIPTION, PROMPT, REMOTE_TRIGGER_TOOL_NAME } from './prompt.js'
|
||||
import { renderToolResultMessage, renderToolUseMessage } from './UI.js'
|
||||
|
||||
const inputSchema = lazySchema(() =>
|
||||
z.strictObject({
|
||||
action: z.enum(['list', 'get', 'create', 'update', 'run']),
|
||||
trigger_id: z
|
||||
.string()
|
||||
.regex(/^[\w-]+$/)
|
||||
.optional()
|
||||
.describe('Required for get, update, and run'),
|
||||
body: z
|
||||
.record(z.string(), z.unknown())
|
||||
.optional()
|
||||
.describe('JSON body for create and update'),
|
||||
}),
|
||||
)
|
||||
type InputSchema = ReturnType<typeof inputSchema>
|
||||
export type Input = z.infer<InputSchema>
|
||||
|
||||
const outputSchema = lazySchema(() =>
|
||||
z.object({
|
||||
status: z.number(),
|
||||
json: z.string(),
|
||||
}),
|
||||
)
|
||||
type OutputSchema = ReturnType<typeof outputSchema>
|
||||
export type Output = z.infer<OutputSchema>
|
||||
|
||||
const TRIGGERS_BETA = 'ccr-triggers-2026-01-30'
|
||||
|
||||
export const RemoteTriggerTool = buildTool({
|
||||
name: REMOTE_TRIGGER_TOOL_NAME,
|
||||
searchHint: 'manage scheduled remote agent triggers',
|
||||
maxResultSizeChars: 100_000,
|
||||
shouldDefer: true,
|
||||
get inputSchema(): InputSchema {
|
||||
return inputSchema()
|
||||
},
|
||||
get outputSchema(): OutputSchema {
|
||||
return outputSchema()
|
||||
},
|
||||
isEnabled() {
|
||||
return (
|
||||
getFeatureValue_CACHED_MAY_BE_STALE('tengu_surreal_dali', false) &&
|
||||
isPolicyAllowed('allow_remote_sessions')
|
||||
)
|
||||
},
|
||||
isConcurrencySafe() {
|
||||
return true
|
||||
},
|
||||
isReadOnly(input: Input) {
|
||||
return input.action === 'list' || input.action === 'get'
|
||||
},
|
||||
toAutoClassifierInput(input: Input) {
|
||||
return `RemoteTrigger ${input.action}${input.trigger_id ? ` ${input.trigger_id}` : ''}`
|
||||
},
|
||||
async description() {
|
||||
return DESCRIPTION
|
||||
},
|
||||
async prompt() {
|
||||
return PROMPT
|
||||
},
|
||||
async call(input: Input, context: ToolUseContext) {
|
||||
await checkAndRefreshOAuthTokenIfNeeded()
|
||||
const accessToken = getClaudeAIOAuthTokens()?.accessToken
|
||||
if (!accessToken) {
|
||||
throw new Error(
|
||||
'Not authenticated with a claude.ai account. Run /login and try again.',
|
||||
)
|
||||
}
|
||||
const orgUUID = await getOrganizationUUID()
|
||||
if (!orgUUID) {
|
||||
throw new Error('Unable to resolve organization UUID.')
|
||||
}
|
||||
|
||||
const base = `${getOauthConfig().BASE_API_URL}/v1/code/triggers`
|
||||
const headers = {
|
||||
Authorization: `Bearer ${accessToken}`,
|
||||
'Content-Type': 'application/json',
|
||||
'anthropic-version': '2023-06-01',
|
||||
'anthropic-beta': TRIGGERS_BETA,
|
||||
'x-organization-uuid': orgUUID,
|
||||
}
|
||||
|
||||
const { action, trigger_id, body } = input
|
||||
let method: 'GET' | 'POST'
|
||||
let url: string
|
||||
let data: unknown
|
||||
switch (action) {
|
||||
case 'list':
|
||||
method = 'GET'
|
||||
url = base
|
||||
break
|
||||
case 'get':
|
||||
if (!trigger_id) throw new Error('get requires trigger_id')
|
||||
method = 'GET'
|
||||
url = `${base}/${trigger_id}`
|
||||
break
|
||||
case 'create':
|
||||
if (!body) throw new Error('create requires body')
|
||||
method = 'POST'
|
||||
url = base
|
||||
data = body
|
||||
break
|
||||
case 'update':
|
||||
if (!trigger_id) throw new Error('update requires trigger_id')
|
||||
if (!body) throw new Error('update requires body')
|
||||
method = 'POST'
|
||||
url = `${base}/${trigger_id}`
|
||||
data = body
|
||||
break
|
||||
case 'run':
|
||||
if (!trigger_id) throw new Error('run requires trigger_id')
|
||||
method = 'POST'
|
||||
url = `${base}/${trigger_id}/run`
|
||||
data = {}
|
||||
break
|
||||
}
|
||||
|
||||
const res = await axios.request({
|
||||
method,
|
||||
url,
|
||||
headers,
|
||||
data,
|
||||
timeout: 20_000,
|
||||
signal: context.abortController.signal,
|
||||
validateStatus: () => true,
|
||||
})
|
||||
|
||||
return {
|
||||
data: {
|
||||
status: res.status,
|
||||
json: jsonStringify(res.data),
|
||||
},
|
||||
}
|
||||
},
|
||||
mapToolResultToToolResultBlockParam(output, toolUseID) {
|
||||
return {
|
||||
tool_use_id: toolUseID,
|
||||
type: 'tool_result',
|
||||
content: `HTTP ${output.status}\n${output.json}`,
|
||||
}
|
||||
},
|
||||
renderToolUseMessage,
|
||||
renderToolResultMessage,
|
||||
} satisfies ToolDef<InputSchema, Output>)
|
||||
@@ -0,0 +1,17 @@
|
||||
import React from 'react';
|
||||
import { MessageResponse } from '../../components/MessageResponse.js';
|
||||
import { Text } from '../../ink.js';
|
||||
import { countCharInString } from '../../utils/stringUtils.js';
|
||||
import type { Input, Output } from './RemoteTriggerTool.js';
|
||||
export function renderToolUseMessage(input: Partial<Input>): React.ReactNode {
|
||||
return `${input.action ?? ''}${input.trigger_id ? ` ${input.trigger_id}` : ''}`;
|
||||
}
|
||||
export function renderToolResultMessage(output: Output): React.ReactNode {
|
||||
const lines = countCharInString(output.json, '\n') + 1;
|
||||
return <MessageResponse>
|
||||
<Text>
|
||||
HTTP {output.status} <Text dimColor>({lines} lines)</Text>
|
||||
</Text>
|
||||
</MessageResponse>;
|
||||
}
|
||||
//# sourceMappingURL=data:application/json;charset=utf-8;base64,eyJ2ZXJzaW9uIjozLCJuYW1lcyI6WyJSZWFjdCIsIk1lc3NhZ2VSZXNwb25zZSIsIlRleHQiLCJjb3VudENoYXJJblN0cmluZyIsIklucHV0IiwiT3V0cHV0IiwicmVuZGVyVG9vbFVzZU1lc3NhZ2UiLCJpbnB1dCIsIlBhcnRpYWwiLCJSZWFjdE5vZGUiLCJhY3Rpb24iLCJ0cmlnZ2VyX2lkIiwicmVuZGVyVG9vbFJlc3VsdE1lc3NhZ2UiLCJvdXRwdXQiLCJsaW5lcyIsImpzb24iLCJzdGF0dXMiXSwic291cmNlcyI6WyJVSS50c3giXSwic291cmNlc0NvbnRlbnQiOlsiaW1wb3J0IFJlYWN0IGZyb20gJ3JlYWN0J1xuaW1wb3J0IHsgTWVzc2FnZVJlc3BvbnNlIH0gZnJvbSAnLi4vLi4vY29tcG9uZW50cy9NZXNzYWdlUmVzcG9uc2UuanMnXG5pbXBvcnQgeyBUZXh0IH0gZnJvbSAnLi4vLi4vaW5rLmpzJ1xuaW1wb3J0IHsgY291bnRDaGFySW5TdHJpbmcgfSBmcm9tICcuLi8uLi91dGlscy9zdHJpbmdVdGlscy5qcydcbmltcG9ydCB0eXBlIHsgSW5wdXQsIE91dHB1dCB9IGZyb20gJy4vUmVtb3RlVHJpZ2dlclRvb2wuanMnXG5cbmV4cG9ydCBmdW5jdGlvbiByZW5kZXJUb29sVXNlTWVzc2FnZShpbnB1dDogUGFydGlhbDxJbnB1dD4pOiBSZWFjdC5SZWFjdE5vZGUge1xuICByZXR1cm4gYCR7aW5wdXQuYWN0aW9uID8/ICcnfSR7aW5wdXQudHJpZ2dlcl9pZCA/IGAgJHtpbnB1dC50cmlnZ2VyX2lkfWAgOiAnJ31gXG59XG5cbmV4cG9ydCBmdW5jdGlvbiByZW5kZXJUb29sUmVzdWx0TWVzc2FnZShvdXRwdXQ6IE91dHB1dCk6IFJlYWN0LlJlYWN0Tm9kZSB7XG4gIGNvbnN0IGxpbmVzID0gY291bnRDaGFySW5TdHJpbmcob3V0cHV0Lmpzb24sICdcXG4nKSArIDFcbiAgcmV0dXJuIChcbiAgICA8TWVzc2FnZVJlc3BvbnNlPlxuICAgICAgPFRleHQ+XG4gICAgICAgIEhUVFAge291dHB1dC5zdGF0dXN9IDxUZXh0IGRpbUNvbG9yPih7bGluZXN9IGxpbmVzKTwvVGV4dD5cbiAgICAgIDwvVGV4dD5cbiAgICA8L01lc3NhZ2VSZXNwb25zZT5cbiAgKVxufVxuIl0sIm1hcHBpbmdzIjoiQUFBQSxPQUFPQSxLQUFLLE1BQU0sT0FBTztBQUN6QixTQUFTQyxlQUFlLFFBQVEscUNBQXFDO0FBQ3JFLFNBQVNDLElBQUksUUFBUSxjQUFjO0FBQ25DLFNBQVNDLGlCQUFpQixRQUFRLDRCQUE0QjtBQUM5RCxjQUFjQyxLQUFLLEVBQUVDLE1BQU0sUUFBUSx3QkFBd0I7QUFFM0QsT0FBTyxTQUFTQyxvQkFBb0JBLENBQUNDLEtBQUssRUFBRUMsT0FBTyxDQUFDSixLQUFLLENBQUMsQ0FBQyxFQUFFSixLQUFLLENBQUNTLFNBQVMsQ0FBQztFQUMzRSxPQUFPLEdBQUdGLEtBQUssQ0FBQ0csTUFBTSxJQUFJLEVBQUUsR0FBR0gsS0FBSyxDQUFDSSxVQUFVLEdBQUcsSUFBSUosS0FBSyxDQUFDSSxVQUFVLEVBQUUsR0FBRyxFQUFFLEVBQUU7QUFDakY7QUFFQSxPQUFPLFNBQVNDLHVCQUF1QkEsQ0FBQ0MsTUFBTSxFQUFFUixNQUFNLENBQUMsRUFBRUwsS0FBSyxDQUFDUyxTQUFTLENBQUM7RUFDdkUsTUFBTUssS0FBSyxHQUFHWCxpQkFBaUIsQ0FBQ1UsTUFBTSxDQUFDRSxJQUFJLEVBQUUsSUFBSSxDQUFDLEdBQUcsQ0FBQztFQUN0RCxPQUNFLENBQUMsZUFBZTtBQUNwQixNQUFNLENBQUMsSUFBSTtBQUNYLGFBQWEsQ0FBQ0YsTUFBTSxDQUFDRyxNQUFNLENBQUMsQ0FBQyxDQUFDLElBQUksQ0FBQyxRQUFRLENBQUMsQ0FBQyxDQUFDRixLQUFLLENBQUMsT0FBTyxFQUFFLElBQUk7QUFDakUsTUFBTSxFQUFFLElBQUk7QUFDWixJQUFJLEVBQUUsZUFBZSxDQUFDO0FBRXRCIiwiaWdub3JlTGlzdCI6W119
|
||||
@@ -0,0 +1,15 @@
|
||||
export const REMOTE_TRIGGER_TOOL_NAME = 'RemoteTrigger'
|
||||
|
||||
export const DESCRIPTION =
|
||||
'Manage scheduled remote Claude Code agents (triggers) via the claude.ai CCR API. Auth is handled in-process — the token never reaches the shell.'
|
||||
|
||||
export const PROMPT = `Call the claude.ai remote-trigger API. Use this instead of curl — the OAuth token is added automatically in-process and never exposed.
|
||||
|
||||
Actions:
|
||||
- list: GET /v1/code/triggers
|
||||
- get: GET /v1/code/triggers/{trigger_id}
|
||||
- create: POST /v1/code/triggers (requires body)
|
||||
- update: POST /v1/code/triggers/{trigger_id} (requires body, partial update)
|
||||
- run: POST /v1/code/triggers/{trigger_id}/run
|
||||
|
||||
The response is the raw JSON from the API.`
|
||||
Reference in New Issue
Block a user