From 27defda03538a57bab0d501e6a76aa13cfa3766d Mon Sep 17 00:00:00 2001
From: stefanfaistenauer <stefan@superglue.cloud>
Date: Wed, 5 Mar 2025 23:15:48 -0800
Subject: [PATCH] added logging to when schema is generated

---
 packages/core/utils/schema.ts | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/packages/core/utils/schema.ts b/packages/core/utils/schema.ts
index 0a3411b..9356d11 100644
--- a/packages/core/utils/schema.ts
+++ b/packages/core/utils/schema.ts
@@ -19,7 +19,9 @@ export async function generateSchema(instruction: string, responseData: string)
 
   while (retryCount <= MAX_RETRIES) {
     try {
-      return await attemptSchemaGeneration(messages, retryCount);
+      const schema = await attemptSchemaGeneration(messages, retryCount);
+      console.log(`Schema generated`);
+      return schema;
     } catch (error) {
       retryCount++;
       if (retryCount > MAX_RETRIES) {
@@ -41,7 +43,7 @@ async function attemptSchemaGeneration(
   messages: ChatCompletionMessageParam[],
   retry: number
 ): Promise<string> {
-  console.log(`Generating schema: ${retry ? `(retry ${retry})` : ""}`);
+  console.log(`Generating schema${retry ? `: (retry ${retry})` : ""}`);
   const openai = new OpenAI({
     apiKey: process.env.OPENAI_API_KEY,
     baseURL: process.env.OPENAI_API_BASE_URL
@@ -72,6 +74,7 @@ async function attemptSchemaGeneration(
     throw new Error("No schema generated");
   }
   const validator = new Validator();
-  const validation = validator.validate({}, generatedSchema);
+  validator.validate({}, generatedSchema);
+
   return generatedSchema;
 }