From 5e3d012eab102addcef0033dddc5a4f6db239e25 Mon Sep 17 00:00:00 2001 From: Hugo Saporetti Junior Date: Thu, 17 Oct 2024 20:54:32 -0300 Subject: [PATCH] fix tests due to direct new param in actionplan --- src/main/requirements.txt | 47 ++++++++++++++++++++++++++ src/test/fixtures/action_plan_stubs.py | 7 ++-- 2 files changed, 52 insertions(+), 2 deletions(-) create mode 100644 src/main/requirements.txt diff --git a/src/main/requirements.txt b/src/main/requirements.txt new file mode 100644 index 00000000..d6966e93 --- /dev/null +++ b/src/main/requirements.txt @@ -0,0 +1,47 @@ +###### AUTO-GENERATED Requirements file for: AskAI ###### + +hspylib>=1.12.50 +hspylib-clitt>=0.9.138 +hspylib-setman>=0.10.41 +retry2>=0.9.5 +pause>=0.3 +tqdm>=4.66.5 +pyperclip>=1.9.0 +python-magic>=0.4.27 +pytz>=2024.1 +langchain>=0.3.0 +langchain-openai>=0.2.0 +langchain-community>=0.3.1 +langchain-google-community>=2.0.0 +openai-whisper>=20231117 +openai>=1.48.0 +google-api-python-client>=2.147.0 +fake_useragent>=1.5.1 +requests>=2.32.3 +urllib3>=1.26.20 +protobuf>=4.25.4 +aiohttp>=3.10.5 +html2text>=2024.2.26 +rich>=13.8.1 +textual>=0.80.1 +soundfile>=0.12.1 +PyAudio>=0.2.14 +SpeechRecognition>=3.10.4 +opencv-python>=4.10.0.84 +pyautogui>=0.9.54 +torch>=2.2.0 +torchvision>=0.17.2 +open-clip-torch +opentelemetry-api>=1.27.0 +opentelemetry-sdk>=1.27.0 +opentelemetry-proto>=1.27.0 +transformers>=4.44.2 +unstructured>=0.15.8 +unstructured[md]>=0.15.8 +tiktoken>=0.7.0 +stanza==1.1.1 +nltk>=3.9.1 +faiss-cpu~=1.8.0 +chromadb>=0.5.5 +deepl>=1.18.0 +argostranslate==1.9.1 diff --git a/src/test/fixtures/action_plan_stubs.py b/src/test/fixtures/action_plan_stubs.py index 1d4818a0..3b26fcfc 100644 --- a/src/test/fixtures/action_plan_stubs.py +++ b/src/test/fixtures/action_plan_stubs.py @@ -12,6 +12,7 @@ def stub_response(index: int) -> Optional[ActionPlan]: "list my downloads", "I will help you list the contents of your 'Downloads' folder.", "List the contents of the 'Downloads' folder", + False, [], [ SimpleNamespace( @@ -26,16 +27,18 @@ def stub_response(index: int) -> Optional[ActionPlan]: ), ActionPlan( "hello", - "I will greet the user and initiate the conversation.", + "Hello! How can I assist you today?", "Respond to the user's greeting.", + True, + [], [], - [SimpleNamespace(**{"id": "1", "task": "Direct: 'Hello! How can I assist you today?'"})], ModelResult.default(), ), ActionPlan( "List my downloads and let me know if there is any image.", "I will list the contents of your downloads folder and check for any image files present.", "List the contents of the downloads folder and identify any image files", + False, [ SimpleNamespace(**{"id": "1", "sub_goal": "List the contents of the downloads folder"}), SimpleNamespace(**{"id": "2", "sub_goal": "Identify image files in the downloads folder"}),