A lightweight Python framework for building modular AI pipelines with function nodes and agent nodes.
Project description
Pipeline Node Agents
A lightweight Python framework for building modular AI pipelines with function nodes and agent nodes. Designed to work well with lightweight local LLMs by giving you full control over context and task complexity at each step.
Usage Guide
Requirements
- OS: Ubuntu 20.04 or later
- RAM: 8 GB minimum (16 GB recommended)
- Disk: 10 GB free space
- curl (any recent version)
- Python 3.11 - 3.13
1) Setup Ollama
-
Install Ollama (if not already installed):
curl -fsSL https://ollama.com/install.sh | sh
-
Run Ollama in a separate terminal:
ollama serve -
Install LLM (default:
llama3.2:latest):ollama pull llama3.2:latest
-
Make sure the required LLM is installed using command:
ollama list
Expected output:
NAME ID SIZE MODIFIED
llama3.2:latest 9f1c3d6a5b8e 2.0 GB 1 minute ago
2) Install Package
-
(Optional, but recommended) create python virtual environment:
python3 -m venv .venv && source .venv/bin/activate
-
Install package:
pip install pipeline-node-agents
3) Verify Installation
from pipeline_node_agents import greet
print(greet())
Expected output:
Hello, World! Pipeline Node Agents <version> is working.
4) Run Pipelines
Option 1: With defined model and logger (recommended)
from crewai import LLM
from pipeline_node_agents import init_pipeline_logger, get_logger, TripPlannerPipeline
init_pipeline_logger(pipeline_name="trip_planner_pipeline", project_root=".")
logger = get_logger(__name__)
# Default model is llama3.2, change if needed
ollama_llm = LLM(model="ollama/llama3.2", base_url="http://localhost:11434")
pipeline = TripPlannerPipeline(ollama_llm=ollama_llm, logger=logger)
pipeline.run()
Option 2: Without logger
from pipeline_node_agents import TripPlannerPipeline
# Default model is llama3.2
pipeline = TripPlannerPipeline()
pipeline.run()
Available tools configuration
As default, agent nodes work as pure LLMs (without tools). You can allow them to search for information by themselves.
Note: Usage of tools is not recommended for light-weight LLMs (<8 billion hyperparameters).
To enable tools for specific agents in TripPlannerPipeline:
-
Enable tools for local expert agent only:
pipeline = TripPlannerPipeline(local_expert_tools_enabled=True)
-
Enable tools for travel concierge agent only:
pipeline = TripPlannerPipeline(travel_concierge_tools_enabled=True)
-
Enable tools for both agents:
pipeline = TripPlannerPipeline(local_expert_tools_enabled=True, travel_concierge_tools_enabled=True)
Available pipelines
The following pipelines are included as examples:
-
ConditioningPipeline: Randomly chooses whether to go to a park or cinema, and suggests either a film or a park in Vienna. -
InputCheckerPipeline: Takes text as input and returns true if the text is a list of populated places (cities). Can be run as a loop, returning to the input step as long as the input is invalid:pipeline.run(loop=True) -
RandomMeanPipeline: Simple pipeline without LLM usage that generates random numbers and calculates their mean. -
RandomMeanPipelineCrewAI: Generates random numbers and summarizes them using an LLM. -
SearchAndSummarizePipeline: Searches for the best country for business using DuckDuckGo and summarizes results using an LLM, returning the best country for business. -
TripPlannerPipeline: Takes a list of cities and trip dates as input, chooses the city based on weather conditions, and creates a full 7-day trip itinerary for the provided dates.
To run one of them, replace "TripPlannerPipeline" in the code example with a class from the list above.
Example with SearchAndSummarizePipeline:
from pipeline_node_agents import SearchAndSummarizePipeline
pipeline = SearchAndSummarizePipeline()
pipeline.run()
Maintenance Guide
For detailed maintenance instructions, including how to create nodes, build pipelines, and extend the framework, please refer to the Maintenance Guide.
Project details
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 pipeline_node_agents-0.1.5.tar.gz.
File metadata
- Download URL: pipeline_node_agents-0.1.5.tar.gz
- Upload date:
- Size: 18.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.9.6
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
6f3c57d8da5fc4ced23596bc6f2f5c091779ad62851eb18ea93a0c326ed25d0b
|
|
| MD5 |
9db4a4ebfec24329eabc40aa65e925db
|
|
| BLAKE2b-256 |
990e5bae8a0f1640d2b9847592af76186614c83acb76c60870fd4c4c8ffa2995
|
File details
Details for the file pipeline_node_agents-0.1.5-py3-none-any.whl.
File metadata
- Download URL: pipeline_node_agents-0.1.5-py3-none-any.whl
- Upload date:
- Size: 31.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.9.6
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
3151537749ac90485e772bf25659640596d1e83a6b8fcecc9bbc65d740d7ee55
|
|
| MD5 |
a673cda9f7588f2720ed43dc5437a996
|
|
| BLAKE2b-256 |
eaef98282d9bab72d606f900805671869f8529a708dbfba2d3bc03ce4402012e
|