@tuturuuu/ai 0.0.10

package/src/mind/types.ts

@@ -0,0 +1,20 @@
+export type {
+  MindAiPatch,
+  MindAiPatchRecord,
+  MindBoard,
+  MindBoardSnapshot,
+  MindBoardStatus,
+  MindBoardSummary,
+  MindEdge,
+  MindEdgeType,
+  MindEntityLinkType,
+  MindGroup,
+  MindHorizon,
+  MindJsonObject,
+  MindNode,
+  MindNodeLink,
+  MindNodeStatus,
+  MindNodeType,
+  MindPatchOperation,
+  MindTag,
+} from '@tuturuuu/types/db';

package/src/object/core.ts

@@ -0,0 +1,3 @@
+import { experimental_useObject as useObject } from '@ai-sdk/react';
+
+export { useObject };

package/src/object/flashcards/route.ts

@@ -0,0 +1,140 @@
+import { google } from '@ai-sdk/google';
+import {
+  createAdminClient,
+  createClient,
+} from '@tuturuuu/supabase/next/server';
+import { Output, streamText } from 'ai';
+import { NextResponse } from 'next/server';
+import { withAiMemory } from '../../memory';
+import { flashcardSchema } from '../types';
+
+const DEFAULT_MODEL_NAME = 'gemini-3.1-flash-lite';
+
+export async function POST(req: Request) {
+  const sbAdmin = await createAdminClient();
+
+  const { wsId, context } = (await req.json()) as {
+    wsId?: string;
+    context?: string;
+  };
+
+  try {
+    // if (!id) return new Response('Missing chat ID', { status: 400 });
+    if (!wsId) return new Response('Missing workspace ID', { status: 400 });
+    if (!context) return new Response('Missing context', { status: 400 });
+
+    // eslint-disable-next-line no-undef
+    const apiKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+    if (!apiKey) return new Response('Missing API key', { status: 400 });
+
+    const supabase = await createClient();
+
+    const {
+      data: { user },
+    } = await supabase.auth.getUser();
+
+    if (!user) return new Response('Unauthorized', { status: 401 });
+
+    const { count, error } = await sbAdmin
+      .from('workspace_secrets')
+      .select('*', { count: 'exact', head: true })
+      .eq('ws_id', wsId)
+      .eq('name', 'ENABLE_CHAT')
+      .eq('value', 'true');
+
+    if (error) return new Response(error.message, { status: 500 });
+    if (count === 0)
+      return new Response('You are not allowed to use this feature.', {
+        status: 401,
+      });
+
+    // let chatId = id;
+
+    // if (!chatId) {
+    //   const { data, error } = await sbAdmin
+    //     .from('ai_chats')
+    //     .select('id')
+    //     .eq('creator_id', user?.id)
+    //     .order('created_at', { ascending: false })
+    //     .limit(1)
+    //     .single();
+
+    //   if (error) return new Response(error.message, { status: 500 });
+    //   if (!data) return new Response('Internal Server Error', { status: 500 });
+
+    //   chatId = data.id;
+    // }
+
+    const result = streamText({
+      model: await withAiMemory({
+        customId: `flashcards-${Date.now()}`,
+        model: google(DEFAULT_MODEL_NAME),
+        product: 'education',
+        source: 'flashcard_generation',
+        surface: 'flashcard_generation',
+        userId: user.id,
+        wsId,
+      }),
+      prompt: `Generate 10 flashcards with the following context (in the same language as the provided context): ${context}`,
+      output: Output.object({ schema: flashcardSchema }),
+      providerOptions: {
+        google: {
+          safetySettings: [
+            {
+              category: 'HARM_CATEGORY_DANGEROUS_CONTENT',
+              threshold: 'BLOCK_NONE',
+            },
+            {
+              category: 'HARM_CATEGORY_HATE_SPEECH',
+              threshold: 'BLOCK_NONE',
+            },
+            {
+              category: 'HARM_CATEGORY_HARASSMENT',
+              threshold: 'BLOCK_NONE',
+            },
+            {
+              category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',
+              threshold: 'BLOCK_NONE',
+            },
+          ],
+        },
+      },
+      // onFinish: async (response) => {
+      // console.log('AI Response:', response);
+      // if (!response.object) {
+      //   console.log('No content found');
+      //   throw new Error('No content found');
+      // }
+      // const { error } = await sbAdmin.from('ai_chat_messages').insert({
+      //   chat_id: chatId,
+      //   creator_id: user.id,
+      //   content: response.text,
+      //   role: 'ASSISTANT',
+      //   model: model.toLowerCase(),
+      //   finish_reason: response.finishReason,
+      //   prompt_tokens: response.usage.promptTokens,
+      //   completion_tokens: response.usage.completionTokens,
+      //   metadata: { source: 'Tuturuuu' },
+      // });
+      // if (error) {
+      //   console.log('ERROR ORIGIN: ROOT COMPLETION');
+      //   console.log(error);
+      //   throw new Error(error.message);
+      // }
+      // console.log('AI Response saved to database');
+      // },
+    });
+
+    return result.toTextStreamResponse();
+  } catch (error) {
+    console.log(error);
+    return NextResponse.json(
+      {
+        message: `## Edge API Failure\nCould not complete the request. Please view the **Stack trace** below.\n\`\`\`bash\n${(error as Error)?.stack || 'No stack trace available'}`,
+      },
+      {
+        status: 500,
+      }
+    );
+  }
+}

package/src/object/quizzes/explanation/route.ts

@@ -0,0 +1,145 @@
+import { google } from '@ai-sdk/google';
+import {
+  createAdminClient,
+  createClient,
+} from '@tuturuuu/supabase/next/server';
+import { Output, streamText } from 'ai';
+import { NextResponse } from 'next/server';
+import { withAiMemory } from '../../../memory';
+import { quizOptionExplanationSchema } from '../../types';
+
+const DEFAULT_MODEL_NAME = 'gemini-3.1-flash-lite';
+
+export async function POST(req: Request) {
+  const sbAdmin = await createAdminClient();
+
+  const { wsId, question, option } = (await req.json()) as {
+    wsId?: string;
+    question?: string;
+    option?: {
+      is_correct?: boolean;
+      value?: string;
+    };
+  };
+
+  try {
+    // if (!id) return new Response('Missing chat ID', { status: 400 });
+    if (!wsId) return new Response('Missing workspace ID', { status: 400 });
+    if (!question || !option)
+      return new Response('Missing context', { status: 400 });
+
+    // eslint-disable-next-line no-undef
+    const apiKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+    if (!apiKey) return new Response('Missing API key', { status: 400 });
+
+    const supabase = await createClient();
+
+    const {
+      data: { user },
+    } = await supabase.auth.getUser();
+
+    if (!user) return new Response('Unauthorized', { status: 401 });
+
+    const { count, error } = await sbAdmin
+      .from('workspace_secrets')
+      .select('*', { count: 'exact', head: true })
+      .eq('ws_id', wsId)
+      .eq('name', 'ENABLE_CHAT')
+      .eq('value', 'true');
+
+    if (error) return new Response(error.message, { status: 500 });
+    if (count === 0)
+      return new Response('You are not allowed to use this feature.', {
+        status: 401,
+      });
+
+    // let chatId = id;
+
+    // if (!chatId) {
+    //   const { data, error } = await sbAdmin
+    //     .from('ai_chats')
+    //     .select('id')
+    //     .eq('creator_id', user?.id)
+    //     .order('created_at', { ascending: false })
+    //     .limit(1)
+    //     .single();
+
+    //   if (error) return new Response(error.message, { status: 500 });
+    //   if (!data) return new Response('Internal Server Error', { status: 500 });
+
+    //   chatId = data.id;
+    // }
+
+    const result = streamText({
+      model: await withAiMemory({
+        customId: `quiz-explanation-${Date.now()}`,
+        model: google(DEFAULT_MODEL_NAME),
+        product: 'education',
+        source: 'quiz_explanation',
+        surface: 'quiz_explanation',
+        userId: user.id,
+        wsId,
+      }),
+      prompt: `Generate an explanation with the following context: \n\n"""Question: ${question}""" \n\n"""Option: ${option.value}"""\n\nIs this option correct? ${option.is_correct ? 'Yes' : 'No'}\n\nNOTE: Provide it in the same language as the question and option, be concise and clear.`,
+      output: Output.object({ schema: quizOptionExplanationSchema }),
+      providerOptions: {
+        google: {
+          safetySettings: [
+            {
+              category: 'HARM_CATEGORY_DANGEROUS_CONTENT',
+              threshold: 'BLOCK_NONE',
+            },
+            {
+              category: 'HARM_CATEGORY_HATE_SPEECH',
+              threshold: 'BLOCK_NONE',
+            },
+            {
+              category: 'HARM_CATEGORY_HARASSMENT',
+              threshold: 'BLOCK_NONE',
+            },
+            {
+              category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',
+              threshold: 'BLOCK_NONE',
+            },
+          ],
+        },
+      },
+      // onFinish: async (response) => {
+      //   console.log('AI Response:', response);
+      // if (!response.object) {
+      //   console.log('No content found');
+      //   throw new Error('No content found');
+      // }
+      // const { error } = await sbAdmin.from('ai_chat_messages').insert({
+      //   chat_id: chatId,
+      //   creator_id: user.id,
+      //   content: response.text,
+      //   role: 'ASSISTANT',
+      //   model: model.toLowerCase(),
+      //   finish_reason: response.finishReason,
+      //   prompt_tokens: response.usage.promptTokens,
+      //   completion_tokens: response.usage.completionTokens,
+      //   metadata: { source: 'Tuturuuu' },
+      // });
+      // if (error) {
+      //   console.log('ERROR ORIGIN: ROOT COMPLETION');
+      //   console.log(error);
+      //   throw new Error(error.message);
+      // }
+      // console.log('AI Response saved to database');
+      // },
+    });
+
+    return result.toTextStreamResponse();
+  } catch (error) {
+    console.log(error);
+    return NextResponse.json(
+      {
+        message: `## Edge API Failure\nCould not complete the request. Please view the **Stack trace** below.\n\`\`\`bash\n${(error as Error)?.stack || 'No stack trace available'}`,
+      },
+      {
+        status: 500,
+      }
+    );
+  }
+}

package/src/object/quizzes/route.ts

@@ -0,0 +1,142 @@
+import { google } from '@ai-sdk/google';
+import {
+  createAdminClient,
+  createClient,
+} from '@tuturuuu/supabase/next/server';
+import { Output, streamText } from 'ai';
+import { NextResponse } from 'next/server';
+import { withAiMemory } from '../../memory';
+import { quizSchema } from '../types';
+
+const DEFAULT_MODEL_NAME = 'gemini-3.1-flash-lite';
+
+export async function POST(req: Request) {
+  const sbAdmin = await createAdminClient();
+
+  const { wsId, context } = (await req.json()) as {
+    wsId?: string;
+    context?: string;
+  };
+
+  try {
+    // if (!id) return new Response('Missing chat ID', { status: 400 });
+    if (!wsId) return new Response('Missing workspace ID', { status: 400 });
+    if (!context) return new Response('Missing context', { status: 400 });
+
+    // eslint-disable-next-line no-undef
+    const apiKey = process.env.GOOGLE_GENERATIVE_AI_API_KEY;
+    if (!apiKey) return new Response('Missing API key', { status: 400 });
+
+    const supabase = await createClient();
+
+    const {
+      data: { user },
+    } = await supabase.auth.getUser();
+
+    if (!user) return new Response('Unauthorized', { status: 401 });
+
+    const { count, error } = await sbAdmin
+      .from('workspace_secrets')
+      .select('*', { count: 'exact', head: true })
+      .eq('ws_id', wsId)
+      .eq('name', 'ENABLE_CHAT')
+      .eq('value', 'true');
+
+    if (error) return new Response(error.message, { status: 500 });
+    if (count === 0)
+      return new Response('You are not allowed to use this feature.', {
+        status: 401,
+      });
+
+    // let chatId = id;
+
+    // if (!chatId) {
+    //   const { data, error } = await sbAdmin
+    //     .from('ai_chats')
+    //     .select('id')
+    //     .eq('creator_id', user?.id)
+    //     .order('created_at', { ascending: false })
+    //     .limit(1)
+    //     .single();
+
+    //   if (error) return new Response(error.message, { status: 500 });
+    //   if (!data) return new Response('Internal Server Error', { status: 500 });
+
+    //   chatId = data.id;
+    // }
+
+    const result = streamText({
+      model: await withAiMemory({
+        customId: `quiz-${Date.now()}`,
+        model: google(DEFAULT_MODEL_NAME),
+        product: 'education',
+        source: 'quiz_generation',
+        surface: 'quiz_generation',
+        userId: user.id,
+        wsId,
+      }),
+      prompt:
+        `Generate 10 quizzes with the following context (in the same language as the provided context): ` +
+        context,
+      output: Output.object({ schema: quizSchema }),
+      providerOptions: {
+        google: {
+          safetySettings: [
+            {
+              category: 'HARM_CATEGORY_DANGEROUS_CONTENT',
+              threshold: 'BLOCK_NONE',
+            },
+            {
+              category: 'HARM_CATEGORY_HATE_SPEECH',
+              threshold: 'BLOCK_NONE',
+            },
+            {
+              category: 'HARM_CATEGORY_HARASSMENT',
+              threshold: 'BLOCK_NONE',
+            },
+            {
+              category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',
+              threshold: 'BLOCK_NONE',
+            },
+          ],
+        },
+      },
+      // onFinish: async (response) => {
+      //   console.log('AI Response:', response);
+      // if (!response.object) {
+      //   console.log('No content found');
+      //   throw new Error('No content found');
+      // }
+      // const { error } = await sbAdmin.from('ai_chat_messages').insert({
+      //   chat_id: chatId,
+      //   creator_id: user.id,
+      //   content: response.text,
+      //   role: 'ASSISTANT',
+      //   model: model.toLowerCase(),
+      //   finish_reason: response.finishReason,
+      //   prompt_tokens: response.usage.promptTokens,
+      //   completion_tokens: response.usage.completionTokens,
+      //   metadata: { source: 'Tuturuuu' },
+      // });
+      // if (error) {
+      //   console.log('ERROR ORIGIN: ROOT COMPLETION');
+      //   console.log(error);
+      //   throw new Error(error.message);
+      // }
+      // console.log('AI Response saved to database');
+      // },
+    });
+
+    return result.toTextStreamResponse();
+  } catch (error) {
+    console.log(error);
+    return NextResponse.json(
+      {
+        message: `## Edge API Failure\nCould not complete the request. Please view the **Stack trace** below.\n\`\`\`bash\n${(error as Error)?.stack || 'No stack trace available'}`,
+      },
+      {
+        status: 500,
+      }
+    );
+  }
+}

package/src/object/types.ts

@@ -0,0 +1,187 @@
+import { z } from 'zod';
+
+export const flashcardSchema = z.object({
+  flashcards: z.array(
+    z.object({
+      front: z.string().describe('Question. Do not use emojis or links.'),
+      back: z.string().describe('Answer. Do not use emojis or links.'),
+    })
+  ),
+});
+
+export const quizSchema = z.object({
+  quizzes: z.array(
+    z.object({
+      question: z.string().describe('Question. Do not use emojis or links.'),
+      quiz_options: z.array(
+        z.object({
+          value: z.string().describe('Option. Do not use emojis or links.'),
+          explanation: z
+            .string()
+            .describe(
+              'Explain why this option is correct or incorrect, if it is incorrect, explain possible misconceptions and what made the option wrong with respect to the question, if it is correct, explain why it is correct. Be as detailed as possible.'
+            ),
+          is_correct: z.boolean().describe('This option is a correct answer.'),
+        })
+      ),
+    })
+  ),
+});
+
+export const quizOptionExplanationSchema = z.object({
+  explanation: z
+    .string()
+    .describe(
+      'Explain why this option is correct or incorrect, if it is incorrect, explain possible misconceptions and what made the option wrong with respect to the question, if it is correct, explain why it is correct. Be as detailed as possible.'
+    ),
+});
+
+export const quickJournalTaskSchema = z.object({
+  tasks: z
+    .array(
+      z.object({
+        title: z
+          .string()
+          .min(1)
+          .max(120)
+          .describe('Concise, action-oriented title for the task'),
+        description: z
+          .string()
+          .max(4000)
+          .optional()
+          .describe(
+            'Optional summary of the task with actionable next steps when requested'
+          ),
+        priority: z
+          .enum(['critical', 'high', 'normal', 'low'])
+          .optional()
+          .describe('Relative urgency for the task'),
+        labelSuggestions: z
+          .array(
+            z
+              .string()
+              .min(1)
+              .max(48)
+              .describe('Short keyword or phrase recommended as a label')
+          )
+          .max(6)
+          .optional()
+          .describe('Suggested labels that align with the task content'),
+        dueDate: z
+          .string()
+          .optional()
+          .describe(
+            'ISO 8601 date (YYYY-MM-DD) when an explicit due date is mentioned; omit otherwise'
+          ),
+      })
+    )
+    .min(1)
+    .max(50)
+    .describe('List of actionable tasks derived from the journal entry'),
+});
+
+const resourceSchema = z.object({
+  title: z.string().describe('Title of the resource'),
+  url: z.string().describe('URL or link to the resource'),
+  type: z
+    .enum(['video', 'article', 'book', 'course', 'other'])
+    .describe('Type of resource'),
+});
+
+const taskSchema = z.object({
+  title: z.string().describe('Title of the task'),
+  description: z
+    .string()
+    .describe('Detailed description of what needs to be done'),
+  priority: z
+    .enum(['high', 'medium', 'low'])
+    .describe('Priority level of the task'),
+  start_date: z.string().describe('Start date of the task (ISO date string)'),
+  end_date: z.string().describe('End date of the task (ISO date string)'),
+  status: z
+    .enum(['not-started', 'in-progress', 'completed', 'blocked'])
+    .describe('Current status of the task'),
+  estimatedHours: z
+    .number()
+    .min(0)
+    .describe('Estimated hours to complete the task'),
+  // dependencies: z
+  //   .array(z.string())
+  //   .optional()
+  //   .describe('List of dependencies for this task'),
+  resources: z
+    .array(resourceSchema)
+    .optional()
+    .describe('Learning resources for this task'),
+});
+
+const milestoneSchema = z.object({
+  title: z.string().describe('Title of the milestone'),
+  description: z
+    .string()
+    .describe('Detailed description of what needs to be achieved'),
+  start_date: z
+    .string()
+    .describe('Start date of the milestone (ISO date string)'),
+  end_date: z.string().describe('End date of the milestone (ISO date string)'),
+  tasks: z.array(taskSchema).describe('List of tasks for this milestone'),
+  // progress: z
+  //   .number()
+  //   .min(0)
+  //   .max(100)
+  //   .describe('Progress percentage of the milestone'),
+  // objectives: z
+  //   .array(z.string())
+  //   .describe('Learning objectives for this milestone'),
+  // keyOutcomes: z
+  //   .array(z.string())
+  //   .describe('Key outcomes expected from this milestone'),
+});
+
+const quarterSchema = z.object({
+  quarter: z.number().min(1).max(4).describe('Quarter number (1-4)'),
+  focus: z.string().describe('Main focus and objectives for this quarter'),
+  milestones: z
+    .array(milestoneSchema)
+    .describe('List of milestones for this quarter'),
+  start_date: z
+    .string()
+    .describe('Start date of the quarter (ISO date string)'),
+  end_date: z.string().describe('End date of the quarter (ISO date string)'),
+  // learningObjectives: z
+  //   .array(z.string())
+  //   .describe('Learning objectives for this quarter'),
+  // expectedOutcomes: z
+  //   .array(z.string())
+  //   .describe('Expected outcomes for this quarter'),
+  // progress: z
+  //   .number()
+  //   .min(0)
+  //   .max(100)
+  //   .optional()
+  //   .describe('Progress percentage of the quarter'),
+});
+
+export const yearPlanSchema = z.object({
+  yearPlan: z.object({
+    overview: z
+      .string()
+      .describe(
+        'A high-level overview of the year plan and how the goals will be achieved'
+      ),
+    quarters: z
+      .array(quarterSchema)
+      .describe('List of quarters in the year plan'),
+    recommendations: z
+      .array(z.string())
+      .describe('Additional recommendations, tips, or considerations'),
+    start_date: z
+      .string()
+      .describe('Start date of the year plan (ISO date string)'),
+    end_date: z
+      .string()
+      .describe('End date of the year plan (ISO date string)'),
+    // metadata: metadataSchema.describe('Additional metadata about the plan'),
+    // progress: progressSchema.describe('Overall progress tracking'),
+  }),
+});