move files around, clean up

ai-service/src/handlers/chat.handler.ts

@@ -8,8 +8,8 @@ import {
     streamText,
     TextPart
 } from 'ai';
-import { createRuntimeContext, WriterType } from '../../lib/ai/mastra/utils/runtime-context';
-import { mastra } from '../../lib/ai/mastra';
+import { createRuntimeContext, WriterType } from '../lib/ai/mastra/utils/runtime-context';
+import { mastra } from '../lib/ai/mastra';
 import { createImagineClient } from '@/lib/imagine/create-artifact-client';
 import { anthropic } from '@ai-sdk/anthropic';
 import { getOrCreateArtifactSandbox, startDevServer } from '@/lib/daytona/daytona-sandbox';
@@ -20,7 +20,7 @@ import {
 import {
     getOrCreateConversation,
     updateConversationHistory
-} from '@/lib/message-history';
+} from './message-history';
 
 export const routeHandler = async (c: Context) => {
     try {

ai-service/src/handlers/conversations.handler.ts

@@ -1,5 +1,5 @@
 import { getArtifactSandbox, startDevServer } from "@/lib/daytona/daytona-sandbox";
-import { convertDBMessagesToImagineUIMessages, getOrCreateConversation } from "@/lib/message-history";
+import { convertDBMessagesToImagineUIMessages, getOrCreateConversation } from "./message-history";
 import { prisma } from "@/lib/prisma";
 import { ImagineUIMessage } from "@/shared-types";
 import { Context } from "hono";

ai-service/src/handlers/message-history.ts

@@ -1,6 +1,6 @@
 import { ImagineUIMessage } from "@/shared-types";
-import { Message, Prisma } from "./generated/prisma";
-import { prisma } from "./prisma";
+import { Message, Prisma } from "@/lib/generated/prisma";
+import { prisma } from "@/lib/prisma";
 
 type ConversationWithMessages = Prisma.ConversationGetPayload<{
   include: {

ai-service/src/lib/daytona-image.ts

@@ -1,29 +0,0 @@
-import { Image } from '@daytonaio/sdk';
-import { daytona } from './daytona-client';
-
-export async function buildImage() {
-    const image = Image.base("daytonaio/sandbox")
-        .workdir('/home/daytona/artifact')
-        .runCommands('npm -v');
-
-    console.log(' image', image);
-
-    const snapshot = await daytona.snapshot.create(
-        {
-            name: 'imgn-vite:v3',
-            image,
-            resources: {
-                cpu: 1,
-                memory: 1,
-                disk: 3
-            }
-        },
-        {
-            onLogs: console.log
-        }
-    );
-
-    console.log('snapshot', snapshot);
-}
-
-buildImage();

ai-service/src/lib/daytona/cleanup-sandboxes.ts

@@ -1,4 +1,4 @@
-import { daytona } from "@/lib/daytona-client";
+import { daytona } from "@/lib/daytona/daytona-client";
 
 async function main() {
     const sandboxes = await daytona.list();

ai-service/src/lib/daytona/daytona-sandbox.ts

@@ -1,5 +1,5 @@
 import { Sandbox } from '@daytonaio/sdk';
-import { daytona } from '../daytona-client';
+import { daytona } from './daytona-client';
 import { OnStepUpdateFn, WorkspaceStepId } from '@/lib/ai/custom-parts/workspace-state';
 import { prisma } from '@/lib/prisma';
 

ai-service/src/server.ts

@@ -8,8 +8,8 @@ import { Hono } from 'hono';
 import { cors } from 'hono/cors';
 import { HTTPException } from 'hono/http-exception';
 import { fileURLToPath } from 'url';
-import { handleChatRequest } from '@/handlers/chat/route';
-import { getConversation, getConversations } from './handlers/conversation';
+import { handleChatRequest } from '@/handlers/chat.handler';
+import { getConversation, getConversations } from './handlers/conversations.handler';
 import { contextStorage } from "hono/context-storage";
 
 const app = new Hono();

ai-service/src/shared-types/index.ts

@@ -3,7 +3,7 @@ import { thinkingUIDataPartSchema } from '@/lib/ai/custom-parts/thinking';
 import { workspaceStateUIDataPartSchema } from '@/lib/ai/custom-parts/workspace-state';
 import { fileTools } from '@/lib/ai/mastra/tools/file-tools';
 import { InferUIDataParts, InferUITool, ToolUIPart, UIMessage } from 'ai';
-export type { GetConversationResult, GetConversationsResult } from '@/handlers/conversation';
+export type { GetConversationResult, GetConversationsResult } from '@/handlers/conversations.handler';
 
 export type ImagineTools = {
     readFile: InferUITool<typeof fileTools.readFileTool>;