build(deps): bump @codesandbox/sdk from 0.6.2 to 2.2.0 (#1652)

* build(deps): bump @codesandbox/sdk from 0.6.2 to 2.2.0

Bumps [@codesandbox/sdk](https://github.com/codesandbox/codesandbox-sdk) from 0.6.2 to 2.2.0.
- [Release notes](https://github.com/codesandbox/codesandbox-sdk/releases)
- [Changelog](https://github.com/codesandbox/codesandbox-sdk/blob/main/CHANGELOG.md)
- [Commits](codesandbox/codesandbox-sdk@v0.6.2...v2.2.0)

---
updated-dependencies:
- dependency-name: "@codesandbox/sdk"
  dependency-version: 2.2.0
  dependency-type: direct:production
  update-type: version-update:semver-major
...

Signed-off-by: dependabot[bot] <support@github.com>

* chore: adapt code to new sandbox version

---------

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Gerard <clos.gerard@gmail.com>

Author: dependabot[bot]

apps/gateway/src/presenters/documentPresenter.ts

@@ -40,14 +40,14 @@ export function documentPresenterWithProviderAndMetadata({
   const parameters =
     rawParams.length > 0
       ? rawParams.reduce(
-        (acc, rawParam) => {
-          if (acc[rawParam]) return acc
-          acc[rawParam] = { type: ParameterType.Text }
+          (acc, rawParam) => {
+            if (acc[rawParam]) return acc
+            acc[rawParam] = { type: ParameterType.Text }
 
-          return acc
-        },
-        { ...configParams },
-      )
+            return acc
+          },
+          { ...configParams },
+        )
       : configParams
 
   return {

packages/core/package.json

@@ -144,7 +144,7 @@
     "@aws-sdk/client-s3": "3.850.0",
     "@aws-sdk/s3-request-presigner": "3.850.0",
     "@google-cloud/storage": "^7.7.0",
-    "@codesandbox/sdk": "0.6.2",
+    "@codesandbox/sdk": "2.2.1",
     "@kubernetes/client-node": "1.0.0",
     "@latitude-data/compiler": "workspace:^",
     "@latitude-data/constants": "workspace:*",

packages/core/src/services/latitudeTools/runCode/sandbox.ts

@@ -18,8 +18,8 @@ async function createSandbox() {
       return Result.error(new BadRequestError('CODESANDBOX_API_KEY is not set'))
     }
     const sdk = new CodeSandbox(env.CODESANDBOX_API_KEY)
-    const sandbox = await sdk.sandbox.create()
-    return Result.ok(sandbox)
+    const sandbox = await sdk.sandboxes.create()
+    return Result.ok({ sandbox, sdk })
   } catch (error) {
     return Result.error(error as LatitudeError)
   }
@@ -34,7 +34,7 @@ export const withSafeSandbox = async <
 ): Promise<T> => {
   const sandboxResult = await createSandbox()
   if (sandboxResult.error) throw sandboxResult.error
-  const sandbox = sandboxResult.unwrap()
+  const { sandbox, sdk } = sandboxResult.unwrap()
 
   const timeoutPromise = new Promise<T>((resolve) => {
     setTimeout(() => {
@@ -51,6 +51,6 @@ export const withSafeSandbox = async <
   } catch (error) {
     return Result.error(error as LatitudeError) as T
   } finally {
-    sandbox.shutdown()
+    await sdk.sandboxes.shutdown(sandbox.id)
   }
 }

packages/core/src/services/latitudeTools/runCode/withDependencies.ts

@@ -35,25 +35,24 @@ export async function runCodeWithDependencies({
   dependencies,
 }: CodeToolArgs): PromisedResult<CodeRunResult, LatitudeError> {
   return withSafeSandbox(async (sandbox) => {
+    const client = await sandbox.connect()
     const filename = getFilename({ language })
     const buildDependency = getDependencyBuilder({ language })
 
-    await sandbox.fs.writeFile(filename, new TextEncoder().encode(code))
+    await client.fs.writeFile(filename, new TextEncoder().encode(code))
 
     for await (const dep of dependencies!) {
-      const installResult = await sandbox.shells.run(buildDependency(dep))
-      if (installResult.exitCode !== 0) {
+      const installResult = await client.commands.run(buildDependency(dep))
+      if (!installResult) {
         return Result.error(
-          new BadRequestError(
-            `Failed to install dependency: '${dep}'\n${installResult.output}`,
-          ),
+          new BadRequestError(`Failed to install dependency: '${dep}'`),
         )
       }
     }
 
-    const scriptResult = await sandbox.shells.run(
+    const scriptResult = await client.commands.run(
       getRunCommand({ language, filename }),
     )
-    return Result.ok(normalizedResult(scriptResult))
+    return Result.ok(normalizedResult({ output: scriptResult, exitCode: 0 }))
   })
 }

packages/core/src/services/latitudeTools/runCode/withoutDependencies.ts

@@ -9,14 +9,15 @@ export async function runCodeWithoutDependencies({
   language,
 }: CodeToolArgs): PromisedResult<CodeRunResult, LatitudeError> {
   return withSafeSandbox(async (sandbox) => {
+    const client = await sandbox.connect()
     const interpreter = (() => {
-      if (language === 'python') return sandbox.shells.python
-      if (language === 'javascript') return sandbox.shells.js
+      if (language === 'python') return client.interpreters.python
+      if (language === 'javascript') return client.interpreters.javascript
 
       throw new BadRequestError(`Language ${language} is not supported`)
     })()
 
-    const result = await interpreter.run(code)
-    return Result.ok(normalizedResult(result))
+    const result = await interpreter(code)
+    return Result.ok(normalizedResult({ output: result, exitCode: 0 }))
   })
 }

packages/core/src/tests/factories/documentLogs.ts

@@ -66,8 +66,8 @@ async function generateProviderLogs({
         typeof message.content === 'string'
           ? message.content
           : message.content
-            .map((c) => (c.type === 'text' ? c.text : ''))
-            .join('')
+              .map((c) => (c.type === 'text' ? c.text : ''))
+              .join('')
       return acc + content.length
     }, 0)
     const completionTokens = mockedResponse.length
@@ -133,7 +133,7 @@ export async function createDocumentLog({
   const duration =
     totalDuration ??
     Math.floor(Math.random() * 100) +
-    providerLogs.reduce((acc, log) => acc + (log?.duration ?? 0), 0)
+      providerLogs.reduce((acc, log) => acc + (log?.duration ?? 0), 0)
 
   let documentLog = await ogCreateDocumentLog({
     commit,