Spaces:
Runtime error
Runtime error
File size: 2,960 Bytes
4782147 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 | import z from "zod";
import { ChatMentionSchema } from "./chat";
import { VisibilitySchema } from "./util";
export type AgentIcon = {
type: "emoji";
value: string;
style?: Record<string, string>;
};
export const AgentInstructionsSchema = z.object({
role: z.string().optional(),
systemPrompt: z.string().optional(),
mentions: z.array(ChatMentionSchema).optional(),
});
export const AgentCreateSchema = z
.object({
name: z.string().min(1).max(100),
description: z.string().max(8000).optional(),
icon: z
.object({
type: z.literal("emoji"),
value: z.string(),
style: z.record(z.string(), z.string()).optional(),
})
.optional(),
userId: z.string(),
instructions: AgentInstructionsSchema,
visibility: VisibilitySchema.optional().default("private"),
})
.strip();
export const AgentUpdateSchema = z
.object({
name: z.string().min(1).max(100).optional(),
description: z.string().max(8000).optional(),
icon: z
.object({
type: z.literal("emoji"),
value: z.string(),
style: z.record(z.string(), z.string()).optional(),
})
.optional(),
instructions: AgentInstructionsSchema.optional(),
visibility: VisibilitySchema.optional(),
})
.strip();
export const AgentQuerySchema = z.object({
type: z.enum(["all", "mine", "shared", "bookmarked"]).default("all"),
filters: z.string().optional(),
limit: z.coerce.number().min(1).max(100).default(50),
});
export type AgentVisibility = z.infer<typeof VisibilitySchema>;
export type AgentSummary = {
id: string;
name: string;
description?: string;
icon?: AgentIcon;
userId: string;
visibility: AgentVisibility;
createdAt: Date;
updatedAt: Date;
userName?: string;
userAvatar?: string;
isBookmarked?: boolean;
};
export type Agent = AgentSummary & {
instructions: z.infer<typeof AgentInstructionsSchema>;
};
export type AgentRepository = {
insertAgent(agent: z.infer<typeof AgentCreateSchema>): Promise<Agent>;
selectAgentById(id: string, userId: string): Promise<Agent | null>;
selectAgentsByUserId(userId: string): Promise<Agent[]>;
updateAgent(
id: string,
userId: string,
agent: z.infer<typeof AgentUpdateSchema>,
): Promise<Agent>;
deleteAgent(id: string, userId: string): Promise<void>;
selectAgents(
currentUserId: string,
filters?: ("all" | "mine" | "shared" | "bookmarked")[],
limit?: number,
): Promise<AgentSummary[]>;
checkAccess(
agentId: string,
userId: string,
destructive?: boolean,
): Promise<boolean>;
};
export const AgentGenerateSchema = z.object({
name: z.string().describe("Agent name"),
description: z.string().describe("Agent description"),
instructions: z.string().describe("Agent instructions"),
role: z.string().describe("Agent role"),
tools: z
.array(z.string())
.describe("Agent allowed tools name")
.optional()
.default([]),
});
|