mirror of
https://github.com/simstudioai/sim.git
synced 2026-09-24 15:45:35 +08:00
Added code execution functionality for function block
This commit is contained in:
@@ -0,0 +1,85 @@
|
||||
import { ToolConfig, ToolResponse } from '../types'
|
||||
|
||||
interface CodeExecutionInput {
|
||||
code: Array<{content: string, id: string}> | string
|
||||
input?: Record<string, any>
|
||||
}
|
||||
|
||||
interface CodeExecutionOutput extends ToolResponse {
|
||||
output: Record<string, any>
|
||||
}
|
||||
|
||||
export const functionExecuteTool: ToolConfig<CodeExecutionInput, CodeExecutionOutput> = {
|
||||
id: 'function.execute',
|
||||
name: 'Function Execute',
|
||||
description: 'Execute code in a sandboxed environment',
|
||||
version: '1.0.0',
|
||||
|
||||
params: {
|
||||
code: {
|
||||
type: 'string',
|
||||
required: true,
|
||||
description: 'The code to execute',
|
||||
}
|
||||
},
|
||||
|
||||
request: {
|
||||
url: 'https://emkc.org/api/v2/piston/execute',
|
||||
method: 'POST',
|
||||
headers: () => ({
|
||||
'Content-Type': 'application/json',
|
||||
'Accept': 'application/json'
|
||||
}),
|
||||
body: (params) => {
|
||||
const codeContent = Array.isArray(params.code)
|
||||
? params.code.map(c => c.content).join('\n')
|
||||
: params.code;
|
||||
|
||||
return {
|
||||
language: 'js',
|
||||
version: '*',
|
||||
files: [{
|
||||
name: 'code.js',
|
||||
content: codeContent
|
||||
}],
|
||||
stdin: '',
|
||||
args: [],
|
||||
compile_timeout: 10000,
|
||||
run_timeout: 3000,
|
||||
compile_memory_limit: -1,
|
||||
run_memory_limit: -1
|
||||
};
|
||||
},
|
||||
},
|
||||
|
||||
transformResponse: async (response) => {
|
||||
const result = await response.json();
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(result.message || 'Execution failed');
|
||||
}
|
||||
|
||||
if (result.run?.stderr) {
|
||||
throw new Error(result.run.stderr);
|
||||
}
|
||||
|
||||
const stdout = result.run?.stdout || '';
|
||||
|
||||
try {
|
||||
// Try parsing the output as JSON
|
||||
const parsed = JSON.parse(stdout);
|
||||
return { output: parsed };
|
||||
} catch {
|
||||
// If not JSON, wrap it in a JSON object
|
||||
return {
|
||||
output: {
|
||||
result: stdout
|
||||
}
|
||||
};
|
||||
}
|
||||
},
|
||||
|
||||
transformError: (error: any) => {
|
||||
return error.message || 'Code execution failed';
|
||||
},
|
||||
}
|
||||
Reference in New Issue
Block a user