Support single-file prompts (#20)

* Support single-file prompts * `prompt-file` -> `prompts-file` * Add arg dump
docker · Sep 9, 2024 · 76acc60 · 76acc60
1 parent bc41cb1
commit 76acc60
Show file tree

Hide file tree

Showing 6 changed files with 60 additions and 271 deletions.
diff --git a/prompts/poem/local:/Users/colinmcneil/Dev/labs-ai-tools-for-devs/prompts/poem.md b/prompts/poem/local:/Users/colinmcneil/Dev/labs-ai-tools-for-devs/prompts/poem.md
diff --git a/src/extension/ui/src/App.tsx b/src/extension/ui/src/App.tsx
@@ -176,8 +176,11 @@ export function App() {
     ]);
 
     runOutput.updateOutput({ method: 'message', params: { debug: 'Running prompts...' } })
+
     const args = getRunArgs(selectedPrompt!, selectedProject!, "", client.host.platform)
 
+    runOutput.updateOutput({ method: 'message', params: { debug: `Running prompt image with args ${args}` } })
+
     client.docker.cli.exec("run", args, {
       stream: {
         splitOutputLines: true,

diff --git a/src/extension/ui/src/args.ts b/src/extension/ui/src/args.ts
@@ -1,13 +1,13 @@
 export const getRunArgs = (promptRef: string, projectDir: string, username: string, platform: string, render = false) => {
     const isLocal = promptRef.startsWith('local://');
+    const isMarkdown = promptRef.toLowerCase().endsWith('.md');
     let promptArgs: string[] = ["--prompts", promptRef];
     let mountArgs: string[] = [];
-
     if (isLocal) {
         const localPromptPath = promptRef.replace('local://', '');
         const pathSeparator = platform === 'win32' ? '\\' : '/';
         promptRef = localPromptPath.split(pathSeparator).pop() || 'unknown-local-prompt';
-        promptArgs = ["--prompts-dir", `/app/${promptRef}`];
+        promptArgs = [isMarkdown ? "--prompts-file" : "--prompts-dir", `/app/${promptRef}`];
         mountArgs = ["--mount", `type=bind,source=${localPromptPath},target=/app/${promptRef}`];
     }
 

diff --git a/src/extension/ui/src/components/Projects.tsx b/src/extension/ui/src/components/Projects.tsx
@@ -21,16 +21,18 @@ const Projects: React.FC<ProjectsProps> = ({ projects, selectedProject, setProje
             <Stack direction='row' spacing={1} sx={{ mt: 1 }} alignItems={'center'} justifyContent={'space-between'}>
                 <Button sx={{ padding: 1 }} onClick={() => {
                     client.desktopUI.dialog.showOpenDialog({
-                        properties: ['openDirectory', 'multiSelections']
+                        title: 'Import projects',
+                        properties: ['openDirectory', 'multiSelections'],
                     }).then((result) => {
                         if (result.canceled) {
                             return;
                         }
                         const newProjects = result.filePaths;
                         setProjects([...projects, ...newProjects]);
+                        setSelectedProject(newProjects[0]);
                     });
                 }}>
-                    Import projects
+                    Import project(s)
                 </Button>
             </Stack>
             <List>
@@ -57,7 +59,7 @@ const Projects: React.FC<ProjectsProps> = ({ projects, selectedProject, setProje
                     </ListItem>
                 ))}
             </List>
-        </Paper>
+        </Paper >
     );
 };
 

diff --git a/src/extension/ui/src/components/Prompts.tsx b/src/extension/ui/src/components/Prompts.tsx
@@ -37,15 +37,18 @@ const Prompts: React.FC<PromptsProps> = ({ prompts, selectedPrompt, promptInput,
                 )}
                 <Button onClick={() => {
                     client.desktopUI.dialog.showOpenDialog({
-                        properties: ['openDirectory', 'multiSelections']
+                        title: 'Select prompt',
+                        properties: ['openDirectory', 'openFile', 'multiSelections'],
+                        filters: [{ name: 'Markdown', extensions: ['.md'] }],
                     }).then((result) => {
                         if (result.canceled) {
                             return;
                         }
                         track('DockerPromptsAddLocalPrompt');
                         setPrompts([...prompts, ...result.filePaths.map(p => `local://${p}`)]);
+                        setSelectedPrompt(`local://${result.filePaths[0]}`);
                     });
-                }}>Add local prompt</Button>
+                }}>Add local prompt(s)</Button>
             </Stack>
             <List>
                 {prompts.map((prompt) => (