| import os |
|
|
| import hydra |
|
|
| from aiflows.backends.api_info import ApiInfo |
| from aiflows.messages import InputMessage |
| from aiflows.utils.general_helpers import read_yaml_file |
|
|
| from aiflows import logging |
| from aiflows.flow_cache import CACHING_PARAMETERS, clear_cache |
|
|
| CACHING_PARAMETERS.do_caching = False |
|
|
| logging.set_verbosity_debug() |
| logging.auto_set_dir() |
|
|
| dependencies = [ |
| {"url": "aiflows/MemoryReadingFlowModule", "revision": "main"}, |
| ] |
|
|
| from aiflows import flow_verse |
|
|
| flow_verse.sync_dependencies(dependencies) |
|
|
| if __name__ == "__main__": |
| current_dir = os.getcwd() |
| cfg_path = os.path.join(current_dir, "MemoryReadingAtomicFlow.yaml") |
| cfg = read_yaml_file(cfg_path) |
|
|
| MemReadFlow = hydra.utils.instantiate(cfg, _recursive_=False, _convert_="partial") |
|
|
| input_data = { |
| "memory_files": { |
| "plan": os.path.join(current_dir, "example_plan.txt"), |
| "logs": os.path.join(current_dir, "example_logs.txt"), |
| "code_library": os.path.join(current_dir, "example_library.py") |
| } |
| } |
| input_message = InputMessage.build( |
| data_dict=input_data, |
| src_flow="Launcher", |
| dst_flow=MemReadFlow.name |
| ) |
|
|
| |
| output_message = MemReadFlow(input_message) |
|
|
| print(output_message.data) |
