From 9eba370afb3541ece151de610df259f84469ee64 Mon Sep 17 00:00:00 2001
From: Dan D'Avella <dan.davella@pixee.ai>
Date: Wed, 24 Apr 2024 10:39:14 -0400
Subject: [PATCH] Add `ai` field to result

---
 codetf.schema.json | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/codetf.schema.json b/codetf.schema.json
index 2e89063..1b54ae2 100644
--- a/codetf.schema.json
+++ b/codetf.schema.json
@@ -152,6 +152,10 @@
           "type": "array",
           "items": { "$ref": "#/definitions/change" },
           "minItems": 1
+        },
+        "ai": {
+          "description": "Metadata about AI tool used to generate the fix",
+          "$ref": "#/definitions/ai"
         }
       },
       "required": ["path", "diff", "changes"]
@@ -212,6 +216,24 @@
       "required": ["action", "result", "package"]
     },
 
+    "ai": {
+      "type": "object",
+      "properties": {
+        "provider": {
+          "type": "string",
+          "description": "Name of the AI provider used to generate this fix"
+        },
+        "model": {
+          "type": "string",
+          "description": "Name of the model used to generate this fix"
+        },
+        "tokens": {
+          "type": "integer",
+          "description": "Total number of tokens used to generate the fix"
+        }
+      }
+    },
+
     "detectionTool": {
       "type": "object",
       "properties": {