From 8987c6d96942318ddfb8f7b88055338b5e67b7aa Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Sat, 16 Sep 2023 00:05:03 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(test=5Fagents=5Ftemplate.py)?=
 =?UTF-8?q?:=20fix=20incorrect=20variable=20name=20in=20test=5Fzero=5Fshot?=
 =?UTF-8?q?=5Fagent=20function=20=F0=9F=90=9B=20fix(test=5Fendpoints.py):?=
 =?UTF-8?q?=20add=20missing=20pytest.mark.async=5Ftest=20decorator=20to=20?=
 =?UTF-8?q?test=5Fasync=5Ftask=5Fprocessing=20function?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_agents_template.py | 6 +++---
 tests/test_endpoints.py       | 1 +
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/tests/test_agents_template.py b/tests/test_agents_template.py
index 85f689eda..b12ad7dee 100644
--- a/tests/test_agents_template.py
+++ b/tests/test_agents_template.py
@@ -31,15 +31,15 @@ def test_zero_shot_agent(client: TestClient, logged_in_headers):
     }
 
     # Additional assertions for other template variables
-    assert template["callback_service"] == {
+    assert template["callback_manager"] == {
         "required": False,
         "dynamic": False,
         "placeholder": "",
         "show": False,
         "multiline": False,
         "password": False,
-        "name": "callback_service",
-        "type": "BaseCallbackService",
+        "name": "callback_manager",
+        "type": "BaseCallbackManager",
         "list": False,
         "advanced": False,
         "info": "",
diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py
index da67cbb1a..cb3755db0 100644
--- a/tests/test_endpoints.py
+++ b/tests/test_endpoints.py
@@ -452,6 +452,7 @@ def test_basic_chat_with_two_session_ids_and_names(client, added_flow, created_a
 
 
 # Test function without loop
+@pytest.mark.async_test
 def test_async_task_processing(client, added_flow, created_api_key):
     headers = {"x-api-key": created_api_key.api_key}
     post_data = {"inputs": {"text": "Hi, My name is Gabriel"}}