875c7f9b91
- Add full Telegram bot functionality with Z.AI API integration
- Implement 4 tools: Bash, FileEdit, WebSearch, Git
- Add 3 agents: Code Reviewer, Architect, DevOps Engineer
- Add 6 skills for common coding tasks
- Add systemd service file for 24/7 operation
- Add nginx configuration for HTTPS webhook
- Add comprehensive documentation
- Implement WebSocket server for real-time updates
- Add logging system with Winston
- Add environment validation
🤖 zCode CLI X - Agentic coder with Z.AI + Telegram integration
73 lines
2.7 KiB
TypeScript
73 lines
2.7 KiB
TypeScript
import { transformJSONSchema } from '../..//lib/transform-json-schema';
|
|
import type { infer as zodInfer, ZodType } from 'zod';
|
|
import * as z from 'zod/v4';
|
|
import { AnthropicError } from '../../core/error';
|
|
import { AutoParseableBetaOutputFormat } from '../../lib/beta-parser';
|
|
import { BetaRunnableTool, Promisable } from '../../lib/tools/BetaRunnableTool';
|
|
import { BetaToolResultContentBlockParam } from '../../resources/beta';
|
|
/**
|
|
* Creates a JSON schema output format object from the given Zod schema.
|
|
*
|
|
* If this is passed to the `.parse()` method then the response message will contain a
|
|
* `.parsed_output` property that is the result of parsing the content with the given Zod object.
|
|
*
|
|
* This can be passed directly to the `.create()` method but will not
|
|
* result in any automatic parsing, you'll have to parse the response yourself.
|
|
*/
|
|
export function betaZodOutputFormat<ZodInput extends ZodType>(
|
|
zodObject: ZodInput,
|
|
): AutoParseableBetaOutputFormat<zodInfer<ZodInput>> {
|
|
let jsonSchema = z.toJSONSchema(zodObject, { reused: 'ref' });
|
|
|
|
jsonSchema = transformJSONSchema(jsonSchema);
|
|
|
|
return {
|
|
type: 'json_schema',
|
|
schema: {
|
|
...jsonSchema,
|
|
},
|
|
parse: (content) => {
|
|
const output = zodObject.safeParse(JSON.parse(content));
|
|
|
|
if (!output.success) {
|
|
throw new AnthropicError(
|
|
`Failed to parse structured output: ${output.error.message} cause: ${output.error.issues}`,
|
|
);
|
|
}
|
|
|
|
return output.data;
|
|
},
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Creates a tool using the provided Zod schema that can be passed
|
|
* into the `.toolRunner()` method. The Zod schema will automatically be
|
|
* converted into JSON Schema when passed to the API. The provided function's
|
|
* input arguments will also be validated against the provided schema.
|
|
*/
|
|
export function betaZodTool<InputSchema extends ZodType>(options: {
|
|
name: string;
|
|
inputSchema: InputSchema;
|
|
description: string;
|
|
run: (args: zodInfer<InputSchema>) => Promisable<string | Array<BetaToolResultContentBlockParam>>;
|
|
}): BetaRunnableTool<zodInfer<InputSchema>> {
|
|
const jsonSchema = z.toJSONSchema(options.inputSchema, { reused: 'ref' });
|
|
|
|
if (jsonSchema.type !== 'object') {
|
|
throw new Error(`Zod schema for tool "${options.name}" must be an object, but got ${jsonSchema.type}`);
|
|
}
|
|
|
|
// TypeScript doesn't narrow the type after the runtime check, so we need to assert it
|
|
const objectSchema = jsonSchema as typeof jsonSchema & { type: 'object' };
|
|
|
|
return {
|
|
type: 'custom',
|
|
name: options.name,
|
|
input_schema: objectSchema,
|
|
description: options.description,
|
|
run: options.run,
|
|
parse: (args: unknown) => options.inputSchema.parse(args) as zodInfer<InputSchema>,
|
|
};
|
|
}
|