From 241838e7f544a7ec45f658d2382cdd25cbd37230 Mon Sep 17 00:00:00 2001 From: Kye Date: Tue, 28 Nov 2023 12:38:43 -0800 Subject: [PATCH] [CODE QUALITY] --- swarms/prompts/agent_system_prompts.py | 4 +--- swarms/prompts/tools.py | 1 - swarms/structs/agent.py | 3 ++- tests/structs/test_task.py | 1 + 4 files changed, 4 insertions(+), 5 deletions(-) diff --git a/swarms/prompts/agent_system_prompts.py b/swarms/prompts/agent_system_prompts.py index 01b94d9d..c6aaa4f2 100644 --- a/swarms/prompts/agent_system_prompts.py +++ b/swarms/prompts/agent_system_prompts.py @@ -1,5 +1,3 @@ - - # System prompt FLOW_SYSTEM_PROMPT = """ You are an autonomous agent granted autonomy in a autonomous loop structure. @@ -10,4 +8,4 @@ and accomplish tasks bestowed by the user. You can have internal dialogues with yourself or can interact with the user to aid in these complex tasks. Your responses should be coherent, contextually relevant, and tailored to the task at hand. -""" \ No newline at end of file +""" diff --git a/swarms/prompts/tools.py b/swarms/prompts/tools.py index 17886126..db6c5807 100644 --- a/swarms/prompts/tools.py +++ b/swarms/prompts/tools.py @@ -1,4 +1,3 @@ - # Prompts DYNAMIC_STOP_PROMPT = """ diff --git a/swarms/structs/agent.py b/swarms/structs/agent.py index 610bef73..29bba651 100644 --- a/swarms/structs/agent.py +++ b/swarms/structs/agent.py @@ -20,10 +20,11 @@ from swarms.utils.pdf_to_text import pdf_to_text from swarms.prompts.tools import ( DYNAMIC_STOP_PROMPT, DYNAMICAL_TOOL_USAGE, - SCENARIOS + SCENARIOS, ) from swarms.prompts.agent_system_prompts import FLOW_SYSTEM_PROMPT + def autonomous_agent_prompt( tools_prompt: str = DYNAMICAL_TOOL_USAGE, dynamic_stop_prompt: str = DYNAMIC_STOP_PROMPT, diff --git a/tests/structs/test_task.py b/tests/structs/test_task.py index af6ac42e..cc6be26f 100644 --- a/tests/structs/test_task.py +++ b/tests/structs/test_task.py @@ -43,6 +43,7 @@ def test_agent_run_task(llm): assert "dashboard_data" in result # Add more assertions as needed + @pytest.fixture def task(): agents = [Agent(llm=llm, id=f"Agent_{i}") for i in range(5)]