python_tool.h

#pragma once
 
#include <agents-cpp/tool.h>
 
namespace agents {
namespace tools {

class PythonTool : public Tool {
public:
    PythonTool();

    ToolResult execute(const JsonObject& params) const override;

protected:
    void setupParameters();

    bool validatePythonCode(const String& code) const;

    bool isDangerousPythonCode(const String& code) const;

    ToolResult executePythonCode(const String& code) const;

    ToolResult formatPythonResult(const String& code, const String& output, bool success, const String& error = "") const;
private:
    static bool pythonInitialized;

    static void initializePython();

    static void finalizePython();
};
 
} // namespace tools
} // namespace agents