chore: Include invocation result (#265)

- Include invocation result in message endpoint - Remove `public` from run-configs
inferablehq · Dec 10, 2024 · b4c3add · b4c3add
1 parent 24b58d0
commit b4c3add
Show file tree

Hide file tree

Showing 13 changed files with 1,644 additions and 66 deletions.
diff --git a/app/app/clusters/[clusterId]/configs/[promptId]/edit/page.tsx b/app/app/clusters/[clusterId]/configs/[promptId]/edit/page.tsx
@@ -77,7 +77,6 @@ export default function EditPromptTemplate({
     initialPrompt?: string;
     systemPrompt?: string;
     attachedFunctions: string;
-    public: boolean;
     resultSchema?: string;
     inputSchema?: string;
   }) => {
@@ -90,7 +89,6 @@ export default function EditPromptTemplate({
           systemPrompt:
             formData.systemPrompt === "" ? undefined : formData.systemPrompt,
           name: formData.name === "" ? undefined : formData.name,
-          public: formData.public,
           resultSchema: formData.resultSchema
             ? JSON.parse(formData.resultSchema)
             : undefined,
@@ -195,7 +193,6 @@ export default function EditPromptTemplate({
                             attachedFunctions: version.attachedFunctions,
                             resultSchema: version.resultSchema,
                             inputSchema: version.inputSchema,
-                            public: version.public,
                           });
                           setSelectedVersion(version.version);
                           toast.success(
@@ -231,7 +228,6 @@ export default function EditPromptTemplate({
               initialPrompt: promptTemplate.initialPrompt ?? undefined,
               systemPrompt: promptTemplate.systemPrompt ?? undefined,
               attachedFunctions: promptTemplate.attachedFunctions,
-              public: promptTemplate.public,
               resultSchema: promptTemplate.resultSchema
                 ? promptTemplate.resultSchema
                 : undefined,

diff --git a/app/app/clusters/[clusterId]/configs/new/page.tsx b/app/app/clusters/[clusterId]/configs/new/page.tsx
@@ -21,7 +21,6 @@ export default function NewPromptTemplate({
     name: string;
     initialPrompt?: string;
     systemPrompt?: string;
-    public: boolean;
     attachedFunctions: string;
     resultSchema?: string;
     inputSchema?: string;
@@ -41,7 +40,6 @@ export default function NewPromptTemplate({
             formData.initialPrompt === "" ? undefined : formData.initialPrompt,
           systemPrompt:
             formData.systemPrompt === "" ? undefined : formData.systemPrompt,
-          public: formData.public,
           resultSchema: formData.resultSchema
             ? JSON.parse(formData.resultSchema)
             : undefined,
@@ -94,7 +92,6 @@ export default function NewPromptTemplate({
         initialData={{
           name: "",
           initialPrompt: "",
-          public: false,
           attachedFunctions: [],
         }}
         onSubmit={handleSubmit}

diff --git a/app/client/contract.ts b/app/client/contract.ts
@@ -1092,7 +1092,6 @@ export const definition = {
         attachedFunctions: z.array(z.string()),
         resultSchema: anyObject.nullable(),
         inputSchema: anyObject.nullable(),
-        public: z.boolean(),
         createdAt: z.date(),
         updatedAt: z.date(),
         versions: z.array(
@@ -1104,7 +1103,6 @@ export const definition = {
             attachedFunctions: z.array(z.string()),
             resultSchema: anyObject.nullable(),
             inputSchema: anyObject.nullable(),
-            public: z.boolean(),
           }),
         ),
       }),
@@ -1133,7 +1131,6 @@ export const definition = {
       attachedFunctions: z.array(z.string()).optional(),
       resultSchema: anyObject.optional(),
       inputSchema: z.object({}).passthrough().optional().nullable(),
-      public: z.boolean(),
     }),
     responses: {
       201: z.object({ id: z.string() }),
@@ -1161,7 +1158,6 @@ export const definition = {
       attachedFunctions: z.array(z.string()).optional(),
       resultSchema: z.object({}).passthrough().optional().nullable(),
       inputSchema: z.object({}).passthrough().optional().nullable(),
-      public: z.boolean(),
     }),
     responses: {
       200: z.object({

diff --git a/app/components/chat/prompt-template-form.tsx b/app/components/chat/prompt-template-form.tsx
@@ -17,7 +17,6 @@ import { Button } from "../ui/button";
 import { Input } from "../ui/input";
 import { Textarea } from "../ui/textarea";
 import { StructuredOutputEditor } from "./structured-output-editor";
-import { Switch } from "../ui/switch";
 import Link from "next/link";
 
 const formSchema = z.object({
@@ -27,7 +26,6 @@ const formSchema = z.object({
   attachedFunctions: z.string(),
   resultSchema: z.string().optional(),
   inputSchema: z.string().optional(),
-  public: z.boolean().default(false),
 });
 
 type PromptTemplateFormProps = {
@@ -38,7 +36,6 @@ type PromptTemplateFormProps = {
     attachedFunctions: string[];
     resultSchema?: unknown;
     inputSchema?: unknown;
-    public: boolean;
   };
   onSubmit: (data: z.infer<typeof formSchema>) => Promise<void>;
   isLoading: boolean;
@@ -58,7 +55,6 @@ export function PromptTemplateForm({
       attachedFunctions: initialData.attachedFunctions?.join(", "),
       resultSchema: JSON.stringify(initialData.resultSchema, null, 2),
       inputSchema: JSON.stringify(initialData.inputSchema, null, 2),
-      public: initialData.public,
     },
   });
 
@@ -252,36 +248,6 @@ export function PromptTemplateForm({
             )}
           />
         </div>
-        <FormField
-          control={form.control}
-          name="public"
-          render={({ field }) => (
-            <FormItem>
-              <FormLabel>Public (Front end usage)</FormLabel>
-              <FormDescription>
-                Should this Run Configuration be avilable for use from front end
-                applications.
-                <br />
-                Please see our{" "}
-                <Link
-                  href="https://docs.inferable.ai/pages/frontend"
-                  target="_blank"
-                  className="underline"
-                >
-                  docs
-                </Link>{" "}
-                for more information
-              </FormDescription>
-              <FormControl>
-                <Switch
-                  checked={field.value}
-                  onCheckedChange={field.onChange}
-                />
-              </FormControl>
-              <FormMessage />
-            </FormItem>
-          )}
-        />
         <Button type="submit" disabled={isLoading}>
           {isLoading ? "Saving..." : "Save Run Configuration"}
         </Button>

diff --git a/app/components/workflow.tsx b/app/components/workflow.tsx
@@ -269,7 +269,9 @@ export function Run({
     })) || [];
 
   const eventElements =
-    runTimeline?.messages.map((m) => ({
+    runTimeline?.messages
+      .filter((m) => ["human", "agent", "template"].includes(m.type))
+      .map((m) => ({
       element: (
         <ElementWrapper
           mutableId={mutableId}

diff --git a/control-plane/drizzle/0191_perfect_wasp.sql b/control-plane/drizzle/0191_perfect_wasp.sql
@@ -0,0 +1 @@
+ALTER TABLE "prompt_templates" DROP COLUMN IF EXISTS "public";
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		ALTER TABLE "prompt_templates" DROP COLUMN IF EXISTS "public";