From ab263fc0d4974f49ff4e9e36f0b2d46995daeb41 Mon Sep 17 00:00:00 2001
From: jmansdorfer <jmansdorfer19@gmail.com>
Date: Tue, 16 Jul 2024 13:29:25 -0400
Subject: [PATCH] adding input to streaming and input/output to vision

---
 ...ard-Prediction-Guard-API-1.0-resolved.yaml | 45 ++++++++++++++++++-
 1 file changed, 43 insertions(+), 2 deletions(-)

diff --git a/fern/openapi/Prediction-Guard-Prediction-Guard-API-1.0-resolved.yaml b/fern/openapi/Prediction-Guard-Prediction-Guard-API-1.0-resolved.yaml
index f643e5d..cba0959 100644
--- a/fern/openapi/Prediction-Guard-Prediction-Guard-API-1.0-resolved.yaml
+++ b/fern/openapi/Prediction-Guard-Prediction-Guard-API-1.0-resolved.yaml
@@ -283,7 +283,13 @@ paths:
                   temperature: 1.0,
                   top_p: 1.0,
                   top_k: 50,
-                  stream: true
+                  stream: true,
+                  input: [
+                    {
+                      pii: "replace",
+                      pii_replace_method: "random"
+                    }
+                  ]
                 }
       responses:
         '403':
@@ -462,6 +468,29 @@ paths:
                 top_k:
                   type: integer
                   description: The diversity of the generated text based on top-k sampling.
+                output:
+                  type: object
+                  description: Options to affect the output of the response.
+                  properties:
+                    factuality:
+                      type: boolean
+                      description: Set to true to turn on factuality processing.
+                    toxicity:
+                      type: boolean
+                      description: Set to true to turn on toxicity processing.
+                input:
+                  type: object
+                  description: Options to affect the input of the request.
+                  properties:
+                    block_prompt_injection:
+                      type: boolean
+                      description: Set to true to turn on factuality processing.
+                    pii:
+                      type: string
+                      description: Set to either 'block' or 'replace'.
+                    pii_replace_method:
+                      type: string
+                      description: Set to either 'random', 'fake', 'category', 'mask'.
             examples:
               basic:
                 summary: A basic example of using the API.
@@ -487,7 +516,19 @@ paths:
                   max_tokens: 1000,
                   temperature: 1.0,
                   top_p: 1.0,
-                  top_k: 50
+                  top_k: 50,
+                  output: [
+                    {
+                      factuality: true,
+                      toxicity: true
+                    }
+                  ],
+                  input: [
+                    {
+                      pii: "replace",
+                      pii_replace_method: "random"
+                    }
+                  ]
                 }
       responses:
         '403':