105 lines
3.4 KiB
TypeScript
105 lines
3.4 KiB
TypeScript
import { useContextKey } from '@kevisual/context'
|
||
import { createSkill, type QueryRouterServer, tool, type QueryRouter, type Skill } from './route.ts'
|
||
import { type App } from './app.ts'
|
||
import { type Plugin } from "@opencode-ai/plugin"
|
||
|
||
import { filter } from '@kevisual/js-filter';
|
||
export const addCallFn = (app: QueryRouterServer) => {
|
||
app.route({
|
||
path: 'call',
|
||
key: '',
|
||
description: '调用',
|
||
middleware: ['auth'],
|
||
metadata: {
|
||
tags: ['opencode'],
|
||
...createSkill({
|
||
skill: 'call-app',
|
||
title: '调用app应用',
|
||
summary: '调用router的应用, 参数path, key, payload',
|
||
args: {
|
||
path: tool.schema.string().describe('应用路径，例如 cnb'),
|
||
key: tool.schema.string().optional().describe('应用key，例如 list-repos'),
|
||
payload: tool.schema.object({}).optional().describe('调用参数'),
|
||
}
|
||
})
|
||
},
|
||
}).define(async (ctx) => {
|
||
const { path, key = '' } = ctx.query;
|
||
if (!path) {
|
||
ctx.throw('路径path不能为空');
|
||
}
|
||
const res = await ctx.run({ path, key, payload: ctx.query.payload || {} }, {
|
||
...ctx
|
||
});
|
||
ctx.forward(res);
|
||
}).addTo(app)
|
||
}
|
||
export const createRouterAgentPluginFn = (opts?: {
|
||
router?: QueryRouter,
|
||
//** 过滤比如，WHERE metadata.tags includes 'opencode' */
|
||
query?: string
|
||
}) => {
|
||
let router = opts?.router
|
||
if (!router) {
|
||
const app = useContextKey<App>('app')
|
||
router = app.router
|
||
}
|
||
if (!router) {
|
||
throw new Error('Router 参数缺失')
|
||
}
|
||
if (!router.hasRoute('call', '')) {
|
||
addCallFn(router as QueryRouterServer)
|
||
}
|
||
const _routes = filter(router.routes, opts?.query || '')
|
||
const routes = _routes.filter(r => {
|
||
const metadata = r.metadata as Skill
|
||
if (metadata && metadata.tags && metadata.tags.includes('opencode')) {
|
||
return !!metadata.skill
|
||
}
|
||
return false
|
||
})
|
||
// opencode run "查看系统信息"
|
||
const AgentPlugin: Plugin = async ({ project, client, $, directory, worktree }) => {
|
||
return {
|
||
'tool': {
|
||
...routes.reduce((acc, route) => {
|
||
const metadata = route.metadata as Skill
|
||
acc[metadata.skill!] = {
|
||
name: metadata.title || metadata.skill,
|
||
description: metadata.summary || '',
|
||
args: metadata.args || {},
|
||
async execute(args: Record<string, any>) {
|
||
const res = await router.run({
|
||
path: route.path,
|
||
key: route.key,
|
||
payload: args
|
||
},
|
||
{ appId: router.appId! });
|
||
if (res.code === 200) {
|
||
if (res.data?.content) {
|
||
return res.data.content;
|
||
}
|
||
if (res.data?.final) {
|
||
return '调用程序成功';
|
||
}
|
||
const str = JSON.stringify(res.data || res, null, 2);
|
||
if (str.length > 10000) {
|
||
return str.slice(0, 10000) + '... (truncated)';
|
||
}
|
||
return str;
|
||
}
|
||
return `Error: ${res?.message || '无法获取结果'}`;
|
||
}
|
||
}
|
||
return acc;
|
||
}, {} as Record<string, any>)
|
||
},
|
||
'tool.execute.before': async (opts) => {
|
||
// console.log('CnbPlugin: tool.execute.before', opts.tool);
|
||
// delete toolSkills['cnb-login-verify']
|
||
}
|
||
}
|
||
}
|
||
return AgentPlugin
|
||
}
|