feat(prompts): create prompt mutation (#5812)

Arize-ai · Jan 3, 2025 · 20a3d0a · 20a3d0a
1 parent 1d3919d
commit 20a3d0a
Show file tree

Hide file tree

Showing 20 changed files with 620 additions and 177 deletions.
diff --git a/app/schema.graphql b/app/schema.graphql
@@ -244,6 +244,12 @@ input CreateDatasetInput {
   metadata: JSON
 }
 
+input CreatePromptInput {
+  name: String!
+  description: String = null
+  promptVersion: PromptVersionInput!
+}
+
 input CreateSpanAnnotationInput {
   spanId: GlobalID!
   name: String!
@@ -1092,7 +1098,11 @@ type JSONPromptMessage {
 }
 
 type JSONSchema {
-  schema: JSON!
+  definition: JSON!
+}
+
+input JSONSchemaInput {
+  definition: JSON!
 }
 
 type LabelFraction {
@@ -1169,6 +1179,7 @@ type Mutation {
   exportClusters(clusters: [ClusterInput!]!, fileName: String): ExportedFile!
   deleteProject(id: GlobalID!): Query!
   clearProject(input: ClearProjectInput!): Query!
+  createPrompt(input: CreatePromptInput!): Prompt!
   createSpanAnnotations(input: [CreateSpanAnnotationInput!]!): SpanAnnotationMutationPayload!
   patchSpanAnnotations(input: [PatchAnnotationInput!]!): SpanAnnotationMutationPayload!
   deleteSpanAnnotations(input: DeleteAnnotationsInput!): SpanAnnotationMutationPayload!
@@ -1387,7 +1398,6 @@ type Prompt implements Node {
 }
 
 type PromptChatTemplate {
-  Version: String!
   messages: [PromptTemplateMessage!]!
 }
 
@@ -1477,6 +1487,18 @@ type PromptVersionEdge {
   node: PromptVersion!
 }
 
+input PromptVersionInput {
+  description: String = null
+  templateType: PromptTemplateType!
+  templateFormat: PromptTemplateFormat!
+  template: JSON!
+  invocationParameters: JSON! = {}
+  tools: [ToolDefinitionInput!]! = []
+  outputSchema: JSONSchemaInput = null
+  modelProvider: String!
+  modelName: String!
+}
+
 type PromptVersionTag implements Node {
   """The Globally Unique ID of this object"""
   id: GlobalID!
@@ -1858,6 +1880,10 @@ type ToolDefinition {
   definition: JSON!
 }
 
+input ToolDefinitionInput {
+  definition: JSON!
+}
+
 type Trace implements Node {
   """The Globally Unique ID of this object"""
   id: GlobalID!

diff --git a/app/src/pages/prompt/PromptCodeExportCard.tsx b/app/src/pages/prompt/PromptCodeExportCard.tsx
@@ -28,7 +28,7 @@ export function PromptCodeExportCard({
         modelName
         modelProvider
         outputSchema {
-          schema
+          definition
         }
         tools {
           definition

diff --git a/app/src/pages/prompt/__generated__/PromptCodeExportCard__main.graphql.ts b/app/src/pages/prompt/__generated__/PromptCodeExportCard__main.graphql.ts
diff --git a/app/src/pages/prompt/__generated__/promptLoaderQuery.graphql.ts b/app/src/pages/prompt/__generated__/promptLoaderQuery.graphql.ts
diff --git a/app/src/pages/prompt/__generated__/promptVersionLoaderQuery.graphql.ts b/app/src/pages/prompt/__generated__/promptVersionLoaderQuery.graphql.ts