diff --git a/instantiation/controller/agent/agent.py b/instantiation/controller/agent/agent.py index f5a96411..ecfabeb3 100644 --- a/instantiation/controller/agent/agent.py +++ b/instantiation/controller/agent/agent.py @@ -3,7 +3,7 @@ from typing import Dict, List -from controller.prompter.agent_prompter import FilterPrompter, PrefillPrompter +from instantiation.controller.prompter.agent_prompter import FilterPrompter, PrefillPrompter from ufo.agents.agent.basic import BasicAgent diff --git a/instantiation/controller/env/env_manager.py b/instantiation/controller/env/env_manager.py index 0c5b8792..51e9056f 100644 --- a/instantiation/controller/env/env_manager.py +++ b/instantiation/controller/env/env_manager.py @@ -2,7 +2,7 @@ import re import time -from config.config import Config +from instantiation.config.config import Config from fuzzywuzzy import fuzz from pywinauto import Desktop diff --git a/instantiation/controller/instantiation_process.py b/instantiation/controller/instantiation_process.py index 23b6046c..e3116af7 100644 --- a/instantiation/controller/instantiation_process.py +++ b/instantiation/controller/instantiation_process.py @@ -7,7 +7,7 @@ from enum import Enum from typing import Any, Dict -from config.config import Config +from instantiation.config.config import Config from ufo.module.basic import BaseSession @@ -108,10 +108,10 @@ def instantiate_single_file(self, task_object: TaskObject) -> None: Execute the process for one task. :param task_object: The TaskObject containing task details. """ - from controller.env.env_manager import WindowsAppEnv - from controller.workflow.choose_template_flow import ChooseTemplateFlow - from controller.workflow.filter_flow import FilterFlow - from controller.workflow.prefill_flow import PrefillFlow + from instantiation.controller.env.env_manager import WindowsAppEnv + from instantiation.controller.workflow.choose_template_flow import ChooseTemplateFlow + from instantiation.controller.workflow.filter_flow import FilterFlow + from instantiation.controller.workflow.prefill_flow import PrefillFlow # Initialize the app environment and the task file name. app_object = task_object.app_object diff --git a/instantiation/controller/workflow/choose_template_flow.py b/instantiation/controller/workflow/choose_template_flow.py index f7ed8d04..1fac860b 100644 --- a/instantiation/controller/workflow/choose_template_flow.py +++ b/instantiation/controller/workflow/choose_template_flow.py @@ -7,7 +7,7 @@ from pathlib import Path from typing import Dict -from config.config import Config +from instantiation.config.config import Config from langchain.embeddings import CacheBackedEmbeddings from langchain.storage import LocalFileStore from langchain_community.embeddings import HuggingFaceEmbeddings diff --git a/instantiation/controller/workflow/filter_flow.py b/instantiation/controller/workflow/filter_flow.py index a2016cd3..16a447bc 100644 --- a/instantiation/controller/workflow/filter_flow.py +++ b/instantiation/controller/workflow/filter_flow.py @@ -4,8 +4,8 @@ import time from typing import Dict, Tuple -from config.config import Config -from controller.agent.agent import FilterAgent +from instantiation.config.config import Config +from instantiation.controller.agent.agent import FilterAgent from ufo.module.basic import BaseSession diff --git a/instantiation/controller/workflow/prefill_flow.py b/instantiation/controller/workflow/prefill_flow.py index 443499cc..2fcbd536 100644 --- a/instantiation/controller/workflow/prefill_flow.py +++ b/instantiation/controller/workflow/prefill_flow.py @@ -4,9 +4,9 @@ import time from typing import Any, Dict, List, Tuple -from config.config import Config -from controller.agent.agent import PrefillAgent -from controller.env.env_manager import WindowsAppEnv +from instantiation.config.config import Config +from instantiation.controller.agent.agent import PrefillAgent +from instantiation.controller.env.env_manager import WindowsAppEnv from ufo.agents.processors.app_agent_processor import AppAgentProcessor from ufo.automator.ui_control.inspector import ControlInspectorFacade diff --git a/instantiation/instantiation.py b/instantiation/instantiation.py index b692f7ab..2524092a 100644 --- a/instantiation/instantiation.py +++ b/instantiation/instantiation.py @@ -31,7 +31,7 @@ def main() -> None: task_dir_name = parse_arguments().task.lower() - from controller.instantiation_process import InstantiationProcess + from instantiation.controller.instantiation_process import InstantiationProcess InstantiationProcess().instantiate_files(task_dir_name) diff --git a/ufo/agents/agent/basic.py b/ufo/agents/agent/basic.py index 123ed06e..1f54e08e 100644 --- a/ufo/agents/agent/basic.py +++ b/ufo/agents/agent/basic.py @@ -237,7 +237,7 @@ def process_resume(self) -> None: self.processor.resume() - def process_asker(self, configs = configs) -> None: + def process_asker(self, ask_user: bool = True) -> None: """ Ask for the process. :param ask_user: Whether to ask the user for the questions.