Connectors for your agent to the outside world.
- Discord connector with (voice and chat, DMs coming)
- Twitter connector (feed only, DMs coming)
- Admin Panel - simple web interface to chat with your agent and upload files
pip install agentcomms
This module uses a set of environment variables to interact with Twitter, so you'll need to set the following before using:
TWITTER_EMAIL
: The email address for your Twitter account.TWITTER_USERNAME
: The username for your Twitter account.TWITTER_PASSWORD
: The password for your Twitter account.
Before you can start using the Twitter connector, you have to initialize it. The initialization is done using the start_twitter_connector
function.
import twitter
twitter.start_twitter_connector()
This will start the twitter connector with default parameters. If you wish to customize the email, username, password, or session storage path, you can use the start_twitter
function like so:
twitter.start_twitter(email="[email protected]", username="my_username", password="my_password", session_storage_path="my_session.cookies")
To like a tweet, you can use the like_tweet
function. Pass in the id of the tweet you wish to like.
twitter.like_tweet("1234567890")
To reply to a tweet, you can use the reply_to_tweet
function. Pass in the message you wish to send, and the id of the tweet you're replying to.
twitter.reply_to_tweet("This is a great tweet!", "1234567890")
To post a new tweet, you can use the tweet
function. Pass in the message you wish to tweet. You can optionally pass in a media object to attach to the tweet.
twitter.tweet("Hello, Twitter!")
Feed handlers are functions that get called whenever there are new tweets in the feed. They can be registered using the register_feed_handler
function.
def my_feed_handler(tweet):
print(f"New tweet from {tweet['user']['name']}: {tweet['text']}")
twitter.register_feed_handler(my_feed_handler)
You can also unregister a handler using the unregister_feed_handler
function.
twitter.unregister_feed_handler(my_feed_handler)
To get the current account object, you can use the get_account
function.
account = twitter.get_account()
print(account.username)
This will print out the username of the current Twitter account.
The Discord connector works with both voice and text. For voice, you will need an Elevenlabs API key.
Before you start, you need to set the environment variables for the bot to function correctly. Create a .env
file in your project directory and set these variables:
DISCORD_API_TOKEN=your_discord_bot_token
ELEVENLABS_API_KEY=your_elevenlabs_api_key
ELEVENLABS_VOICE=voice_you_want_to_use
ELEVENLABS_MODEL=model_you_want_to_use
DISCORD_API_TOKEN
is your Discord bot token, which you get when you create a new bot on the Discord developer portal.ELEVENLABS_API_KEY
is your Eleven Labs API key for their TTS service.ELEVENLABS_VOICE
is the voice you want to use for the TTS. You will have to check the Eleven Labs API documentation for the voices they support.ELEVENLABS_MODEL
is the TTS model you want to use. Again, you will have to check the Eleven Labs API documentation for the supported models.
After setting your environment variables, you can run your bot by calling the start_connector
function:
start_connector()
Message handlers are functions that are executed when certain events happen in Discord, such as receiving a message. Here's how you can register a message handler:
First, you need to create a function that will be executed when a message is received. This function should take one argument, which is the message that was received. The message object will contain all the information about the message, such as the content of the message, the author, and the channel where it was sent.
Here's an example of a simple message handler function:
def handle_message(message):
print(f"Received a message from {message.author}: {message.content}")
This function will simply print the author and content of every message that is received.
To register the handler function, you use the register_feed_handler
function and pass the handler function as an argument:
register_feed_handler(handle_message)
After calling this function, the handle_message
function will be executed every time a message is received on Discord.
This function is used to add a message to the queue. The message will be sent to the channel with the ID specified.
send_message("Hello world!", 1234567890)
This function is used to start the bot and the event loop, setting the bot to listen for events on Discord.
start_connector("your_discord_api_token")
This function is used to register a new function as a feed handler. Feed handlers are functions that process or respond to incoming data in some way.
def my_func(data):
print(data)
register_feed_handler(my_func)
This function is used to remove a function from the list of feed handlers.
unregister_feed_handler(my_func)
- Start the server: You can start the server with uvicorn like this:
import os
if __name__ == "__main__":
import uvicorn
uvicorn.run("agentcomms:start_server", host="0.0.0.0", port=int(os.getenv("PORT", 8000)))
This will start the server at https://localhost:8000
.
- Get a file:
Once the server is up and running, you can retrieve file content by sending a GET request to
/file/{path}
endpoint, where{path}
is the path to the file relative to the server's current storage directory.
from agentcomms import get_file
# Fetches the content of the file located at "./files/test.txt"
file_content = get_file("test.txt")
print(file_content)
- Save a file:
Similarly, you can save content to a file by sending a POST request to
/file/
endpoint, with JSON data containing thepath
andcontent
parameters.
from agentcomms import add_file
# Creates a file named "test.txt" in the current storage directory
# and writes "Hello, world!" to it.
add_file("test.txt", "Hello, world!")
AgentFS provides the following public functions:
Starts the FastAPI server. If a storage_path
is provided, it sets the storage directory to the given path.
Arguments:
storage_path
(str, optional): The path to the storage directory.
Returns:
- None
Example:
from agentcomms import start_server
start_server("/my/storage/directory")
Returns the FastAPI application instance.
Arguments:
- None
Returns:
- FastAPI application instance.
Example:
from agentcomms import get_server
app = get_server()
Sets the storage directory to the provided path.
Arguments:
new_path
(str): The path to the new storage directory.
Returns:
True
if the path was successfully set,False
otherwise.
Example:
from agentcomms import set_storage_path
set_storage_path("/my/storage/directory")
Creates a file at the specified path and writes the provided content to it.
Arguments:
path
(str): The path to the new file.content
(str): The content to be written to the file.
Returns:
True
if the file was successfully created.
Example:
from agentcomms import add_file
add_file("test.txt", "Hello, world!")
Removes the file at the specified path.
Arguments:
path
(str): The path to the file to be removed.
Returns:
True
if the file was successfully removed.
Example:
from agentcomms import remove_file
remove_file("test.txt")
Appends the provided content to the file at the specified path.
Arguments:
path
(str): The path to the file to be updated.content
(str): The content to be appended to the file.
Returns:
True
if the file was successfully updated.
Example:
from agentcomms import update_file
update_file("test.txt", "New content")
Lists all files in the specified directory.
Arguments:
path
(str, optional): The path to the directory. Defaults to'.'
(current directory).
Returns:
- A list of file names in the specified directory.
Example:
from agentcomms import list_files
files = list_files()
Lists all files in the specified directory as a formatted string. Convenient!
Arguments:
path
(str, optional): The path to the directory. Defaults to'.'
(current directory).
Returns:
- A string containing a list of file names in the specified directory.
Example:
from agentcomms import list_files
files = list_files()
Returns the content of the file at the specified path.
Arguments:
path
(str): The path to the file.
Returns:
- A string containing the content of the file.
Example:
from agentcomms import get_file
content = get_file("test.txt")
If you like this library and want to contribute in any way, please feel free to submit a PR and I will review it. Please note that the goal here is simplicity and accesibility, using common language and few dependencies.