feat: add response_format to invocation parameters

Arize-ai · Oct 31, 2024 · 0ac67c2 · 0ac67c2
1 parent 6330b20
commit 0ac67c2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/app/schema.graphql b/app/schema.graphql
@@ -936,6 +936,7 @@ input InvocationParameters {
   stop: [String!]
   seed: Int
   toolChoice: JSON
+  responseFormat: JSON
 }
 
 """

diff --git a/app/src/pages/playground/__generated__/PlaygroundOutputSubscription.graphql.ts b/app/src/pages/playground/__generated__/PlaygroundOutputSubscription.graphql.ts
diff --git a/src/phoenix/server/api/input_types/InvocationParameters.py b/src/phoenix/server/api/input_types/InvocationParameters.py
@@ -18,3 +18,4 @@ class InvocationParameters:
     stop: Optional[list[str]] = UNSET
     seed: Optional[int] = UNSET
     tool_choice: Optional[JSON] = UNSET
+    response_format: Optional[JSON] = UNSET