select-ai 1.1.0__tar.gz → 1.2.0rc1__tar.gz

{select_ai-1.1.0/src/select_ai.egg-info → select_ai-1.2.0rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: select_ai
-Version: 1.1.0
+Version: 1.2.0rc1
 Summary: Select AI for Python
 Author-email: Abhishek Singh <abhishek.o.singh@oracle.com>
 Maintainer-email: Abhishek Singh <abhishek.o.singh@oracle.com>

select_ai-1.2.0rc1/samples/agent/agent_create.py

@@ -0,0 +1,27 @@
+import os
+
+import select_ai
+from select_ai.agent import (
+    Agent,
+    AgentAttributes,
+)
+
+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. "
+    "Your can help answer a variety of questions related to movies. ",
+    enable_human_tool=False,
+)
+agent = Agent(
+    agent_name="MOVIE_ANALYST",
+    attributes=agent_attributes,
+)
+agent.create(enabled=True, replace=True)
+print("Created Agent:", agent)

select_ai-1.2.0rc1/samples/agent/movie_analyst.py

@@ -0,0 +1,63 @@
+import os
+import uuid
+
+import select_ai
+from select_ai.agent import (
+    Agent,
+    AgentAttributes,
+    Task,
+    TaskAttributes,
+    Team,
+    TeamAttributes,
+)
+
+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="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,
+)
+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,
+)
+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,
+)
+team.create(enabled=True, replace=True)
+print(
+    team.run(
+        prompt="In the movie Titanic, was there enough space for Jack ? ",
+        params={"conversation_id": str(uuid.uuid4())},
+    )
+)

select_ai-1.2.0rc1/samples/agent/task_create.py

@@ -0,0 +1,31 @@
+import os
+from pprint import pformat
+
+import select_ai
+import select_ai.agent
+from select_ai.agent import Task, TaskAttributes
+
+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)
+
+attributes = 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,
+)
+
+task = Task(
+    task_name="ANALYZE_MOVIE_TASK",
+    description="Movie task involving a human",
+    attributes=attributes,
+)
+
+task.create(replace=True)
+
+print(task.task_name)
+print(pformat(task.attributes))

select_ai-1.2.0rc1/samples/agent/tasks_list.py

@@ -0,0 +1,13 @@
+import os
+
+import select_ai
+import select_ai.agent
+
+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)
+
+for task in select_ai.agent.Task.list():
+    print(task)

select_ai-1.2.0rc1/samples/agent/team_create.py

@@ -0,0 +1,33 @@
+import os
+import uuid
+
+import select_ai
+from select_ai.agent import (
+    Team,
+    TeamAttributes,
+)
+
+conversation_id = str(uuid.uuid4())
+
+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)
+
+# Team
+team = Team(
+    team_name="MOVIE_AGENT_TEAM",
+    attributes=TeamAttributes(
+        agents=[{"name": "MOVIE_ANALYST", "task": "ANALYZE_MOVIE_TASK"}],
+        process="sequential",
+    ),
+)
+team.create(enabled=True, replace=True)
+
+print(
+    team.run(
+        prompt="Could you list the movies in the database?",
+        params={"conversation_id": conversation_id},
+    )
+)

select_ai-1.2.0rc1/samples/agent/tool_create.py

@@ -0,0 +1,55 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+# -----------------------------------------------------------------------------
+# tool_create.py
+#
+# Create a tool for
+# -----------------------------------------------------------------------------
+
+import os
+from pprint import pformat
+
+import select_ai
+import select_ai.agent
+
+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)
+
+profile_attributes = select_ai.ProfileAttributes(
+    credential_name="my_oci_ai_profile_key",
+    object_list=[
+        {"owner": user, "name": "MOVIE"},
+        {"owner": user, "name": "ACTOR"},
+        {"owner": user, "name": "DIRECTOR"},
+    ],
+    provider=select_ai.OCIGenAIProvider(
+        region="us-chicago-1",
+        oci_apiformat="GENERIC",
+        model="meta.llama-4-maverick-17b-128e-instruct-fp8",
+    ),
+)
+profile = select_ai.Profile(
+    profile_name="LLAMA_4_MAVERICK",
+    attributes=profile_attributes,
+    description="MY OCI AI Profile",
+    replace=True,
+)
+
+# Use the OCI AI Profile to perform natural
+# language SQL translation
+sql_tool = select_ai.agent.Tool.create_sql_tool(
+    tool_name="MOVIE_SQL_TOOL",
+    description="My Select AI MOVIE SQL agent tool",
+    profile_name="LLAMA_4_MAVERICK",
+    replace=True,
+)
+print(sql_tool.tool_name)
+print(pformat(sql_tool.attributes))

select_ai-1.2.0rc1/samples/agent/tool_create_human.py

@@ -0,0 +1,35 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+# -----------------------------------------------------------------------------
+# tool_create_human.py
+#
+# Create a vector index for Retrieval Augmented Generation (RAG)
+# -----------------------------------------------------------------------------
+
+import os
+
+import select_ai
+import select_ai.agent
+
+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)
+
+attributes = select_ai.agent.ToolAttributes(
+    tool_params=select_ai.agent.HumanToolParams()
+)
+
+human_tool = select_ai.agent.Tool(
+    attributes=attributes,
+    tool_name="Human",
+    description="My Select AI agent human tool",
+)
+
+human_tool.create()

select_ai-1.2.0rc1/samples/agent/tool_delete.py

@@ -0,0 +1,25 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+# -----------------------------------------------------------------------------
+# tool_delete.py
+#
+# Delete an AI tool
+# -----------------------------------------------------------------------------
+
+import os
+
+import select_ai.agent
+
+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)
+
+tool = select_ai.agent.Tool("LLM_CHAT_TOOL")
+tool.delete(force=True)

select_ai-1.2.0rc1/samples/agent/tool_list.py

@@ -0,0 +1,26 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+# -----------------------------------------------------------------------------
+# tool_list.py
+#
+# List all the Agent Tools
+# -----------------------------------------------------------------------------
+
+import os
+
+import select_ai
+import select_ai.agent
+
+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)
+
+for tool in select_ai.agent.Tool.list():
+    print(tool)

select_ai-1.2.0rc1/samples/agent/websearch_agent.py

@@ -0,0 +1,101 @@
+import os
+
+import select_ai
+from select_ai.agent import (
+    Agent,
+    AgentAttributes,
+    Task,
+    TaskAttributes,
+    Team,
+    TeamAttributes,
+    Tool,
+)
+
+OPEN_AI_CREDENTIAL_NAME = "OPENAI_CRED"
+OPEN_AI_PROFILE_NAME = "OPENAI_PROFILE"
+SELECT_AI_AGENT_NAME = "WEB_SEARCH_AGENT"
+SELECT_AI_TASK_NAME = "WEB_SEARCH_TASK"
+SELECT_AI_TOOL_NAME = "WEB_SEARCH_TOOL"
+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",
+    "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",
+}
+
+# connect
+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)
+
+# Create Open AI credential
+select_ai.create_credential(
+    credential={
+        "credential_name": OPEN_AI_CREDENTIAL_NAME,
+        "username": "OPENAI",
+        "password": os.getenv("OPEN_AI_API_KEY"),
+    },
+    replace=True,
+)
+print("Created credential: ", OPEN_AI_CREDENTIAL_NAME)
+
+# # Create Open AI Profile
+profile = select_ai.Profile(
+    profile_name=OPEN_AI_PROFILE_NAME,
+    attributes=select_ai.ProfileAttributes(
+        credential_name=OPEN_AI_CREDENTIAL_NAME,
+        provider=select_ai.OpenAIProvider(model="gpt-4.1"),
+    ),
+    description="My Open AI Profile",
+    replace=True,
+)
+print("Created profile: ", OPEN_AI_PROFILE_NAME)
+
+# Create an AI Agent team
+team = Team(
+    team_name=SELECT_AI_TEAM_NAME,
+    attributes=TeamAttributes(
+        agents=[{"name": SELECT_AI_AGENT_NAME, "task": SELECT_AI_TASK_NAME}]
+    ),
+)
+team.create(replace=True)
+
+# Agent
+agent = Agent(
+    agent_name=SELECT_AI_AGENT_NAME,
+    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.",
+    ),
+)
+agent.create(replace=True)
+
+# Task
+task = Task(
+    task_name=SELECT_AI_TASK_NAME,
+    attributes=TaskAttributes(
+        instruction="Answer the user question about the provided URL:{query}",
+        enable_human_tool=False,
+        tools=[SELECT_AI_TOOL_NAME],
+    ),
+)
+task.create(replace=True)
+
+# Tool
+web_search_tool = Tool.create_websearch_tool(
+    tool_name=SELECT_AI_TOOL_NAME,
+    credential_name=OPEN_AI_CREDENTIAL_NAME,
+    description="Web Search Tool using OpenAI",
+    replace=True,
+)
+print("Created tool: ", SELECT_AI_TOOL_NAME)
+
+# Run the Agent Team
+for conversation_id, prompt in USER_QUERIES.items():
+    response = team.run(
+        prompt=prompt, params={"conversation_id": conversation_id}
+    )
+    print(response)

{select_ai-1.1.0 → select_ai-1.2.0rc1}/samples/enable_ai_provider.py

@@ -23,6 +23,6 @@ select_ai_user = os.getenv("SELECT_AI_USER")
 
 select_ai.connect(user=admin_user, password=password, dsn=dsn)
 select_ai.enable_provider(
-    users=select_ai_user, provider_endpoint="*.openai.azure.com"
+    users=select_ai_user, provider_endpoint="api.OPENAI.com"
 )
 print("Enabled AI provider for user: ", select_ai_user)

select_ai-1.2.0rc1/src/select_ai/agent/__init__.py

@@ -0,0 +1,24 @@
+# -----------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+
+from .core import Agent, AgentAttributes
+from .task import Task, TaskAttributes
+from .team import Team, TeamAttributes
+from .tool import (
+    EmailNotificationToolParams,
+    HTTPToolParams,
+    HumanToolParams,
+    RAGToolParams,
+    SlackNotificationToolParams,
+    SQLToolParams,
+    Tool,
+    ToolAttributes,
+    ToolParams,
+    ToolType,
+    WebSearchToolParams,
+)

select_ai-1.2.0rc1/src/select_ai/agent/core.py

@@ -0,0 +1,235 @@
+# ------------------------------------------------------------------------------
+# 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.
+# -----------------------------------------------------------------------------
+
+import json
+from abc import ABC
+from dataclasses import dataclass
+from typing import (
+    Any,
+    AsyncGenerator,
+    Iterator,
+    List,
+    Mapping,
+    Optional,
+    Union,
+)
+
+import oracledb
+
+from select_ai import BaseProfile
+from select_ai._abc import SelectAIDataClass
+from select_ai._enums import StrEnum
+from select_ai.agent.sql import (
+    GET_USER_AI_AGENT,
+    GET_USER_AI_AGENT_ATTRIBUTES,
+    LIST_USER_AI_AGENTS,
+)
+from select_ai.async_profile import AsyncProfile
+from select_ai.db import async_cursor, cursor
+from select_ai.errors import AgentNotFoundError
+from select_ai.profile import Profile
+
+
+@dataclass
+class AgentAttributes(SelectAIDataClass):
+    """AI Agent Attributes
+
+    :param str profile_name: Name of the AI Profile which agent will
+     use to send request to LLM
+    :param str role: Agent's role also sent to LLM
+    :param bool enable_human_tool: Enable human tool support. Agent
+     will ask question to the user for any clarification
+    """
+
+    profile_name: str
+    role: str
+    enable_human_tool: Optional[bool] = True
+
+
+class BaseAgent(ABC):
+
+    def __init__(
+        self,
+        agent_name: Optional[str] = None,
+        description: Optional[str] = None,
+        attributes: Optional[AgentAttributes] = None,
+    ):
+        if attributes is not None and not isinstance(
+            attributes, AgentAttributes
+        ):
+            raise TypeError(
+                "attributes must be an object of type "
+                "select_ai.agent.AgentAttributes"
+            )
+        self.agent_name = agent_name
+        self.description = description
+        self.attributes = attributes
+
+    def __repr__(self):
+        return (
+            f"{self.__class__.__name__}("
+            f"agent_name={self.agent_name}, "
+            f"attributes={self.attributes}, description={self.description})"
+        )
+
+
+class Agent(BaseAgent):
+    """
+    select_ai.agent.Agent class lets you create, delete, enable, disable
+    and list AI agents
+
+    :param str agent_name: The name of the AI Agent
+    :param str description: Optional description of the AI agent
+    :param select_ai.agent.AgentAttributes attributes: AI agent attributes
+
+    """
+
+    @staticmethod
+    def _get_attributes(agent_name: str) -> AgentAttributes:
+        with cursor() as cr:
+            cr.execute(GET_USER_AI_AGENT, agent_name=agent_name.upper())
+            attributes = cr.fetchall()
+            if attributes:
+                post_processed_attributes = {}
+                for k, v in attributes:
+                    if isinstance(v, oracledb.LOB):
+                        post_processed_attributes[k] = v.read()
+                    else:
+                        post_processed_attributes[k] = v
+                return AgentAttributes(**post_processed_attributes)
+            else:
+                raise AgentNotFoundError(agent_name=agent_name)
+
+    def create(
+        self, enabled: Optional[bool] = True, replace: Optional[bool] = False
+    ):
+        """
+        Register a new AI Agent within the Select AI framework
+
+        :param bool enabled: Whether the AI Agent should be enabled.
+         Default value is True.
+
+        :param bool replace: Whether the AI Agent should be replaced.
+         Default value is False.
+
+        """
+        if self.agent_name is None:
+            raise AttributeError("Agent must have a name")
+        if self.attributes is None:
+            raise AttributeError("Agent must have attributes")
+
+        parameters = {
+            "agent_name": self.agent_name,
+            "attributes": self.attributes.json(),
+        }
+        if self.description:
+            parameters["description"] = self.description
+
+        if not enabled:
+            parameters["status"] = "disabled"
+
+        with cursor() as cr:
+            try:
+                cr.callproc(
+                    "DBMS_CLOUD_AI_AGENT.CREATE_AGENT",
+                    keyword_parameters=parameters,
+                )
+            except oracledb.Error as err:
+                (err_obj,) = err.args
+                if err_obj.code in (20050, 20052) and replace:
+                    self.delete(force=True)
+                    cr.callproc(
+                        "DBMS_CLOUD_AI_AGENT.CREATE_AGENT",
+                        keyword_parameters=parameters,
+                    )
+                else:
+                    raise
+
+    def delete(self, force: Optional[bool] = False):
+        """
+        Delete AI Agent from the database
+
+        :param bool force: Force the deletion. Default value is False.
+
+        """
+        with cursor() as cr:
+            cr.callproc(
+                "DBMS_CLOUD_AI_AGENT.DROP_AGENT",
+                keyword_parameters={
+                    "agent_name": self.agent_name,
+                    "force": force,
+                },
+            )
+
+    def disable(self):
+        """
+        Disable AI Agent
+
+        """
+        pass
+
+    def enable(self):
+        """
+        Enable AI Agent
+        """
+        pass
+
+    @classmethod
+    def fetch(cls, agent_name: str) -> "Agent":
+        """
+        Fetch AI Agent attributes from the Database and build a proxy object in
+        the Python layer
+
+        :param str agent_name: The name of the AI Agent
+
+        :return: select_ai.agent.Agent
+
+        :raises select_ai.errors.AgentNotFoundError:
+         If the AI Agent is not found
+
+        """
+        pass
+
+    @classmethod
+    def list(
+        cls, agent_name_pattern: Optional[str] = ".*"
+    ) -> Iterator["Agent"]:
+        """
+        List AI agents matching a pattern
+
+        :param str agent_name_pattern: Regular expressions can be used
+         to specify a pattern. Function REGEXP_LIKE is used to perform the
+         match. Default value is ".*" i.e. match all agent names.
+
+        :return: Iterator[Agent]
+        """
+        with cursor() as cr:
+            cr.execute(
+                LIST_USER_AI_AGENTS,
+                agent_name_pattern=agent_name_pattern,
+            )
+            for row in cr.fetchall():
+                agent_name = row[0]
+                if row[1]:
+                    description = row[1].read()  # Oracle.LOB
+                else:
+                    description = None
+                attributes = cls._get_attributes(agent_name=agent_name)
+                yield cls(
+                    agent_name=agent_name,
+                    description=description,
+                    attributes=attributes,
+                )
+
+    def set_attributes(self, attributes: AgentAttributes) -> None:
+        """
+        Set AI Agent attributes
+        """
+        pass
+
+    def set_attribute(self, attribute_name: str, attribute_value: Any) -> None:
+        pass