Post Intent

prisma/migrations/20240509145855_add_domain_to_tool/migration.sql

@@ -0,0 +1,3 @@
+-- AlterTable
+ALTER TABLE "Tool"
+  ADD COLUMN "domain" TEXT;

prisma/schema.prisma

@@ -2822,7 +2822,6 @@ model ChatMessage {
   chatId             Int
   content            String
   contentType        ChatMessageType @default(Markdown)
-  // isSystemMessage Boolean @default(false)
   referenceMessageId Int?
   editedAt           DateTime?
   // TODO deletedAt DateTime
@@ -2998,6 +2997,7 @@ model Tool {
   createdAt  DateTime    @default(now())
   enabled    Boolean     @default(true)
   type       ToolType
+  domain     String?
   imageTools ImageTool[]
 }
 

src/components/Image/ImageDropzone/ImageDropzone.tsx

@@ -51,7 +51,13 @@ export function ImageDropzone({
 
   const handleDropCapture = async (e: DragEvent) => {
     const url = e.dataTransfer.getData('text/uri-list');
-    if (!url.startsWith('https://orchestration.civitai.com')) return;
+    if (
+      !(
+        url.startsWith('https://orchestration.civitai.com') ||
+        url.startsWith('https://orchestration-stage.civitai.com')
+      )
+    )
+      return;
     const blob = await fetchBlob(url);
     if (!blob) return;
     const file = new File([blob], url.substring(url.lastIndexOf('/')), { type: blob.type });

src/components/Image/ImageDropzone/MediaDropzone.tsx

@@ -5,9 +5,6 @@ import { DragEvent } from 'react';
 import { constants } from '~/server/common/constants';
 import { IMAGE_MIME_TYPE, MIME_TYPES, VIDEO_MIME_TYPE } from '~/server/common/mime-types';
 import { fetchBlob } from '~/utils/file-utils';
-import { formatBytes } from '~/utils/number-helpers';
-import { PreprocessFileReturnType } from '~/utils/media-preprocessors';
-import { MediaUploadOnCompleteProps, useMediaUpload } from '~/hooks/useMediaUpload';
 
 const MAX_IMAGE_SIZE = constants.mediaUpload.maxImageFileSize;
 
@@ -39,7 +36,13 @@ export function MediaDropzone({
   // #region [handle drop]
   const handleDropCapture = async (e: DragEvent) => {
     const url = e.dataTransfer.getData('text/uri-list');
-    if (!url.startsWith('https://orchestration.civitai.com')) return;
+    if (
+      !(
+        url.startsWith('https://orchestration.civitai.com') ||
+        url.startsWith('https://orchestration-stage.civitai.com')
+      )
+    )
+      return;
     const blob = await fetchBlob(url);
     if (!blob) return;
     const file = new File([blob], url.substring(url.lastIndexOf('/')), { type: blob.type });