Added test cases for Select AI agents

Author: aosingh

samples/agent/task_create.py

@@ -25,7 +25,7 @@
 
 task = Task(
     task_name="ANALYZE_MOVIE_TASK",
-    description="Movie task involving a human",
+    description="Search for movies in the database",
     attributes=TaskAttributes(
         instruction="Help the user with their request about movies. "
         "User question: {query}. "

src/select_ai/_abc.py

@@ -27,6 +27,17 @@ def _bool(value: Any) -> bool:
         raise ValueError(f"Invalid boolean value: {value}")
 
 
+def _is_json(field, value) -> bool:
+    if field.type in (
+        typing.List[Mapping],
+        typing.Optional[Mapping],
+        typing.Optional[List[str]],
+        typing.Optional[List[typing.Mapping]],
+    ) and isinstance(value, (str, bytes, bytearray)):
+        return True
+    return False
+
+
 @dataclass
 class SelectAIDataClass(ABC):
     """SelectAIDataClass is an abstract container for all data
@@ -65,13 +76,7 @@ def __post_init__(self):
                     setattr(self, field.name, _bool(value))
                 elif field.type is typing.Optional[float]:
                     setattr(self, field.name, float(value))
-                elif field.type is typing.Optional[Mapping] and isinstance(
-                    value, (str, bytes, bytearray)
-                ):
-                    setattr(self, field.name, json.loads(value))
-                elif field.type is typing.Optional[
-                    List[typing.Mapping]
-                ] and isinstance(value, (str, bytes, bytearray)):
+                elif _is_json(field, value):
                     setattr(self, field.name, json.loads(value))
                 else:
                     setattr(self, field.name, value)

src/select_ai/agent/sql.py

@@ -62,19 +62,21 @@
 
 
 GET_USER_AI_AGENT_TEAM = """
-SELECT t.tool_name, t.description
+SELECT t.agent_team_name as team_name, t.description
 FROM USER_AI_AGENT_TEAMS t
-WHERE t.team_name = :team_name
+WHERE t.agent_team_name = :team_name
 """
 
+
 GET_USER_AI_AGENT_TEAM_ATTRIBUTES = """
 SELECT attribute_name, attribute_value
 FROM USER_AI_AGENT_TEAM_ATTRIBUTES
-WHERE team_name = :team_name
+WHERE agent_team_name = :team_name
 """
 
+
 LIST_USER_AI_AGENT_TEAMS = """
-SELECT t.tool_name, t.description
+SELECT t.AGENT_TEAM_NAME as team_name, description
 FROM USER_AI_AGENT_TEAMS t
-WHERE REGEXP_LIKE(t.team_name, :team_name_pattern, 'i')
+WHERE REGEXP_LIKE(t.AGENT_TEAM_NAME, :team_name_pattern, 'i')
 """

src/select_ai/agent/team.py

@@ -110,7 +110,7 @@ def _get_attributes(team_name: str) -> TeamAttributes:
     @staticmethod
     def _get_description(team_name: str) -> Union[str, None]:
         with cursor() as cr:
-            cr.execute(GET_USER_AI_AGENT_TEAM, task_name=team_name.upper())
+            cr.execute(GET_USER_AI_AGENT_TEAM, team_name=team_name.upper())
             team = cr.fetchone()
             if team:
                 if team[1] is not None:

tests/agents/test_3000_tools.py

@@ -6,7 +6,7 @@
 # -----------------------------------------------------------------------------
 
 """
-3000 - Module for testing select_ai agents
+3000 - Module for testing select_ai agent tools
 """
 
 import uuid

tests/agents/test_3100_tasks.py

@@ -0,0 +1,70 @@
+# -----------------------------------------------------------------------------
+# Copyright (c) 2025, Oracle and/or its affiliates.
+#
+# Licensed under the Universal Permissive License v 1.0 as shown at
+# http://oss.oracle.com/licenses/upl.
+# -----------------------------------------------------------------------------
+
+"""
+3100 - Module for testing select_ai agent tasks
+"""
+
+import uuid
+
+import pytest
+import select_ai
+from select_ai.agent import Task, TaskAttributes
+
+PYSAI_3100_TASK_NAME = f"PYSAI_3100_{uuid.uuid4().hex.upper()}"
+PYSAI_3100_SQL_TASK_DESCRIPTION = "PYSAI_3100_SQL_TASK_DESCRIPTION"
+
+
+@pytest.fixture(scope="module")
+def task_attributes():
+    return TaskAttributes(
+        instruction="Help the user with their request about movies. "
+        "User question: {query}. "
+        "You can use SQL tool to search the data from database",
+        tools=["MOVIE_SQL_TOOL"],
+        enable_human_tool=False,
+    )
+
+
+@pytest.fixture(scope="module")
+def task(task_attributes):
+    task = Task(
+        task_name=PYSAI_3100_TASK_NAME,
+        description=PYSAI_3100_SQL_TASK_DESCRIPTION,
+        attributes=task_attributes,
+    )
+    task.create()
+    yield task
+    task.delete(force=True)
+
+
+def test_3100(task, task_attributes):
+    """simple task creation"""
+    assert task.task_name == PYSAI_3100_TASK_NAME
+    assert task.attributes == task_attributes
+    assert task.description == PYSAI_3100_SQL_TASK_DESCRIPTION
+
+
+@pytest.mark.parametrize("task_name_pattern", [None, "^PYSAI_3100_"])
+def test_3101(task_name_pattern):
+    """task list"""
+    if task_name_pattern:
+        tasks = list(select_ai.agent.Task.list(task_name_pattern))
+    else:
+        tasks = list(select_ai.agent.Task.list())
+    task_names = set(task.task_name for task in tasks)
+    task_descriptions = set(task.description for task in tasks)
+    assert PYSAI_3100_TASK_NAME in task_names
+    assert PYSAI_3100_SQL_TASK_DESCRIPTION in task_descriptions
+
+
+def test_3102(task_attributes):
+    """task fetch"""
+    task = select_ai.agent.Task.fetch(PYSAI_3100_TASK_NAME)
+    assert task.task_name == PYSAI_3100_TASK_NAME
+    assert task.attributes == task_attributes
+    assert task.description == PYSAI_3100_SQL_TASK_DESCRIPTION

tests/agents/test_3200_agents.py

@@ -0,0 +1,71 @@
+# -----------------------------------------------------------------------------
+# Copyright (c) 2025, Oracle and/or its affiliates.
+#
+# Licensed under the Universal Permissive License v 1.0 as shown at
+# http://oss.oracle.com/licenses/upl.
+# -----------------------------------------------------------------------------
+
+"""
+3200 - Module for testing select_ai agents
+"""
+import uuid
+
+import pytest
+import select_ai
+from select_ai.agent import Agent, AgentAttributes
+
+PYSAI_3200_AGENT_NAME = f"PYSAI_3200_AGENT_{uuid.uuid4().hex.upper()}"
+PYSAI_3200_AGENT_DESCRIPTION = "PYSAI_3200_AGENT_DESCRIPTION"
+PYSAI_3200_PROFILE_NAME = f"PYSAI_3200_PROFILE_{uuid.uuid4().hex.upper()}"
+
+
+@pytest.fixture(scope="module")
+def python_gen_ai_profile(profile_attributes):
+    profile = select_ai.Profile(
+        profile_name=PYSAI_3200_PROFILE_NAME,
+        description="OCI GENAI Profile",
+        attributes=profile_attributes,
+    )
+    yield profile
+    profile.delete(force=True)
+
+
+@pytest.fixture(scope="module")
+def agent_attributes():
+    agent_attributes = AgentAttributes(
+        profile_name=PYSAI_3200_PROFILE_NAME,
+        role="You are an AI Movie Analyst. "
+        "Your can help answer a variety of questions related to movies. ",
+        enable_human_tool=False,
+    )
+    return agent_attributes
+
+
+@pytest.fixture(scope="module")
+def agent(python_gen_ai_profile, agent_attributes):
+    agent = Agent(
+        agent_name=PYSAI_3200_AGENT_NAME,
+        attributes=agent_attributes,
+        description=PYSAI_3200_AGENT_DESCRIPTION,
+    )
+    agent.create(enabled=True, replace=True)
+    yield agent
+    agent.delete(force=True)
+
+
+def test_3200(agent, agent_attributes):
+    assert agent.agent_name == PYSAI_3200_AGENT_NAME
+    assert agent.attributes == agent_attributes
+    assert agent.description == PYSAI_3200_AGENT_DESCRIPTION
+
+
+@pytest.mark.parametrize("agent_name_pattern", [None, "^PYSAI_3200_AGENT_"])
+def test_3201(agent_name_pattern):
+    if agent_name_pattern:
+        agents = list(select_ai.agent.Agent.list(agent_name_pattern))
+    else:
+        agents = list(select_ai.agent.Agent.list())
+    agent_names = set(agent.agent_name for agent in agents)
+    agent_descriptions = set(agent.description for agent in agents)
+    assert PYSAI_3200_AGENT_NAME in agent_names
+    assert PYSAI_3200_AGENT_DESCRIPTION in agent_descriptions

tests/agents/test_3300_teams.py

@@ -0,0 +1,125 @@
+# -----------------------------------------------------------------------------
+# Copyright (c) 2025, Oracle and/or its affiliates.
+#
+# Licensed under the Universal Permissive License v 1.0 as shown at
+# http://oss.oracle.com/licenses/upl.
+# -----------------------------------------------------------------------------
+
+"""
+3300 - Module for testing select_ai agent teams
+"""
+
+import uuid
+
+import pytest
+import select_ai
+from select_ai.agent import (
+    Agent,
+    AgentAttributes,
+    Task,
+    TaskAttributes,
+    Team,
+    TeamAttributes,
+)
+
+PYSAI_3300_AGENT_NAME = f"PYSAI_3300_AGENT_{uuid.uuid4().hex.upper()}"
+PYSAI_3300_AGENT_DESCRIPTION = "PYSAI_3300_AGENT_DESCRIPTION"
+PYSAI_3300_PROFILE_NAME = f"PYSAI_3300_PROFILE_{uuid.uuid4().hex.upper()}"
+PYSAI_3300_TASK_NAME = f"PYSAI_3300_{uuid.uuid4().hex.upper()}"
+PYSAI_3300_TASK_DESCRIPTION = "PYSAI_3100_SQL_TASK_DESCRIPTION"
+PYSAI_3300_TEAM_NAME = f"PYSAI_3300_TEAM_{uuid.uuid4().hex.upper()}"
+PYSAI_3300_TEAM_DESCRIPTION = "PYSAI_3300_TEAM_DESCRIPTION"
+
+
+@pytest.fixture(scope="module")
+def python_gen_ai_profile(profile_attributes):
+    profile = select_ai.Profile(
+        profile_name=PYSAI_3300_PROFILE_NAME,
+        description="OCI GENAI Profile",
+        attributes=profile_attributes,
+    )
+    yield profile
+    profile.delete(force=True)
+
+
+@pytest.fixture(scope="module")
+def task_attributes():
+    return TaskAttributes(
+        instruction="Help the user with their request about movies. "
+        "User question: {query}. ",
+        enable_human_tool=False,
+    )
+
+
+@pytest.fixture(scope="module")
+def task(task_attributes):
+    task = Task(
+        task_name=PYSAI_3300_TASK_NAME,
+        description=PYSAI_3300_TASK_DESCRIPTION,
+        attributes=task_attributes,
+    )
+    task.create()
+    yield task
+    task.delete(force=True)
+
+
+@pytest.fixture(scope="module")
+def agent(python_gen_ai_profile):
+    agent = Agent(
+        agent_name=PYSAI_3300_AGENT_NAME,
+        description=PYSAI_3300_AGENT_DESCRIPTION,
+        attributes=AgentAttributes(
+            profile_name=PYSAI_3300_PROFILE_NAME,
+            role="You are an AI Movie Analyst. "
+            "Your can help answer a variety of questions related to movies. ",
+            enable_human_tool=False,
+        ),
+    )
+    agent.create(enabled=True, replace=True)
+    yield agent
+    agent.delete(force=True)
+
+
+@pytest.fixture(scope="module")
+def team_attributes(agent, task):
+    return TeamAttributes(
+        agents=[{"name": agent.agent_name, "task": task.task_name}],
+        process="sequential",
+    )
+
+
+@pytest.fixture(scope="module")
+def team(team_attributes):
+    team = Team(
+        team_name=PYSAI_3300_TEAM_NAME,
+        description=PYSAI_3300_TEAM_DESCRIPTION,
+        attributes=team_attributes,
+    )
+    team.create()
+    yield team
+    team.delete(force=True)
+
+
+def test_3300(team, team_attributes):
+    assert team.team_name == PYSAI_3300_TEAM_NAME
+    assert team.description == PYSAI_3300_TEAM_DESCRIPTION
+    assert team.attributes == team_attributes
+
+
+@pytest.mark.parametrize("team_name_pattern", [None, "^PYSAI_3300_TEAM_"])
+def test_3301(team_name_pattern):
+    if team_name_pattern:
+        teams = list(select_ai.agent.Team.list(team_name_pattern))
+    else:
+        teams = list(select_ai.agent.Team.list())
+    team_names = set(team.team_name for team in teams)
+    team_descriptions = set(team.description for team in teams)
+    assert PYSAI_3300_TEAM_NAME in team_names
+    assert PYSAI_3300_TEAM_DESCRIPTION in team_descriptions
+
+
+def test_3302(team_attributes):
+    team = select_ai.agent.Team.fetch(team_name=PYSAI_3300_TEAM_NAME)
+    assert team.team_name == PYSAI_3300_TEAM_NAME
+    assert team.description == PYSAI_3300_TEAM_DESCRIPTION
+    assert team.attributes == team_attributes