-
Access to latest AI Engine features
-
Simple and intuitive API
poetry add ai-engine-sdk
# or
pip install ai-engine-sdk
Before you start to integrate the AI Engine into your app, you might want to get familiar with agent functions.
To find out how to generate an apiKey
check out the documentation regarding Agentverse API keys.
from ai_engine_sdk import AiEngine
ai_engine: AiEngine = AiEngine(api_key)
function_groups: list[FunctionGroup] = await ai_engine.get_function_groups()
public_group = next(
(g for g in function_groups if g.name == "Fetch Verified"), None
)
If you would like to use the functions in your own My Functions function group, you can use this filter instead:
my_group = next(
(g for g in function_groups if g.name == "My Functions"), None
)
Allow to other users to use functions
, under a concrete function-group
, without replicating that function
or allowing them alter those functions
or funtion-group
data.
If you wish to give access to a certain function-group
(use, not alter the data/state of it), you can use the following method in the following way:
# assuming `ai_engine` is a valid instance of AiEngine
what_function_group_identifier_i_want_share="normally-this-is-an-uuid4"
user_email_i_want_to_share_the_function_group_with = "[email protected]"
await ai_engine.share_function_group(
function_group_id=what_function_group_identifier_i_want_share,
target_user_email=user_email_i_want_to_share_the_function_group_with
)
Now, if you were requesting the available function-groups
for the user with email assigned to the target_user_email
argument, the function-group
with the id assigned to function_group_id
.
You can check that by using the AiEngine.get_function_groups
method.
session = await ai_engine.create_session(function_group=public_group.uuid)
await session.start(objective)
You might want to query new messages regularly ...
while True:
messages: list[ApiBaseMessage] = await session.get_messages()
# throttling
sleep(3)
This is the first message that should be sent to the AI Engine for execution the function/s of your choice.
The main difference in here it is the AI Engine won't search, therefore decide for you, what is the apt function to fulfill your needs.
It contains the list of function-ids you want to execute and a function group (for secondary function picks).
Currently only supported by Next Generation personality. Don't use this if you already sent 'start' message.
# init the AI Engine client
from ai_engine_sdk import AiEngine
ai_engine: AiEngine = AiEngine(api_key)
# Create (do not start) a Session
session = await ai_engine.create_session(function_group=function_group.uuid)
# Execute function. You will receive no response.
await session.execute_function(function_ids=[function_uuid], objective="", context="")
# In order to get some feedback, gather the messages as regular.
while True:
messages: list[ApiBaseMessage] = await session.get_messages()
# throttling
sleep(3)
There are 5 different types of messages which are generated by the AI Engine and the SDK implements methods for checking the type of the respective new Message
:
-
task selection message:
is_task_selection_message
- this message is generated for example in the case when the AI engine recommends functions based on the initial objective or in the case when the AI Engine finds multiple options to provide as an input for a function. -
AI Engine message:
is_ai_engine_message
- this message type doesn't expect the user to reply to, this is for notifying the user about something -
confirmation message:
is_confirmation_message
- when the AI Engine has managed to acquire all the inputs for the agent belonging to the to-be-executed function (= it has managed to build the context), it sends this type of message to the user -
agent message:
is_agent_message
- this is a regular question that the user has to reply to with a string -
stop message:
is_stop_message
- this is the message which is sent when the session has stopped and the AI Engine doesn't wait for any replies from the user
All message types (except for the AI engine message and stop message) expect a response from the user and the SDK implements methods for sending reply in response to those different type of messages respectively.
The first argument of all these reply methods is the Message
object to which we want to send back the response.
The SDK methods you can use to reply to:
-
task selection message:
session.submit_task_selection
-
agent message:
session.submit_response
-
confirmation message:
session.submit_confirmation
orsession.reject_confirmation
- depending on if the user wants to confirm or reject the context generated by the AI engine respectively
After finishing the conversation with AI Engine you can delete the session like this:
await session.delete()
If you would like to check out a complete example on how to integrate AI Engine into your app, feel free to checkout examples/run_example.py.
- Test in different environments the function-group creation and sharing.
List the function belonging to the user owning the AV_API_TOKEN.
Share an existing function group by providing the id (you can fetch via 'list function groups'
All contributions are welcome! Remember, contribution includes not only code, but any help with docs or issues raised by other developers. See our contribution guidelines for more details.
We use GitHub Issues for tracking requests and bugs.