diff --git a/backend/src/lib/cosmos-init.ts b/backend/src/lib/cosmos-init.ts
index e214b6d..2995dc7 100644
--- a/backend/src/lib/cosmos-init.ts
+++ b/backend/src/lib/cosmos-init.ts
@@ -13,6 +13,8 @@ const CONTAINER_DEFS: Record<string, ContainerConfig> = {
   note_prompts: { partitionKeyPath: '/userId' },
   note_prompt_schedules: { partitionKeyPath: '/userId' },
   note_prompt_webhooks: { partitionKeyPath: '/userId' },
+  note_shares: { partitionKeyPath: '/workspaceId' },
+  note_versions: { partitionKeyPath: '/workspaceId' },
 };
 
 export async function initCosmosIfNeeded(): Promise<void> {
diff --git a/backend/src/modules/note-prompts/types.ts b/backend/src/modules/note-prompts/types.ts
index 7be5f80..0cb92d9 100644
--- a/backend/src/modules/note-prompts/types.ts
+++ b/backend/src/modules/note-prompts/types.ts
@@ -5,7 +5,7 @@ import { z } from 'zod';
 export const PROMPT_CATEGORIES = ['transform', 'extract', 'generate', 'analyze', 'export'] as const;
 export type PromptCategory = (typeof PROMPT_CATEGORIES)[number];
 
-export const PROMPT_INPUT_TYPES = ['text', 'image', 'text+image'] as const;
+export const PROMPT_INPUT_TYPES = ['text', 'image', 'text+image', 'multi-note'] as const;
 export type PromptInputType = (typeof PROMPT_INPUT_TYPES)[number];
 
 export const PROMPT_OUTPUT_TYPES = ['replace', 'new_note', 'artifact', 'inline'] as const;