NeuroAI data-driven System for multi-Agent Networks
Project description
Neuro-San Data-Driven Agents
Running client and server
Prep
Set up your virtual environment per instructions here
Direct Setup
From the top-level of this repo:
python -m neuro_san.client.agent_cli --connection direct --agent hello_world
Type in this input to the chat client:
I am travelling to a new planet and wish to send greetings to the orb.
What should return is something like:
Hello, world.
... but you are dealing with LLMs. Your results will vary!
Client/Server Setup
Server
In the same terminal window, be sure the environment variable(s) listed above are set before proceeding.
Option 1: Run the service directly. (Most useful for development)
python -m neuro_san.service.agent_main_loop
Option 2: Build and run the docker container for the hosting agent service:
./neuro_san/deploy/build.sh ; ./neuro_san/deploy/run.sh
You will need the leaf-common and leaf-server-common wheel files for this to work.
These build.sh / Dockerfile / run.sh scripts are portable so they can be used with
your own projects' registries and coded_tools work.
Client
In another terminal start the chat client:
python -m neuro_san.client.agent_cli --connection service --agent hello_world
Extra info about agent_cli.py
There is help to be had with --help.
By design, you cannot see all agents registered with the service from the client.
When the chat client is given a newline as input, that implies "send the message". This isn't great when you are copy/pasting multi-line input. For that there is a --first_prompt_file argument where you can specify a file to send as the first message.
You can send private data that does not go into the chat stream as a single escaped string of a JSON dictionary. For example: --sly_data "{ "login": "" }"
Running Python unit/integration tests
To run Python unit/integration tests, follow the instructions here
Creating a new agent network
Agent example files
Look at the hocon files in ./neuro_san/registries for examples of specific agent networks.
The natural question to ask is: What is a hocon file? The simplest answer is that you can think of a hocon file as a JSON file that allows for comments.
Here are some descriptions of the example hocon files provided in this repo. To play with them, specify their stem as the argument for --agent on the agent_cli.py chat client. In some order of complexity, they are:
-
hello_world
This is the initial example used above and demonstrates a front-man agent talking to another agent downstream.
-
esp_decision_assistant
This is Babak's original decision assistant. Very abstract, but also very powerful. A front man agent gathers information about a decision to make in ESP terms. It then calls a prescriptor which in turn calls one or more predictors in order to help make the decision in an LLM-based ESP manner.
When coming up with new hocon files in that same directory, also add an entry for it in the manifest.hocon file.
build.sh / run.sh the service like you did above to re-load the server, and interact with it via the agent_cli.py chat client, making sure you specify your agent correctly (per the hocon file stem).
More agent example files
For more examples of agent networks, documentation and tutorials, see this repo: https://github.com/leaf-ai/neuro-san-demos
Manifest file
All agents used need to have an entry in a single manifest hocon file. For the neuro-san repo, this is: neuro_san/registries/manifest.hocon.
When you create your own repo for your own agents, that will be different and you will need to create your own manifest file. To point the system at your own manifest file, set a new environment variable:
export AGENT_MANIFEST_FILE=<your_repo>/registries/manifest.hocon
Infrastructure
The agent infrastructure is run as a gRPC service. That gRPC service is implemented (client and server) using this interface:
https://github.com/leaf-ai/neuro-san/blob/main/neuro_san/session/agent_session.py
It has 2 main methods:
-
function()
This tells the client what the top-level agent will do for it.
-
streaming_chat()
This is the main entry point. Send some text and it starts a conversation with a "front man" agent. If that guy needs more information it will ask you and you return your answer via another call to the chat() interface. ChatMessage Results from this method are streamed and when the conversation is over, the stream itself closes after the last message has been received.
ChatMessages of various types will come back over the stream. Anything of type AI is the front-man answering you on behalf of the rest of its agent posse, so this is the kind you want to pay the most attention to.
-
Other methods like chat(), logs(), and reset() are legacy
Implementations of the AgentSession interface:
- DirectAgentSession class. Use this if you want to call neuro-san as a library
- ServiceAgentSession class. Use this if you want to call neuro-san as a client to a service
Note that agent_cli uses both of these. You can look at the source code there for examples.
There is also an AsyncServiceAgentSession implementation available
Advanced concepts
Coded Tools
Most of the examples provided here show how no-code agents are put together, but neuro-san agent networks support the notion of coded tools for low-code solutions.
These are most often used when an agent needs to call out to a specific web service, but they can be any kind of Python code as long it derives from the CodedTool interface defined in neuro_san/interfaces/coded_tool.py. See the class and method comments there for more information.
When you develop your own coded tools, there is another environment variable that comes into play:
export AGENT_TOOL_PATH=<your_repo>/coded_tools
Beneath this, classes are dynamically resolved based on their agent name. That is, if you added a new coded tool to your agent, its file path would look like this:
<your_repo>/coded_tools/<your_agent_name>/<your_coded_tool>.py
Creating Clients
To create clients, follow the instructions here
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file neuro_san-0.5.15.tar.gz.
File metadata
- Download URL: neuro_san-0.5.15.tar.gz
- Upload date:
- Size: 846.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.12.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
789c7746d2f61c2b08514d5def18d9538e9d6618e7d0e66773ea6924c606905f
|
|
| MD5 |
4fd8666af28e7a562cd05acdb384a049
|
|
| BLAKE2b-256 |
847e6dbe7d76640ad624aed6440f16696851ae54c338422b3254be5c58d6ff1f
|
File details
Details for the file neuro_san-0.5.15-py3-none-any.whl.
File metadata
- Download URL: neuro_san-0.5.15-py3-none-any.whl
- Upload date:
- Size: 962.4 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.12.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
9dd1c5b39ab8736362eb989ce9dfd69665ba901560f461914408f61e1f7914cc
|
|
| MD5 |
4adb8b5d6eac5603cb9a836109968711
|
|
| BLAKE2b-256 |
cc0b7dc3c60529bb0cfd0d2db8fd4b8d18ddca1ddd109cc1eac6006996b34503
|