AI Agent Compute Platform

// npm install openai @cognitora/sdk
import OpenAI from 'openai';
import { Cognitora } from '@cognitora/sdk';

const openai = new OpenAI({ apiKey: process.env.OPENAI_API_KEY });
const cognitora = new Cognitora({ 
  apiKey: process.env.COGNITORA_API_KEY,
  baseURL: 'https://api.cognitora.dev'
});

async function createCodeInterpreter() {
  // Create a persistent code interpreter session
  const session = await cognitora.codeInterpreter.createSession({
    language: 'python',
    timeout_minutes: 30
  });

  // Define a code execution tool
  const codeExecutionTool = {
    type: "function",
    function: {
      name: "execute_code",
      description: "Execute Python code in a secure sandbox",
      parameters: {
        type: "object",
        properties: {
          code: {
            type: "string", 
            description: "Python code to execute"
          }
        },
        required: ["code"]
      }
    }
  };

  return { session, codeExecutionTool };
}

async function runAICodeInterpreter(userQuery: string) {
  const { session, codeExecutionTool } = await createCodeInterpreter();
  
  const response = await openai.chat.completions.create({
    model: "gpt-4",
    messages: [
      {
        role: "system",
        content: "You are a Python expert. Write and execute code to solve problems."
      },
      {
        role: "user", 
        content: userQuery
      }
    ],
    tools: [codeExecutionTool],
    tool_choice: "auto"
  });

  const toolCall = response.choices[0].message.tool_calls?.[0];
  
  if (toolCall && toolCall.function.name === "execute_code") {
    const { code } = JSON.parse(toolCall.function.arguments);
    
    // Execute code in Cognitora sandbox
    const execution = await cognitora.codeInterpreter.execute({
      code,
      language: 'python',
      session_id: session.data.session_id
    });
    
    return {
      code,
      result: execution.data.outputs,
      status: execution.data.status,
      session_id: session.data.session_id
    };
  }
}

// Example usage
runAICodeInterpreter("Calculate the fibonacci sequence up to 20 terms")
  .then(result => console.log(result));