From 755331ea16929ab544a9c78943c1ed4f53c08fe4 Mon Sep 17 00:00:00 2001
From: Artem Chumachenko <artek.chumak@gmail.com>
Date: Fri, 15 Nov 2024 15:58:39 +0100
Subject: [PATCH] add flag

---
 openapi.yaml | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/openapi.yaml b/openapi.yaml
index a7a85d4..b0f9754 100644
--- a/openapi.yaml
+++ b/openapi.yaml
@@ -419,6 +419,15 @@ paths:
                 wandb_api_key:
                   type: string
                   description: API key for Weights & Biases integration
+                train_on_inputs:
+                  oneOf:
+                    - type: boolean
+                    - type: string
+                      enum:
+                        - auto
+                  type: boolean
+                  default: auto
+                  description: Whether to mask the user messages in conversational data or prompts in instruction data.
                 training_type:
                   type: object
                   oneOf:
@@ -1775,6 +1784,13 @@ components:
           type: number
         eval_steps:
           type: integer
+        train_on_inputs:
+          oneOf:
+            - type: boolean
+            - type: string
+              enum:
+                - auto
+          default: auto
         training_type:
           type: object
           oneOf: