Fix discrepencies between ContentParts with Pydantic, GQL Input, and …

…GQL Output

- Pydantic discriminated union was too strict, need to populate type field with defaults during validation since they are not provided by input
- Strawberry input types need to default to UNSET when using OneOf decorator arg
- unlike Strawberry input type, Strawberry output types need to match the same shape as the pydantic models so that they can be automatically converted from ORM model

app/schema.graphql

@@ -252,13 +252,13 @@ input ClusterInput {
   id: ID
 }
 
-union ContentPart = TextContentValue | ImageContentValue | ToolCallContentValue | ToolResultContentValue
+union ContentPart = TextContentPart | ImageContentPart | ToolCallContentPart | ToolResultContentPart
 
 input ContentPartInput @oneOf {
-  text: TextContentValueInput = null
-  image: ImageContentValueInput = null
-  toolCall: ToolCallContentValueInput = null
-  toolResult: ToolResultContentValueInput = null
+  text: TextContentValueInput
+  image: ImageContentValueInput
+  toolCall: ToolCallContentValueInput
+  toolResult: ToolResultContentValueInput
 }
 
 input CreateApiKeyInput {
@@ -1045,6 +1045,10 @@ input Granularity {
   samplingIntervalMinutes: Int!
 }
 
+type ImageContentPart {
+  image: ImageContentValue!
+}
+
 type ImageContentValue {
   url: String!
 }
@@ -1913,6 +1917,10 @@ type TextChunk implements ChatCompletionSubscriptionPayload {
   content: String!
 }
 
+type TextContentPart {
+  text: TextContentValue!
+}
+
 type TextContentValue {
   text: String!
 }
@@ -1950,12 +1958,16 @@ type ToolCallChunk implements ChatCompletionSubscriptionPayload {
   function: FunctionCallChunk!
 }
 
+type ToolCallContentPart {
+  toolCall: ToolCallContentValue!
+}
+
 type ToolCallContentValue {
-  toolCall: String!
+  toolCallId: String!
 }
 
 input ToolCallContentValueInput {
-  toolCall: String!
+  toolCallId: String!
 }
 
 type ToolDefinition {
@@ -1966,6 +1978,10 @@ input ToolDefinitionInput {
   definition: JSON!
 }
 
+type ToolResultContentPart {
+  toolResult: ToolResultContentValue!
+}
+
 type ToolResultContentValue {
   toolCallId: String!
   result: JSON!

app/src/pages/playground/fetchPlaygroundPrompt.ts

@@ -344,18 +344,26 @@ const fetchPlaygroundPromptQuery = graphql`
                     role
                     content {
                       __typename
-                      ... on TextContentValue {
-                        text
+                      ... on TextContentPart {
+                        text {
+                          text
+                        }
                       }
-                      ... on ImageContentValue {
-                        url
+                      ... on ImageContentPart {
+                        image {
+                          url
+                        }
                       }
-                      ... on ToolCallContentValue {
-                        toolCall
+                      ... on ToolCallContentPart {
+                        toolCall {
+                          toolCallId
+                        }
                       }
-                      ... on ToolResultContentValue {
-                        toolCallId
-                        result
+                      ... on ToolResultContentPart {
+                        toolResult {
+                          toolCallId
+                          result
+                        }
                       }
                     }
                   }

app/src/pages/prompt/PromptChatMessagesCard.tsx

@@ -41,18 +41,26 @@ export function PromptChatMessages({
               role
               content {
                 __typename
-                ... on TextContentValue {
-                  text
+                ... on TextContentPart {
+                  text {
+                    text
+                  }
                 }
-                ... on ImageContentValue {
-                  url
+                ... on ImageContentPart {
+                  image {
+                    url
+                  }
                 }
-                ... on ToolCallContentValue {
-                  toolCall
+                ... on ToolCallContentPart {
+                  toolCall {
+                    toolCallId
+                  }
                 }
-                ... on ToolResultContentValue {
-                  toolCallId
-                  result
+                ... on ToolResultContentPart {
+                  toolResult {
+                    toolCallId
+                    result
+                  }
                 }
               }
             }