- Added support for Select AI Summarize and feedback

- Unified set_attributes and set_attribute implementation across all AI
objects
- Added AI agent, tool, task and team methods

Author: aosingh

.github/workflows/test.yaml

@@ -27,7 +27,7 @@ jobs:
 
       - name: Install select_ai with core dependencies
         run: |
-          python -m pip install --upgrade pip
+          python -m pip install --upgrade pip setuptools
           pip install pytest anyio
           pip install -e .
 

samples/agent/agent_create.py

@@ -1,3 +1,16 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+# -----------------------------------------------------------------------------
+# agent_create.py
+#
+# Create an agent to answer any movie related questions
+# -----------------------------------------------------------------------------
+
 import os
 
 import select_ai
@@ -9,10 +22,7 @@
 user = os.getenv("SELECT_AI_USER")
 password = os.getenv("SELECT_AI_PASSWORD")
 dsn = os.getenv("SELECT_AI_DB_CONNECT_STRING")
-
 select_ai.connect(user=user, password=password, dsn=dsn)
-
-# Agent
 agent_attributes = AgentAttributes(
     profile_name="LLAMA_4_MAVERICK",
     role="You are an AI Movie Analyst. "

samples/agent/movie_analyst.py

@@ -1,3 +1,16 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+# -----------------------------------------------------------------------------
+# movie_analyst.py
+#
+# Demonstrates movie analyst AI agent
+# -----------------------------------------------------------------------------
+
 import os
 import uuid
 
@@ -18,41 +31,38 @@
 select_ai.connect(user=user, password=password, dsn=dsn)
 
 # Agent
-agent_attributes = AgentAttributes(
-    profile_name="oci_ai_profile",
-    role="You are an AI Movie Analyst. "
-    "Your can help answer a variety of questions related to movies. ",
-    enable_human_tool=False,
-)
 agent = Agent(
     agent_name="MOVIE_ANALYST",
-    attributes=agent_attributes,
+    attributes=AgentAttributes(
+        profile_name="oci_ai_profile",
+        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)
 print("Create Agent", agent)
 
 # Task
-task_attributes = TaskAttributes(
-    instruction="Help the user with their request about movies. "
-    "User question: {query}",
-    enable_human_tool=False,
-)
 task = Task(
     task_name="ANALYZE_MOVIE_TASK",
     description="Movie task involving a human",
-    attributes=task_attributes,
+    attributes=TaskAttributes(
+        instruction="Help the user with their request about movies. "
+        "User question: {query}",
+        enable_human_tool=False,
+    ),
 )
 task.create(replace=True)
 print("Created Task", task)
 
 # Team
-team_attributes = TeamAttributes(
-    agents=[{"name": "MOVIE_ANALYST", "task": "ANALYZE_MOVIE_TASK"}],
-    process="sequential",
-)
 team = Team(
     team_name="MOVIE_AGENT_TEAM",
-    attributes=team_attributes,
+    attributes=TeamAttributes(
+        agents=[{"name": "MOVIE_ANALYST", "task": "ANALYZE_MOVIE_TASK"}],
+        process="sequential",
+    ),
 )
 team.create(enabled=True, replace=True)
 print(

samples/agent/task_create.py

@@ -1,3 +1,16 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+# -----------------------------------------------------------------------------
+# task_create.py
+#
+# Create an AI agent task which uses SQL tool to perform NL2SQL
+# -----------------------------------------------------------------------------
+
 import os
 from pprint import pformat
 

samples/agent/tasks_list.py

@@ -1,3 +1,16 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+# -----------------------------------------------------------------------------
+# tasks_list.py
+#
+# List all tasks saved in the database
+# -----------------------------------------------------------------------------
+
 import os
 
 import select_ai

samples/agent/team_create.py

@@ -1,3 +1,16 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+# -----------------------------------------------------------------------------
+# team_create.py
+#
+# Create a team for movie analyst and run it
+# -----------------------------------------------------------------------------
+
 import os
 import uuid
 

samples/agent/tool_create.py

@@ -8,7 +8,7 @@
 # -----------------------------------------------------------------------------
 # tool_create.py
 #
-# Create a tool for
+# Create an in-built SQL tool
 # -----------------------------------------------------------------------------
 
 import os

samples/agent/tool_create_human.py

@@ -1,3 +1,16 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+# -----------------------------------------------------------------------------
+# movie_analyst.py
+#
+# Demonstrates web search AI agent
+# -----------------------------------------------------------------------------
+
 import os
 
 import select_ai
@@ -19,7 +32,8 @@
 SELECT_AI_TEAM_NAME = "WEB_SEARCH_TEAM"
 
 USER_QUERIES = {
-    "d917b055-e8a1-463a-a489-d4328a7b2210": "What are the key features for the product highlighted at this URL https://www.oracle.com/artificial-intelligence/database-machine-learning",
+    "d917b055-e8a1-463a-a489-d4328a7b2210": "What are the key features for the product highlighted at "
+    "this URL https://www.oracle.com/artificial-intelligence/database-machine-learning",
     "c2e3ff20-f56d-40e7-987c-cc72740c75a5": "What is the main topic at this URL https://www.oracle.com/artificial-intelligence/database-machine-learning",
     "25e23a25-07b9-4ed7-be11-f7e5e445d286": "What is the main topic at this URL https://openai.com",
 }
@@ -68,7 +82,8 @@
     attributes=AgentAttributes(
         profile_name=OPEN_AI_PROFILE_NAME,
         enable_human_tool=False,
-        role="You are a specialized web search agent that can access web page contents and respond to questions based on its content.",
+        role="You are a specialized web search agent that can access web page "
+        "contents and respond to questions based on its content.",
     ),
 )
 agent.create(replace=True)

samples/agent/websearch_agent.py

@@ -19,3 +19,5 @@ class Action(StrEnum):
     NARRATE = "narrate"
     CHAT = "chat"
     SHOWPROMPT = "showprompt"
+    FEEDBACK = "feedback"
+    SUMMARIZE = "summarize"