Function run_agent

example/vercel_agent.py:69–154

()

            self.process = None

def run_agent():
    vm = AgentVMClient()
    try:
        vm.start()
        
        # Define tools
        tools = [
            {
                "type": "function",
                "function": {
                    "name": "execute_shell_command",
                    "description": "Execute a shell command in a sandboxed Linux environment. Use this to run Python scripts, check files, or perform calculations.",
                    "parameters": {
                        "type": "object",
                        "properties": {
                            "command": {
                                "type": "string",
                                "description": "The shell command to execute (e.g., \"ls -la\", \"python3 -c ...\")"
                            }
                        },
                        "required": ["command"]
                    }
                }
            }
        ]

        # Check for API Key
        api_key = os.environ.get("OPENAI_API_KEY")
        
        if api_key:
            client = OpenAI(api_key=api_key)
            messages = [
                {"role": "user", "content": "Calculate the 10th Fibonacci number using Python and tell me the result."}
            ]

            MAX_STEPS = 5
            step = 0

            while step < MAX_STEPS:
                step += 1
                response = client.chat.completions.create(
                    model="gpt-4-turbo",
                    messages=messages,
                    tools=tools,
                    tool_choice="auto" 
                )
                
                message = response.choices[0].message
                messages.append(message)
                
                if not message.tool_calls:
                    print("\nAgent Response:")
                    print(message.content)
                    break
                
                for tool_call in message.tool_calls:
                    if tool_call.function.name == "execute_shell_command":
                        args = json.loads(tool_call.function.arguments)