renamed flow to aiflows

ChatWithDemonstrationsFlow.py

@@ -1,7 +1,7 @@
 
 
-from flows.base_flows import SequentialFlow
-from flows.utils import logging
+from aiflows.base_flows import SequentialFlow
+from aiflows.utils import logging
 
 logging.set_verbosity_debug()
 

ChatWithDemonstrationsFlow.yaml

@@ -3,25 +3,25 @@ description: "A sequential flow that answers questions with demonstrations"
 
 subflows_config:
   demonstration_flow:
-    _target_: aiflows.ChatWithDemonstrationsFlowModule.DemonstrationsAtomicFlow.instantiate_from_default_config
+    _target_: flow_modules.aiflows.ChatWithDemonstrationsFlowModule.DemonstrationsAtomicFlow.instantiate_from_default_config
     
   chat_flow:
-    _target_: aiflows.ChatFlowModule.ChatAtomicFlow.instantiate_from_default_config
+    _target_: flow_modules.aiflows.ChatFlowModule.ChatAtomicFlow.instantiate_from_default_config
 
 topology:
   - goal: Get Demonstrations
     input_interface:
-      _target_: flows.interfaces.KeyInterface
+      _target_: aiflows.interfaces.KeyInterface
     flow: demonstration_flow
     output_interface:
-      _target_: flows.interfaces.KeyInterface
+      _target_: aiflows.interfaces.KeyInterface
 
   - goal: Answer the question
     input_interface:
-      _target_: flows.interfaces.KeyInterface
+      _target_: aiflows.interfaces.KeyInterface
     flow: chat_flow
     output_interface:
-      _target_: flows.interfaces.KeyInterface
+      _target_: aiflows.interfaces.KeyInterface
       keys_to_rename:
         api_output: answer  # Rename the api_output to answer
 

DemonstrationsAtomicFlow.py

@@ -1,9 +1,9 @@
 import jinja2
-from flows.base_flows import AtomicFlow
-from flows.utils import logging
-from flows.utils import general_helpers
+from aiflows.base_flows import AtomicFlow
+from aiflows.utils import logging
+from aiflows.utils import general_helpers
 from typing import Dict,Any,Optional,List
-from flows.prompt_template import JinjaPrompt
+from aiflows.prompt_template import JinjaPrompt
 from copy import deepcopy
 import os
 import hydra

DemonstrationsAtomicFlow.yaml

@@ -12,7 +12,7 @@ params:
   demonstrations_k: null
 
 query_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
+  _target_: aiflows.prompt_template.JinjaPrompt
 
 response_prompt_template:
-  _target_: flows.prompt_template.JinjaPrompt
+  _target_: aiflows.prompt_template.JinjaPrompt

README.md

@@ -48,11 +48,11 @@ Default: No default value this field must be set.
     - `demonstrations_k` (int): The number of demonstrations to pass to the ChatFlow.
     If None, all the demonstrations are passed to the ChatFlow. Default: None
 - `query_prompt_template` (Dict[str, Any]): The prompt template used to generate the query of the demonstrations. 
-By default its of type flows.prompt_template.JinjaPrompt. None of the parameters of the prompt are defined by default and therefore need to be defined if one
-wants to use the query_prompt_template. Default parameters are defined in flows.prompt_template.jinja2_prompts.JinjaPrompt.
-- `response_prompt_template` (Dict[str, Any]): The prompt template used to generate the response of the demonstrations. By default its of type flows.prompt_template.JinjaPrompt.
+By default its of type aiflows.prompt_template.JinjaPrompt. None of the parameters of the prompt are defined by default and therefore need to be defined if one
+wants to use the query_prompt_template. Default parameters are defined in aiflows.prompt_template.jinja2_prompts.JinjaPrompt.
+- `response_prompt_template` (Dict[str, Any]): The prompt template used to generate the response of the demonstrations. By default its of type aiflows.prompt_template.JinjaPrompt.
 None of the parameters of the prompt are defined by default and therefore need to be defined if one
-wants to use the response_prompt_template. Default parameters are defined in flows.prompt_template.jinja2_prompts.JinjaPrompt.
+wants to use the response_prompt_template. Default parameters are defined in aiflows.prompt_template.jinja2_prompts.JinjaPrompt.
 
 *Input Interface*:
 

__init__.py

@@ -2,7 +2,7 @@
 dependencies = [
     {"url": "aiflows/ChatFlowModule", "revision": "main"},
 ]
-from flows import flow_verse
+from aiflows import flow_verse
 flow_verse.sync_dependencies(dependencies)
 from .ChatWithDemonstrationsFlow import ChatWithDemonstrationsFlow
 from .DemonstrationsAtomicFlow import DemonstrationsAtomicFlow

demo.yaml

@@ -1,5 +1,5 @@
 flow:  # Overrides the OpenAIChatAtomicFlow config
-  _target_: aiflows.ChatWithDemonstrationsFlowModule.ChatWithDemonstrationsFlow.instantiate_from_default_config
+  _target_: flow_modules.aiflows.ChatWithDemonstrationsFlowModule.ChatWithDemonstrationsFlow.instantiate_from_default_config
   name: "SimpleQA_Flow_with_Demonstrations"
   description: "A sequential flow that answers questions with demonstrations"
   
@@ -41,7 +41,7 @@ flow:  # Overrides the OpenAIChatAtomicFlow config
 
       # ~~~ backend model parameters ~~
       backend:
-        _target_: flows.backends.llm_lite.LiteLLMBackend
+        _target_: aiflows.backends.llm_lite.LiteLLMBackend
         api_infos: ???
         model_name: 
           openai: "gpt-3.5-turbo"
@@ -61,7 +61,7 @@ flow:  # Overrides the OpenAIChatAtomicFlow config
 
       # ~~~ Prompt specification ~~~
       system_message_prompt_template:
-        _target_: flows.prompt_template.JinjaPrompt
+        _target_: aiflows.prompt_template.JinjaPrompt
         template: |2-
           You are a helpful chatbot that truthfully answers questions. Answer in a similar way to your previous replies.
         input_variables: []
@@ -69,7 +69,7 @@ flow:  # Overrides the OpenAIChatAtomicFlow config
  
 
       init_human_message_prompt_template:
-        _target_: flows.prompt_template.JinjaPrompt
+        _target_: aiflows.prompt_template.JinjaPrompt
         template: |2-
           Answer the following question: {{question}}
         input_variables: ["question"]

run.py

@@ -2,20 +2,20 @@ import os
 
 import hydra
 
-import flows
-from flows.flow_launchers import FlowLauncher
-from flows.backends.api_info import ApiInfo
-from flows.utils.general_helpers import read_yaml_file
+import aiflows
+from aiflows.flow_launchers import FlowLauncher
+from aiflows.backends.api_info import ApiInfo
+from aiflows.utils.general_helpers import read_yaml_file
 
-from flows import logging
-from flows.flow_cache import CACHING_PARAMETERS, clear_cache
+from aiflows import logging
+from aiflows.flow_cache import CACHING_PARAMETERS, clear_cache
 
 CACHING_PARAMETERS.do_caching = False  # Set to True in order to disable caching
 # clear_cache() # Uncomment this line to clear the cache
 
 logging.set_verbosity_debug()  # Uncomment this line to see verbose logs
 
-from flows import flow_verse
+from aiflows import flow_verse
 
 
 dependencies = [