@rizom/brain 0.2.0-alpha.145 → 0.2.0-alpha.147

package/dist/plugins.d.ts

@@ -124,6 +124,23 @@ declare const AgentResponseSchema: z.ZodType<AgentResponse, AgentResponse>;
 type MessageRole = "user" | "assistant";
 declare const messageRoleSchema: z.ZodType<MessageRole, MessageRole>;
 
+/**
+ * Query response schemas used throughout the system
+ */
+interface DefaultQuerySource {
+    id: string;
+    type: string;
+    excerpt?: string | undefined;
+    relevance?: number | undefined;
+}
+interface DefaultQueryResponse {
+    message: string;
+    summary?: string | undefined;
+    topics?: string[] | undefined;
+    sources?: DefaultQuerySource[] | undefined;
+    metadata?: Record<string, unknown> | undefined;
+}
+
 /**
  * Options for entity mutation operations (create, update, upsert)
  */
@@ -580,6 +597,8 @@ interface IEntitiesNamespace {
     registerCreateInterceptor(entityType: string, interceptor: CreateInterceptor): void;
     /** Register a raw-upload durable save handler for this plugin's entity type */
     registerUploadSaveHandler(registration: UploadSaveHandlerRegistration): void;
+    /** Look up the registered upload-save handler claiming a media type */
+    getUploadSaveHandler(mediaType: string): UploadSaveHandlerRegistration | undefined;
 }
 interface EmbeddingBackfillResult {
     queued: number;
@@ -755,6 +774,16 @@ declare const entityCountSchema: z.ZodObject<{
     entityType: z.ZodString;
     count: z.ZodNumber;
 }>;
+/**
+ * Content generation configuration - unified config object
+ */
+interface ContentGenerationConfig {
+    prompt: string;
+    templateName: string;
+    conversationHistory?: string;
+    data?: Record<string, unknown>;
+    interfacePermissionGrant?: UserPermissionLevel;
+}
 
 declare const DaemonHealthSchema: z.ZodObject<{
     status: z.ZodEnum<{
@@ -906,6 +935,33 @@ interface MessageContext {
     threadId?: string;
 }
 
+type AIGenerationSchema<T> = ZodType<T>;
+type AspectRatio = "1:1" | "16:9" | "9:16" | "4:3" | "3:4";
+interface ImageGenerationOptions {
+    aspectRatio?: AspectRatio;
+}
+interface ImageGenerationResult {
+    base64: string;
+    dataUrl: string;
+}
+/**
+ * AI namespace for entity plugins — includes generation capabilities
+ */
+interface IEntityAINamespace {
+    /** Query the AI with optional context */
+    query: (prompt: string, context?: Record<string, unknown>) => Promise<DefaultQueryResponse>;
+    /** Generate content using AI with template */
+    generate: <T = unknown>(config: ContentGenerationConfig) => Promise<T>;
+    /** Generate a structured object using AI with a schema parser */
+    generateObject: <T>(prompt: string, schema: AIGenerationSchema<T>) => Promise<{
+        object: T;
+    }>;
+    /** Generate an image using AI (requires AI_API_KEY) */
+    generateImage: (prompt: string, options?: ImageGenerationOptions) => Promise<ImageGenerationResult>;
+    /** Check if image generation is available */
+    canGenerateImages: () => boolean;
+}
+
 type PluginConfig = Record<string, unknown>;
 type PluginConfigInput<T extends {
     _input: unknown;
@@ -1198,6 +1254,7 @@ interface ServicePluginContext extends BasePluginContext {
     readonly templates: IServiceTemplatesNamespace;
     readonly views: IViewsNamespace;
     readonly prompts: IPromptsNamespace;
+    readonly ai: IEntityAINamespace;
     registerInstructions(instructions: string): void;
 }
 interface EntityPluginContext extends BasePluginContext {