@freestyle-sh/with-uv 0.1.0 → 0.2.1

package/README.md

@@ -20,9 +20,9 @@ const { vm } = await freestyle.vms.create({
   },
 });
 
-const res = await vm.uv.runCode(
-  "import json; print(json.dumps({ 'hello': 'world' }))"
-);
+const res = await vm.uv.runCode({
+  code: "import json; print(json.dumps({ 'hello': 'world' }))"
+});
 
 console.log(res);
 // { result: { hello: 'world' }, stdout: '{"hello": "world"}\n', statusCode: 0 }
@@ -44,7 +44,7 @@ new VmUv({
 
 ## API
 
-### `vm.uv.runCode(code: string)`
+### `vm.uv.runCode({ code: string })`
 
 Executes Python code using uv's managed Python runtime.
 

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { VmWith, VmWithInstance, CreateVmOptions } from 'freestyle-sandboxes';
+import { VmWith, VmWithInstance, VmSpec } from 'freestyle-sandboxes';
 import { VmRunCodeInstance, JSONValue, RunCodeResponse, VmRunCode } from '@freestyle-sh/with-type-run-code';
 
 type UvOptions = {
@@ -12,14 +12,16 @@ type UvResolvedOptions = {
 declare class VmUv extends VmWith<VmUvInstance> implements VmRunCode<VmRunCodeInstance> {
     options: UvResolvedOptions;
     constructor(options?: UvOptions);
-    configure(existingConfig: CreateVmOptions): CreateVmOptions | Promise<CreateVmOptions>;
+    configureSnapshotSpec(spec: VmSpec): VmSpec;
     createInstance(): VmUvInstance;
     installServiceName(): string;
 }
 declare class VmUvInstance extends VmWithInstance implements VmRunCodeInstance {
     builder: VmUv;
     constructor(builder: VmUv);
-    runCode<Result extends JSONValue = any>(code: string): Promise<RunCodeResponse<Result>>;
+    runCode<Result extends JSONValue = any>(args: string | {
+        code: string;
+    }): Promise<RunCodeResponse<Result>>;
 }
 
 export { VmUv };

package/dist/index.js

@@ -1,4 +1,4 @@
-import { VmWith, VmTemplate, VmWithInstance } from 'freestyle-sandboxes';
+import { VmWith, VmSpec, VmWithInstance } from 'freestyle-sandboxes';
 
 class VmUv extends VmWith {
   options;
@@ -9,7 +9,7 @@ class VmUv extends VmWith {
       pythonVersion: options?.pythonVersion ?? "3.14"
     };
   }
-  configure(existingConfig) {
+  configureSnapshotSpec(spec) {
     const versionArg = this.options.version ? `UV_VERSION="${this.options.version}" ` : "";
     const installScript = `#!/bin/bash
 set -e
@@ -20,8 +20,9 @@ ${versionArg}curl -LsSf https://astral.sh/uv/install.sh | sh
 `;
     const uvInit = `export PATH="/opt/uv/bin:$PATH"
 `;
-    const uvConfig = {
-      template: new VmTemplate({
+    return this.composeSpecs(
+      spec,
+      new VmSpec({
         additionalFiles: {
           "/opt/install-uv.sh": {
             content: installScript
@@ -42,8 +43,7 @@ ${versionArg}curl -LsSf https://astral.sh/uv/install.sh | sh
           ]
         }
       })
-    };
-    return this.compose(existingConfig, uvConfig);
+    );
   }
   createInstance() {
     return new VmUvInstance(this);
@@ -58,7 +58,8 @@ class VmUvInstance extends VmWithInstance {
     super();
     this.builder = builder;
   }
-  async runCode(code) {
+  async runCode(args) {
+    const code = typeof args === "string" ? args : args.code;
     const result = await this.vm.exec({
       command: `/opt/uv/bin/uv run python -c "${code.replace(/"/g, '\\"')}"`
     });

package/package.json

@@ -1,10 +1,10 @@
 {
   "name": "@freestyle-sh/with-uv",
-  "version": "0.1.0",
+  "version": "0.2.1",
   "private": false,
   "dependencies": {
-    "freestyle-sandboxes": "^0.1.2",
-    "@freestyle-sh/with-type-run-code": "^0.1.0"
+    "freestyle-sandboxes": "^0.1.8",
+    "@freestyle-sh/with-type-run-code": "^0.2.1"
   },
   "type": "module",
   "main": "./dist/index.js",