LLM::Graph

Raku package used to efficiently schedule and combine multiple LLM generation steps.

The package provides the class LLM::Graph with which computations are orchestrated.

The package follows the design discussed in the video "Live CEOing Ep 886: Design Review of LLMGraph", [WRIv1], and the corresponding Wolfram Language function LLMGraph, [WRIf1].

The package implementation heavily relies on the package "LLM::Functions", [AAp1]. Graph functionalities are provided by "Graph", [AAp3].

Installation

Package installations from both sources use zef installer (which should be bundled with the "standard" Rakudo installation file.)

To install the package from Zef ecosystem use the shell command:

zef install LLM::Graph

To install the package from the GitHub repository use the shell command:

zef install https://github.com/antononcube/Raku-LLM-Graph.git

Design

Creation of an LLM::Graph object in which "node_i" evaluates fun_i with results from parent nodes:

LLM::Graph.new({name_1 => fun_1, ...})

LLM::Graph objects are callables. Getting the result of a graph on input:

LLM::Graph.new(...)(input)

Details and options

An LLM::Graph enables efficient scheduling and integration of multiple LLM generation steps optimizing evaluation by managing the concurrency of LLM requests.
Using LLM::Graph requires (LLM) service authentication and internet connectivity.
- Authentication and internet are required if all graph nodes are non-LLM computation specs.
Possible values of the node function spec fun_i are:


`llm-function(...)`	an `llm-function` for LLM submission
`sub (...) {...}`	a sub for Raku computation submission
`%(key_i => val_i ...)`	a `Map` with detailed node specifications `nodespec`

Possible node specifications keys in nodespec are:


"eval-function"	arbitrary Raku sub
"llm-function"	LLM evaluation via an `llm-function`
"listable-llm-function"	threaded LLM evaluation on list input values
"input"	explicit list of nodes required as sub arguments
"test-function"	whether the node should run
"test-function-input"	explicit list of nodes required as test arguments

Each node must be defined with only one of "eval-function", "llm-function", or "listable-llm-function".
The "test-function" specification makes a node evaluation conditional on the results from other nodes.
Possible "llm-function" specifications prompt_i include:


"text"	static text
`["text1", ...]`	a list of strings
`llm-prompt("name")`	a repository prompt
`sub ($arg1..) {"Some $arg1 text"}`	templated text
`llm-function(...)`	an `LLM::Function` object

Any "node_i" result can be provided in input as a named argument. input can have one positional argument and multiple named arguments.
LLM::Graph objects have the attribute llm-evaluator that is used as a default (or fallback) LLM evaluator object. (See [AAp1].)

Usage examples

Three poets

Make an LLM graph with three different poets, and a judge that selects the best of the poet-generated poems:

use LLM::Graph;
use Graph;

my %rules =
        poet1 => "Write a short poem about summer.",
        poet2 => "Write a haiku about winter.",
        poet3 => sub ($topic, $style) {
            "Write a poem about $topic in the $style style."
        },
        judge => sub ($poet1, $poet2, $poet3) {
            [
                "Choose the composition you think is best among these:\n\n",
                "1) Poem1: $poet1",
                "2) Poem2: $poet2",
                "3) Poem3: $poet3",
                "and copy it:"
            ].join("\n\n")
        };

my $gBestPoem = LLM::Graph.new(%rules);

# LLM::Graph(size => 4, nodes => judge, poet1, poet2, poet3)

Calculation with special parameters (topic and style) for the 3rd poet:

$gBestPoem(topic => 'hockey', style => 'limerick');

# LLM::Graph(size => 4, nodes => judge, poet1, poet2, poet3)

Remark Instances of LLM::Graph are callables. Instead of $gBestPoem(...), $gBestPoem.eval(...) can be used.

Computations dependency graph:

$gBestPoem.dot(engine => 'dot', node-shape => 'ellipse', node-width => 1.2 ):svg

The result by the terminal node("judge"):

say $gBestPoem.rules<judge>;

# {eval-function => sub { }, input => [poet3 poet1 poet2], result => I think Poem1 is the best among these. Here's the composition:
# 
# Golden rays that warm the day,  
# Whispering breezes gently sway.  
# Fields alive with blooms so bright,  
# Endless days of pure delight.  
# 
# Laughter dances on the air,  
# Sun-kissed skin without a care.  
# Summer’s song, so sweet and clear,  
# A fleeting gift we hold so dear., test-function-input => [], wrapper => Routine::WrapHandle.new}

Implementation notes

LLM functors introduction

Since the very beginning, the functions produced by all functions were actually blocks (Block:D). It was in my TODO list instead of blocks to produce functors (function objects). For "LLM::Graph" that is/was necessary in order to make the node-specs processing more adequate.
- So, llm-function produces functors (LLM::Function objects) by default now.
- The option "type" can be used to get blocks.

No need for topological sorting

I thought that I should use the graph algorithms for topological sorting in order to navigate node dependencies during evaluation.
Turned out, that is not necessary -- simple recursion is sufficient.
- From the Notes specs, a directed graph (Graph object) is made.
- Graph's method reverse is used to get the directed computational dependency graph.
- That latter graph is used in the node-evaluation recursion.

Wrapping "string templates"

It is convenient to specify LLM functions with "string templates."
Since there are no separate "string template" objects in Raku, subs or blocks are used.
- For example:
- sub ($country, $year) {"What is the GDP of $country in $year"} (sub)
- {"What is the GDP of $^a in $^b?"} (block)
String template subs are wrapped to be executed first and then the result is LLM-submitted.
Since the blocks cannot be wrapped, currently "LLM::Graph" refuses to process them.
- It is planned later versions of "LLM::Graph" to process blocks.

Special graph plotting

Of course it is nice to have the graph visualized.
Instead of the generic graph visualization provided by the package "Graph" (method dot) a more informative graph plot is produced in which the different types of notes have different shapes.
- The graph vertex shapes help distinguishing LLM-nodes from just-Raku-nodes.
- Also, test function dependencies are designated with dashed arrows.

TODO

TODO Implementation
- DONE Initial useful version
  - Just using LLM::Graph.
- DONE Conditional evaluation per node
  - Using a test function
- TODO Front-end simple sub(s)
  - Like llm-graph.
- DONE Special DOT representation
- TODO CLI interface that takes Raku or JSON specs of LLM-graphs
- TODO Asynchronous execution support
TODO Testing
- DONE LLM-graph initialization
- TODO Simple evaluations
TODO Documentation
- DONE Useful README
- TODO Best poet notebook.
- TODO Comprehensive text summary notebook.
- TODO Visual dictionary
- TODO Demo video