Fixed nested description (#7)

Fix estimate for nested descriptions

---------

Co-authored-by: Piotr Kuczko <[email protected]>

Author: piotrkuczko

.gitignore

@@ -1,2 +1,5 @@
+.*
+!.gitignore
+!.github
 node_modules/
 dist/

src/functions.ts

@@ -66,7 +66,7 @@ export function formatFunctionDefinitions(functions: FunctionDef[]) {
 function formatObjectProperties(obj: ObjectProp, indent: number): string {
   const lines = [];
   for (const [name, param] of Object.entries(obj.properties ?? {})) {
-    if (param.description) {
+    if (param.description && indent < 2) {
       lines.push(`// ${param.description}`);
     }
     if (obj.required?.includes(name)) {

tests/token-counts.test.ts

@@ -304,6 +304,83 @@ const TEST_CASES: Example[] = [
     ],
     tokens: 106,
   },
+  {
+    messages: [{ role: 'user', content: 'hello' }],
+    functions: [
+      {
+        name: 'function',
+        description: 'description',
+        parameters: {
+          type: 'object',
+          properties: {
+            quality: {
+              type: 'object',
+              properties: {
+                pros: {
+                  type: 'array',
+                  items: {
+                    type: 'string',
+                  },
+                  description: 'Write 3 points why this text is well written',
+                }
+              },
+            }
+          },
+        }
+      }],
+    tokens: 46,
+  },
+  {
+    messages: [{ role: 'user', content: 'hello' }],
+    functions: [
+      {
+        name: 'function',
+        description: 'desctiption1',
+        parameters: {
+          type: 'object',
+          description: 'desctiption2',
+          properties: {
+            mainField: {
+              type: 'string',
+              description: 'description3',
+            },
+            'field number one': {
+              type: 'object',
+              description: 'description4',
+              properties: {
+                yesNoField: {
+                  type: 'string',
+                  description: 'description5',
+                  enum: [
+                    'Yes',
+                    'No',
+                  ],
+                },
+                howIsInteresting: {
+                  type: 'string',
+                  description: 'description6',
+                },
+                scoreInteresting: {
+                  type: 'number',
+                  description: 'description7',
+                },
+                isInteresting: {
+                  type: 'string',
+                  description: 'description8',
+                  enum: [
+                    'Yes',
+                    'No',
+                  ],
+                },
+              },
+            },
+          },
+
+        }
+      }
+    ],
+    tokens: 96,
+  }
 ];
 
 const validateAll = false;