🛐

src/llm.ts

@@ -209,8 +209,16 @@ export class OpenAIChatModelBuilder<
           : z.object(this._options.output)
 
       const { node } = zodToTs(outputSchema)
+
+      if (node.kind === 152) {
+        // ignore raw strings
+        messages.push({
+          role: 'system',
+          content: dedent`Output a string`
+        })
+      } else {
         const tsTypeString = printNode(node, {
-        removeComments: true,
+          removeComments: false,
           // TODO: this doesn't seem to actually work, so we're doing it manually below
           omitTrailingSemicolon: true,
           noEmitHelpers: true
@@ -226,6 +234,7 @@ export class OpenAIChatModelBuilder<
           \`\`\``
         })
       }
+    }
 
     // TODO: filter/compress messages based on token counts
 

src/temp.ts

@@ -18,8 +18,9 @@ async function main() {
   //   .call()
   // console.log(ex0)
 
-  const ex1 = await $.gpt4(`give me fake data conforming to this schema`)
+  const ex1 = await $.gpt4(`give me fake data`)
     .output(z.object({ foo: z.string(), bar: z.number() }))
+    // .output(z.string())
     // .retry({ attempts: 3 })
     .call()
   console.log(ex1)