Rename Image model to ImageResult

Arize-ai · Jan 14, 2025 · f90aa02 · f90aa02
1 parent dcf9abd
commit f90aa02
Show file tree

Hide file tree

Showing 12 changed files with 45 additions and 45 deletions.
diff --git a/app/schema.graphql b/app/schema.graphql
@@ -1034,19 +1034,19 @@ input Granularity {
   samplingIntervalMinutes: Int!
 }
 
-type Image {
+type ImagePart {
   type: String!
-  url: String!
+  image: ImageResult!
 }
 
-input ImageInput {
+type ImageResult {
   type: String!
   url: String!
 }
 
-type ImagePart {
+input ImageResultInput {
   type: String!
-  image: Image!
+  url: String!
 }
 
 type Inferences {
@@ -1275,7 +1275,7 @@ union Part = TextPart | ImagePart | ToolCallPart | ToolResultPart
 input PartInput {
   type: String!
   text: String = null
-  image: ImageInput = null
+  image: ImageResultInput = null
   toolCall: String = null
   toolResult: ToolResultInput = null
 }

diff --git a/...rc/pages/playground/__generated__/UpsertPromptFromTemplateDialogCreateMutation.graphql.ts b/...rc/pages/playground/__generated__/UpsertPromptFromTemplateDialogCreateMutation.graphql.ts
diff --git a/...rc/pages/playground/__generated__/UpsertPromptFromTemplateDialogUpdateMutation.graphql.ts b/...rc/pages/playground/__generated__/UpsertPromptFromTemplateDialogUpdateMutation.graphql.ts
diff --git a/app/src/pages/playground/__generated__/fetchPlaygroundPromptQuery.graphql.ts b/app/src/pages/playground/__generated__/fetchPlaygroundPromptQuery.graphql.ts
diff --git a/app/src/pages/prompt/__generated__/PromptChatMessagesCard__main.graphql.ts b/app/src/pages/prompt/__generated__/PromptChatMessagesCard__main.graphql.ts
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
diff --git a/schemas/openapi.json b/schemas/openapi.json
@@ -1722,42 +1722,42 @@
         "type": "object",
         "title": "HTTPValidationError"
       },
-      "Image": {
+      "ImagePart": {
         "properties": {
           "type": {
             "type": "string",
-            "const": "image_result",
+            "const": "image",
             "title": "Type"
           },
-          "url": {
-            "type": "string",
-            "title": "Url"
+          "image": {
+            "$ref": "#/components/schemas/ImageResult"
           }
         },
         "type": "object",
         "required": [
           "type",
-          "url"
+          "image"
         ],
-        "title": "Image"
+        "title": "ImagePart"
       },
-      "ImagePart": {
+      "ImageResult": {
         "properties": {
           "type": {
             "type": "string",
-            "const": "image",
+            "const": "image_result",
             "title": "Type"
           },
-          "image": {
-            "$ref": "#/components/schemas/Image"
+          "url": {
+            "type": "string",
+            "title": "Url"
           }
         },
         "type": "object",
         "required": [
           "type",
-          "image"
+          "url"
         ],
-        "title": "ImagePart"
+        "title": "ImageResult"
       },
       "InsertedSpanAnnotation": {
         "properties": {

diff --git a/src/phoenix/server/api/helpers/prompts/models.py b/src/phoenix/server/api/helpers/prompts/models.py
@@ -58,7 +58,7 @@ class TextPart(PartBase):
     text: str
 
 
-class Image(BaseModel):
+class ImageResult(BaseModel):
     type: Literal["image_result"]
     # http url, or base64 encoded image
     url: str
@@ -67,7 +67,7 @@ class Image(BaseModel):
 class ImagePart(PartBase):
     type: Literal["image"]
     # the image data
-    image: Image
+    image: ImageResult
 
 
 class ToolCallPart(PartBase):

diff --git a/src/phoenix/server/api/input_types/PromptVersionInput.py b/src/phoenix/server/api/input_types/PromptVersionInput.py
@@ -4,7 +4,7 @@
 from strawberry.scalars import JSON
 
 from phoenix.server.api.helpers.prompts.models import (
-    Image,
+    ImageResult,
     PromptChatTemplateV1,
     PromptTemplateFormat,
     PromptTemplateType,
@@ -26,8 +26,8 @@ class JSONSchemaInput:
     definition: JSON
 
 
-@strawberry.experimental.pydantic.input(Image)
-class ImageInput:
+@strawberry.experimental.pydantic.input(ImageResult)
+class ImageResultInput:
     type: str
     url: str
 
@@ -43,7 +43,7 @@ class ToolResultInput:
 class PartInput:
     type: str
     text: Optional[str] = None
-    image: Optional[ImageInput] = None
+    image: Optional[ImageResultInput] = None
     tool_call: Optional[str] = None
     tool_result: Optional[ToolResultInput] = None
 

diff --git a/src/phoenix/server/api/types/PromptVersionTemplate.py b/src/phoenix/server/api/types/PromptVersionTemplate.py
@@ -8,8 +8,8 @@
 from typing_extensions import TypeAlias, assert_never
 
 from phoenix.db.models import PromptVersion as ORMPromptVersion
-from phoenix.server.api.helpers.prompts.models import Image as ImageModel
 from phoenix.server.api.helpers.prompts.models import ImagePart as ImagePartModel
+from phoenix.server.api.helpers.prompts.models import ImageResult as ImageResultModel
 from phoenix.server.api.helpers.prompts.models import (
     PromptChatTemplateV1,
     PromptStringTemplateV1,
@@ -25,8 +25,8 @@
 from phoenix.server.api.helpers.prompts.models import ToolResultPart as ToolResultPartModel
 
 
-@strawberry.experimental.pydantic.type(ImageModel)
-class Image:
+@strawberry.experimental.pydantic.type(ImageResultModel)
+class ImageResult:
     type: str
     url: strawberry.auto
 
@@ -47,7 +47,7 @@ class TextPart:
 @strawberry.experimental.pydantic.type(ImagePartModel)
 class ImagePart:
     type: str
-    image: Image
+    image: ImageResult
 
 
 @strawberry.experimental.pydantic.type(ToolCallPartModel)