From 726f1a3185c7b2f8e022447d8a03997e50ce6ca6 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil@katanemo.com>
Date: Mon, 25 Nov 2024 17:51:25 -0800
Subject: [PATCH] add schema change to use enum in arch_config (#304)

---
 arch/arch_config_schema.yaml    | 4 ++++
 demos/hr_agent/arch_config.yaml | 3 ++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/arch/arch_config_schema.yaml b/arch/arch_config_schema.yaml
index 6287661d..a11e9562 100644
--- a/arch/arch_config_schema.yaml
+++ b/arch/arch_config_schema.yaml
@@ -88,6 +88,10 @@ properties:
                 type: string
               type:
                 type: string
+              enum:
+                type: array
+                items:
+                  type: string
             additionalProperties: false
             required:
               - name
diff --git a/demos/hr_agent/arch_config.yaml b/demos/hr_agent/arch_config.yaml
index ea0cfa8f..cde49627 100644
--- a/demos/hr_agent/arch_config.yaml
+++ b/demos/hr_agent/arch_config.yaml
@@ -30,7 +30,7 @@ system_prompt: |
 
 prompt_targets:
     - name: workforce
-      description: Get workforce data like headcount and satisfacton levels by region and staffing type
+      description: Get workforce data like headcount and satisfaction levels by region and staffing type
       endpoint:
         name: app_server
         path: /agent/workforce
@@ -39,6 +39,7 @@ prompt_targets:
           type: str
           description: specific category or nature of employment used by an organization like fte, contract and agency
           required: true
+          enum: [fte, contract, agency]
         - name: region
           type: str
           required: true