templates list pulled from constants

stackblitz-labs · Nov 14, 2024 · e635984 · e635984
1 parent 2a684b4
commit e635984
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 20 deletions.
diff --git a/app/components/chat/ToolManager.tsx b/app/components/chat/ToolManager.tsx
@@ -1,5 +1,3 @@
-import { IconButton } from '../ui/IconButton';
-import { Popover, PopoverContent, PopoverTrigger } from '../ui/Popover';
 import { ToggleSwitch } from '../ui/ToggleSwitch';
 import type { IToolsConfig } from '~/utils/types';
 
@@ -17,7 +15,7 @@ export function ToolManager({ toolConfig, onConfigChange }: ToolManagerProps) {
             <label className="text-sm text-bolt-elements-textSecondary">Tool Calling</label>
             <ToggleSwitch
               checked={toolConfig.enabled}
-              onCheckedChange={(e) => {
+              onCheckedChange={(e: boolean) => {
                 onConfigChange?.({
                   enabled: e,
                   config: toolConfig.config,

diff --git a/app/lib/.server/llm/tools.ts b/app/lib/.server/llm/tools.ts
@@ -1,5 +1,7 @@
 import { z } from 'zod';
 import { tool } from 'ai';
+import { TEMPLATE_LIST } from '~/utils/constants';
+let templates = TEMPLATE_LIST.map(x => x.name) as [string, ...string[]];
 export default {
     SelectCodeTemplate: tool({
         description: `Use this to select one single code template best suited for the user query, 
@@ -8,18 +10,10 @@ export default {
         * START FROM SCRATCH IF User Says 'No'
         * ONLY SELECT ONE TEMPLATE.
         * ONLY Call this tool once.
+        * DO NOT CALL THIS TOOL IF there is no matching templates found
         `,
         parameters: z.object({
-            template: z.enum([
-                'blank',
-                'vite-react-js',
-                'vite-react-ts',
-                'vite-svelte',
-                'vite-vue',
-                'nextjs',
-                'vite-react-tailwind',
-                'vite-react-shadcn',
-                'python']),
+            template: z.enum([...templates]),
         }),
         // execute: async ({ template }) => template,
     }),

diff --git a/app/utils/constants.ts b/app/utils/constants.ts
@@ -52,13 +52,6 @@ const staticModels: ModelInfo[] = [
 
 const codeTemplates: TemplateInfo[] = [
   { name: 'vite-react-js', label: 'Vite React JS', githubRepo: 'thecodacus/vite-react-ts-template' },
-  { name: 'vite-react-ts', label: 'Vite React TS', githubRepo: 'thecodacus/vite-react-ts-template' },
-  { name: 'vite-svelte', label: 'Vite Svelte', githubRepo: 'thecodacus/vite-react-ts-template' },
-  { name: 'vite-vue', label: 'Vite Vue', githubRepo: 'thecodacus/vite-react-ts-template' },
-  { name: 'nextjs', label: 'NextJS', githubRepo: 'thecodacus/vite-react-ts-template' },
-  { name: 'vite-react-tailwind', label: 'Vite React Tailwind', githubRepo: 'thecodacus/vite-react-ts-template' },
-  { name: 'vite-react-shadcn', label: 'Vite React Shadcn', githubRepo: 'thecodacus/vite-react-ts-template' },
-  { name: 'python', label: 'Python', githubRepo: 'thecodacus/vite-react-ts-template' },
 ];
 
 export let MODEL_LIST: ModelInfo[] = [...staticModels];