add console logs for reasoning for debugging (#8042)

* add console logs for reasoning for debugging

Signed-off-by: Jessie Frazelle <[email protected]>

* updates

Signed-off-by: Jessie Frazelle <[email protected]>

* updates

Signed-off-by: Jessie Frazelle <[email protected]>

---------

Signed-off-by: Jessie Frazelle <[email protected]>

Author: jessfraz

src/lib/crossPlatformFetch.ts

@@ -1,6 +1,6 @@
-import isomorphicFetch from 'isomorphic-fetch'
-import { isDesktop } from '@src/lib/isDesktop'
 import env from '@src/env'
+import { isDesktop } from '@src/lib/isDesktop'
+import isomorphicFetch from 'isomorphic-fetch'
 
 const headers = (token?: string): HeadersInit => ({
   'Content-Type': 'application/json',

src/lib/promptToEdit.tsx

@@ -29,6 +29,7 @@ import type { File as KittyCadLibFile } from '@kittycad/lib/dist/types/src/model
 import type { FileMeta } from '@src/lib/types'
 import type { RequestedKCLFile } from '@src/machines/systemIO/utils'
 import { withAPIBaseURL, withSiteBaseURL } from '@src/lib/withBaseURL'
+import { connectReasoningStream } from '@src/lib/reasoningWs'
 
 type KclFileMetaMap = {
   [execStateFileNamesIndex: number]: Extract<FileMeta, { type: 'kcl' }>
@@ -98,6 +99,8 @@ async function submitTextToCadRequest(
     return new Error(errorData.message || 'Unknown error')
   }
 
+  connectReasoningStream(token, data.id)
+
   return data as TextToCadMultiFileIteration_type
 }
 

src/lib/reasoningWs.ts

@@ -0,0 +1,51 @@
+// wsReasoning.ts
+// Logs every frame with the async-op `id`, closes on {"type":"end_of_stream"}
+
+import { withWebSocketURL } from '@src/lib/withBaseURL'
+
+export function connectReasoningStream(token: string, id: string): void {
+  const url = withWebSocketURL('').replace(
+    '/ws/modeling/commands',
+    `/ws/ml/reasoning/${id}`
+  )
+  const ws = new WebSocket(url)
+  const authMessage = {
+    type: 'headers',
+    headers: {
+      Authorization: `Bearer ${token}`,
+    },
+  }
+
+  ws.addEventListener('open', () => {
+    console.log(`[${id}] open ${url}`)
+    ws.send(JSON.stringify(authMessage)) // 🔸 send immediately
+    console.log(`[${id}] →`, authMessage)
+  })
+
+  ws.addEventListener('message', (ev) => {
+    console.log(`[${id}] raw`, ev.data)
+
+    let msg: unknown
+    try {
+      msg = JSON.parse(ev.data as string)
+    } catch {
+      console.error(`[${id}] JSON parse error`, ev.data)
+      return // non-JSON frame
+    }
+
+    if ('error' in (msg as any)) {
+      ws.send(JSON.stringify(authMessage)) // 🔸 send immediately
+      console.log(`[${id}] →`, authMessage)
+    }
+
+    if ('end_of_stream' in (msg as any)) {
+      console.log(`[${id}] end_of_stream → closing`)
+      ws.close(1000, 'done')
+    }
+  })
+
+  ws.addEventListener('close', (e) =>
+    console.log(`[${id}] close`, e.code, e.reason)
+  )
+  ws.addEventListener('error', console.error)
+}

src/lib/textToCad.ts

@@ -1,6 +1,4 @@
 import type { Models } from '@kittycad/lib'
-import toast from 'react-hot-toast'
-import type { NavigateFunction } from 'react-router-dom'
 import {
   ToastTextToCadError,
   ToastTextToCadSuccess,
@@ -9,16 +7,19 @@ import { PROJECT_ENTRYPOINT } from '@src/lib/constants'
 import crossPlatformFetch from '@src/lib/crossPlatformFetch'
 import { getUniqueProjectName } from '@src/lib/desktopFS'
 import { isDesktop } from '@src/lib/isDesktop'
+import { joinOSPaths } from '@src/lib/paths'
+import { connectReasoningStream } from '@src/lib/reasoningWs'
 import { kclManager, systemIOActor } from '@src/lib/singletons'
+import { err, reportRejection } from '@src/lib/trap'
+import { toSync } from '@src/lib/utils'
+import { withAPIBaseURL } from '@src/lib/withBaseURL'
+import { getAllSubDirectoriesAtProjectRoot } from '@src/machines/systemIO/snapshotContext'
 import {
   SystemIOMachineEvents,
   waitForIdleState,
 } from '@src/machines/systemIO/utils'
-import { err, reportRejection } from '@src/lib/trap'
-import { toSync } from '@src/lib/utils'
-import { getAllSubDirectoriesAtProjectRoot } from '@src/machines/systemIO/snapshotContext'
-import { joinOSPaths } from '@src/lib/paths'
-import { withAPIBaseURL } from '@src/lib/withBaseURL'
+import toast from 'react-hot-toast'
+import type { NavigateFunction } from 'react-router-dom'
 
 export async function submitTextToCadPrompt(
   prompt: string,
@@ -51,6 +52,10 @@ export async function submitTextToCadPrompt(
     return new Error('No id returned from Text-to-CAD API')
   }
 
+  if (token) {
+    connectReasoningStream(token, data.id)
+  }
+
   return data
 }