Update run.py

run.py

@@ -18,27 +18,24 @@ logging.set_verbosity_debug()
 
 
 if __name__ == "__main__":
-    # ~~~ Set the API information ~~~
-    # OpenAI backend
-    # api_information = ApiInfo("openai", os.getenv("OPENAI_API_KEY"))
-    # Azure backend
-    api_information = ApiInfo("azure", os.getenv("AZURE_OPENAI_KEY"), os.getenv("AZURE_OPENAI_ENDPOINT"))
+    
 
     # ~~~ Instantiate the Flow ~~~
     root_dir = "."
-    cfg_path = os.path.join(root_dir, "FlowName.yaml")
+    cfg_path = os.path.join(root_dir, "demo.yaml")
     cfg = read_yaml_file(cfg_path)
 
+    # ~~~ Instantiate the Flow ~~~
     flow_with_interfaces = {
         "flow": hydra.utils.instantiate(cfg['flow'], _recursive_=False, _convert_="partial"),
         "input_interface": (
             None
-            if getattr(cfg, "input_interface", None) is None
+            if cfg.get( "input_interface", None) is None
             else hydra.utils.instantiate(cfg['input_interface'], _recursive_=False)
         ),
         "output_interface": (
             None
-            if getattr(cfg, "output_interface", None) is None
+            if cfg.get( "output_interface", None) is None
             else hydra.utils.instantiate(cfg['output_interface'], _recursive_=False)
         ),
     }
@@ -55,7 +52,6 @@ if __name__ == "__main__":
         flow_with_interfaces=flow_with_interfaces,
         data=data,
         path_to_output_file=path_to_output_file,
-        api_information=api_information,
     )
 
     # ~~~ Print the output ~~~