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

{select_ai-1.1.0rc1/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.0rc1
+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.0rc1 → select_ai-1.2.0rc1}/samples/async/vector_index_get_attributes.py

@@ -32,4 +32,5 @@ async def main():
     async_profile = await async_vector_index.get_profile()
     print(async_profile)
 
+
 asyncio.run(main())

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

@@ -6,9 +6,9 @@
 # -----------------------------------------------------------------------------
 
 # -----------------------------------------------------------------------------
-# async/vector_index_get_attributes.py
+# async/vector_index_update_attributes.py
 #
-# Get vector index attributes
+# Update Vector index attributes
 # -----------------------------------------------------------------------------
 
 import asyncio
@@ -34,7 +34,9 @@ async def main():
     await async_vector_index.set_attributes(attributes=updated_attributes)
 
     # Use vector_index.set_attribute to update a single attribute
-    await async_vector_index.set_attribute(attribute_name="similarity_threshold", attribute_value=0.5)
+    await async_vector_index.set_attribute(
+        attribute_name="similarity_threshold", attribute_value=0.5
+    )
     attributes = await async_vector_index.get_attributes()
     print(attributes)
 

{select_ai-1.1.0rc1 → 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.1.0rc1 → select_ai-1.2.0rc1}/samples/vector_index_update_attributes.py

@@ -6,9 +6,9 @@
 # -----------------------------------------------------------------------------
 
 # -----------------------------------------------------------------------------
-# vector_index_get_attributes.py
+# vector_index_update_attributes.py
 #
-# Gets attributes for a vector index
+# Update attributes for a vector index
 # -----------------------------------------------------------------------------
 
 import os
@@ -30,5 +30,7 @@ updated_attributes = select_ai.OracleVectorIndexAttributes(
 vector_index.set_attributes(attributes=updated_attributes)
 
 # Use vector_index.set_attribute to update a single attribute
-vector_index.set_attribute(attribute_name="similarity_threshold", attribute_value=0.5)
+vector_index.set_attribute(
+    attribute_name="similarity_threshold", attribute_value=0.5
+)
 print(vector_index.get_attributes())

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,
+)