import os

from flows.flow_verse import load_class
repository_id = "martinjosifoski/OpenAIChatAtomicFlow"
OpenAIChatAtomicFlow = load_class(repository_id, "OpenAIChatAtomicFlow")


class CodeGenerator(OpenAIChatAtomicFlow):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)