import { z } from "zod";
import BaseSchema from "../schemas/base";
import { zodToJsonSchema } from "../utils";
// Pie chart data schema
const data = z.object({
name: z.string().describe("name of date item."),
value: z.number().describe("value of date item."),
});
const SeriesItemSchema = z.object({
type: z.literal("pie"),
name: z
.string()
.optional()
.describe(
"Series name used for displaying in tooltip and filtering with legend.",
),
label: z
.object({
show: z.boolean().describe("Whether to show label."),
color: z.string().describe("Text color."),
position: z
.enum(["outside", "inside", "center"])
.default("outside")
.describe("The position of label."),
formatter: z.string().describe(`
Data label formatter, which supports string template and callback function. In either form, \n is supported to represent a new line.
String template
Model variation includes:
{a}: series name.
{b}: the name of a data item.
{c}: the value of a data item.
{d}: the percent.
{@xxx}: the value of a dimension named 'xxx', for example, {@product} refers the value of 'product' dimension.
{@[n]}: the value of a dimension at the index of n, for example, {@[3]} refers the value at dimensions[3].
example:
formatter: '{b}: {d}'
`),
})
.default({
show: true,
color: "#000",
position: "outside",
formatter: "{b}: {c} ({d}%)",
})
.describe(
"Text label of pie chart, to explain some data information about graphic item like value, name and so on. ",
),
data: z.array(data).describe("Data array of series"),
// radius: z.array(z.string()).optional().default(["0", "75%"]).describe(`
// Radius of Pie chart. Value can be:
// Array<string>: The first item specifies the inside radius, and the second item specifies the outside radius. Each item follows the definitions above.
// Donut chart can be achieved by setting a inner radius, for example, ["40%", "75%"].
// `),
// itemStyle: z
// .object({
// borderRadius: z
// .number()
// .optional()
// .describe(
// "To specify the roundness for corners of the sectors of the pie chart. It can be either a specific pixel value or a percentage value relative to the radius of the sector.",
// ),
// })
// .optional()
// .default({}),
});
// Pie chart input schema
const schema = {
...BaseSchema,
series: z
.array(SeriesItemSchema)
.describe(
"The pie chart is mainly used for showing proportion of different categories. Each arc length represents the proportion of data quantity.",
),
};
// Pie chart tool descriptor
const tool = {
name: "generate_pie_chart",
description:
"Generate a pie chart preview link(not a chart image) which users can click to show the proportion of parts, such as, market share and budget allocation.",
inputSchema: zodToJsonSchema(schema),
};
export const pie = {
schema,
tool,
};
