@chat-js/cli 0.3.0 → 0.4.0

package/templates/chat-app/providers/chat-input-provider.tsx

@@ -13,7 +13,12 @@ import React, {
 } from "react";
 import type { LexicalChatInputRef } from "@/components/lexical-chat-input";
 import type { AppModelId } from "@/lib/ai/app-models";
-import type { Attachment, UiToolName } from "@/lib/ai/types";
+import {
+  getPrimarySelectedModelId,
+  type Attachment,
+  type SelectedModelValue,
+  type UiToolName,
+} from "@/lib/ai/types";
 import { useChatModels } from "./chat-models-provider";
 import { useDefaultModel, useModelChange } from "./default-model-provider";
 
@@ -24,10 +29,12 @@ interface ChatInputContextType {
   getInputValue: () => string;
   handleInputChange: (value: string) => void;
   handleModelChange: (modelId: AppModelId) => Promise<void>;
+  handleModelSelectionChange: (selection: SelectedModelValue) => Promise<void>;
   handleSubmit: (submitFn: () => void, isEditMode?: boolean) => void;
   isEmpty: boolean;
   isProjectContext: boolean;
   selectedModelId: AppModelId;
+  selectedModelSelection: SelectedModelValue;
   selectedTool: UiToolName | null;
   setAttachments: Dispatch<SetStateAction<Attachment[]>>;
   setSelectedTool: Dispatch<SetStateAction<UiToolName | null>>;
@@ -45,6 +52,7 @@ interface ChatInputProviderProps {
   isProjectContext?: boolean;
   localStorageEnabled?: boolean;
   overrideModelId?: AppModelId; // For message editing where we want to use the original model
+  overrideModelSelection?: SelectedModelValue; // For message editing with multi-model selection
 }
 
 export function ChatInputProvider({
@@ -53,6 +61,7 @@ export function ChatInputProvider({
   initialTool = null,
   initialAttachments = [],
   overrideModelId,
+  overrideModelSelection,
   localStorageEnabled = true,
   isProjectContext = false,
 }: ChatInputProviderProps) {
@@ -95,6 +104,10 @@ export function ChatInputProvider({
   const [selectedModelId, setSelectedModelId] = useState<AppModelId>(
     overrideModelId || defaultModel
   );
+  const [selectedModelSelection, setSelectedModelSelection] =
+    useState<SelectedModelValue>(
+      overrideModelSelection ?? overrideModelId ?? defaultModel
+    );
 
   // IMPORTANT: do not read localStorage during initial render.
   // Next SSRs client components; localStorage is client-only and will cause hydration mismatches
@@ -128,7 +141,7 @@ export function ChatInputProvider({
 
   const { getModelById } = useChatModels();
 
-  const handleModelChange = useCallback(
+  const persistPrimaryModelChange = useCallback(
     async (modelId: AppModelId) => {
       const modelDef = getModelById(modelId);
 
@@ -146,6 +159,29 @@ export function ChatInputProvider({
     [selectedTool, changeModel, getModelById]
   );
 
+  const handleModelChange = useCallback(
+    async (modelId: AppModelId) => {
+      setSelectedModelSelection(modelId);
+      await persistPrimaryModelChange(modelId);
+    },
+    [persistPrimaryModelChange]
+  );
+
+  const handleModelSelectionChange = useCallback(
+    async (selection: SelectedModelValue) => {
+      setSelectedModelSelection(selection);
+
+      const primaryModelId = getPrimarySelectedModelId(selection);
+
+      if (!primaryModelId) {
+        return;
+      }
+
+      await persistPrimaryModelChange(primaryModelId);
+    },
+    [persistPrimaryModelChange]
+  );
+
   const clearInput = useCallback(() => {
     editorRef.current?.clear();
     setLocalStorageInput("");
@@ -207,7 +243,9 @@ export function ChatInputProvider({
         attachments,
         setAttachments,
         selectedModelId,
+        selectedModelSelection,
         handleModelChange,
+        handleModelSelectionChange,
         getInputValue,
         handleInputChange,
         getInitialInput,

package/templates/chat-app/scripts/db-branch-delete.sh

@@ -2,7 +2,13 @@
 set -e
 
 BRANCH_NAME="${1:-dev-local}"
-BRANCH_FILE=".neon-branch"
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+MONOREPO_ROOT="$(cd "$SCRIPT_DIR/../../.." && pwd)"
+if [ -f "$MONOREPO_ROOT/turbo.json" ]; then
+  BRANCH_FILE="$MONOREPO_ROOT/.neon-branch"
+else
+  BRANCH_FILE="$(cd "$SCRIPT_DIR/.." && pwd)/.neon-branch"
+fi
 
 # Check if we're currently on this branch
 if [ -f "$BRANCH_FILE" ] && [ "$(cat "$BRANCH_FILE")" = "$BRANCH_NAME" ]; then

package/templates/chat-app/scripts/db-branch-use.sh

@@ -2,7 +2,13 @@
 # Switch active database branch (like git checkout)
 set -e
 
-BRANCH_FILE=".neon-branch"
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+MONOREPO_ROOT="$(cd "$SCRIPT_DIR/../../.." && pwd)"
+if [ -f "$MONOREPO_ROOT/turbo.json" ]; then
+  BRANCH_FILE="$MONOREPO_ROOT/.neon-branch"
+else
+  BRANCH_FILE="$(cd "$SCRIPT_DIR/.." && pwd)/.neon-branch"
+fi
 BRANCH_NAME="${1:-}"
 
 if [ -z "$BRANCH_NAME" ]; then

package/templates/chat-app/scripts/with-db.sh

@@ -2,7 +2,13 @@
 # Wrapper that uses branch DATABASE_URL if .neon-branch exists, otherwise uses .env.local
 set -e
 
-BRANCH_FILE=".neon-branch"
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+MONOREPO_ROOT="$(cd "$SCRIPT_DIR/../../.." && pwd)"
+if [ -f "$MONOREPO_ROOT/turbo.json" ]; then
+  BRANCH_FILE="$MONOREPO_ROOT/.neon-branch"
+else
+  BRANCH_FILE="$(cd "$SCRIPT_DIR/.." && pwd)/.neon-branch"
+fi
 
 if [ -f "$BRANCH_FILE" ]; then
   BRANCH_NAME=$(cat "$BRANCH_FILE")

package/templates/chat-app/vitest.config.ts

@@ -1,6 +1,8 @@
+import tsconfigPaths from "vite-tsconfig-paths";
 import { defineConfig } from "vitest/config";
 
 export default defineConfig({
+  plugins: [tsconfigPaths()],
   test: {
     exclude: ["**/node_modules/**", "**/*.e2e.ts"],
   },