Update ControllerAtomicFlow.py

ControllerAtomicFlow.py

@@ -4,7 +4,7 @@ import json
 import os.path
 from copy import deepcopy
 from typing import Any, Dict, List
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
 
 from dataclasses import dataclass
 
@@ -16,7 +16,7 @@ class Command:
     input_args: List[str]
 
 
-class ControllerAtomicFlow(OpenAIChatAtomicFlow):
+class ControllerAtomicFlow(ChatAtomicFlow):
     def __init__(
             self,
             commands: List[Command],