Function load_image_generation_tool

host_local_tools.py:121–125

()

        server.load_tool("shell")

    def load_image_generation_tool():
        STEAMSHIP_API_KEY = os.environ.get('STEAMSHIP_API_KEY', None)
        if not STEAMSHIP_API_KEY:
            raise RuntimeError("STEAMSHIP_API_KEY is not provided. Please sign up for a free account at https://steamship.com/account/api, create a new API key, and add it to environment variables.")
        server.load_tool("image_generation")

    def load_hugging_tools():
        HUGGINGFACE_API_KEY = os.environ.get('HUGGINGFACE_API_KEY', None)