import fix + dealing w/ response in function of n parameter in generation

OpenAIChatAtomicFlow.py

@@ -13,7 +13,7 @@ from flows.messages.flow_message import UpdateMessage_ChatMessage
 
 from flows.prompt_template import JinjaPrompt
 
-from backends.llm_lite import LiteLLMBackend
+from flows.backends.llm_lite import LiteLLMBackend
 
 log = logging.get_logger(__name__)
 
@@ -256,5 +256,5 @@ class OpenAIChatAtomicFlow(AtomicFlow):
                 role=self.flow_config["assistant_name"],
                 content=answer
             )
-
+        response = response if len(response) > 1 or len(response) == 0 else response[0]
         return {"api_output": response}

OpenAIChatAtomicFlow.yaml

@@ -9,7 +9,7 @@ user_name: user
 assistant_name: assistant
 
 backend:
-  _target_: backends.llm_lite.LiteLLMBackend
+  _target_: flows.backends.llm_lite.LiteLLMBackend
   api_infos: ???
   model_name: "gpt-3.5-turbo"
   n: 1

run.py

@@ -4,7 +4,7 @@ import hydra
 
 import flows
 from flows.flow_launchers import FlowLauncher
-from backends.api_info import ApiInfo
+from flows.backends.api_info import ApiInfo
 from flows.utils.general_helpers import read_yaml_file
 
 from flows import logging