Update CodeGeneratorAtomicFlow.py

CodeGeneratorAtomicFlow.py

@@ -1,12 +1,12 @@
 import json
 from copy import deepcopy
 from typing import Any, Dict
-from flow_modules.aiflows.OpenAIChatFlowModule import OpenAIChatAtomicFlow
+from flow_modules.aiflows.ChatFlowModule import ChatAtomicFlow
 
 from dataclasses import dataclass
 
 
-class CodeGeneratorAtomicFlow(OpenAIChatAtomicFlow):
+class CodeGeneratorAtomicFlow(ChatAtomicFlow):
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
         self.hint_for_model = """