Adds class section

flarebyte · Jan 8, 2025 · 0c6a3a5 · 0c6a3a5
1 parent d8c0569
commit 0c6a3a5
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/src/incy-wincy-model.ts b/src/incy-wincy-model.ts
@@ -1,4 +1,4 @@
-import {describeEnum} from 'faora-kai';
+import { describeEnum, stringEffectFields, stringFields } from 'faora-kai';
 import { z } from 'zod';
 
 const programmingLanguageKey = ['js', 'ts', 'dart', 'python'] as const;
@@ -37,3 +37,19 @@ const blockKindlabels = {
 const blockKind = z
   .enum(blockKindKey)
   .describe(describeEnum('Language block kind:', blockKindlabels));
+
+const codeSection = {
+  name: stringFields.string1To80.describe('Name of the class'),
+  body: stringFields.string1To1000.describe('The full code of the block'),
+  uncommentedBody: stringFields.string1To1000.describe(
+    'The code of the block with comments removed'
+  ),
+  keywords: stringFields.string1To1000
+    .optional()
+    .describe('Sorted keywords from the code'),
+};
+
+const classSection = z.object({
+  kind: z.literal('class').describe('Class'),
+  ...codeSection,
+});