The library for character-driven AI experiences.
Deploy AI personas for a variety of use cases, each with their own distinct knowledge and style.
Arthas Menethil, World of Warcraft
-
Scoped Knowledge: Using a generic chatbot like ChatGPT for narrow use cases like customer support, a game NPC, or writing code can yield undesired responses, or provide information outside the intended scope of knowledge. You don't want your MMORPG shopkeeper talking about about Ford F-150s or Chick-Fil-A, do you? Ragdoll scrapes a URL you provide as a knowledge source (usually a Wiki style web page, but could be anything - it's very flexible), and uses llamaindex to store and index that knowledge. It handles questions that fall outside of the scope of knowledge gracefully, so it will still feel like the user is interacting with a person even when it doesn't know the answer.
-
Distinct Personalities: Answers to questions are always rephrased from the first-person perspective in the style of a persona that you define. Because you're asked to define things like prose, tone, and even art style, Ragdoll is able to generate the appropriate prompts for your persona, resulting in statements the target persona would perceivably say.
-
Extensible: Ragdoll can be ran as an API, in a React app, as a CLI, or as a a dependency in your application. It uses Ollama for text so you can choose from a wide range of models, and defaults to Stable Diffusion (txt2img) for images.
Retrieval-Augmented Generation (RAG) is a strategy that helps address both LLM hallucinations and out-of-date or off-topic training data. The following diagram shows how data flows into Ragdoll from documents (web pages) and LLM-generated content through an indexed store and query engine before being presented:
You can interact with Ragdoll via this Node/React full stack application.
Note that in a default Terminal you will not see text colors and the image quality will be diminished. Using a Terminal like iTerm2 or Kitty will allow you to view the full resolution (512x512 by default).
In native Terminal with no addons:
Question: "what town are you from"
Answer:
In verbose mode with caching, and high-res image support:
Question: "why are you so mean"
Answer:
In verbose mode when he doesn't know the answer based on the knowledge he has:
Question: what is your favorite memory
For this one, llamaindex could not find any relevant info, resulting in this prompt fragment:
"Arthas's favorite memory is not explicitly mentioned in the context information provided."
Yet the prompt is still robust enough to provide a meaningful response in the style of Arthas:
"In the realm of my existence, a cherished memory lies concealed, veiled by the shadows of time. Its essence, though unspoken, resonates within my being. A tale of valor and darkness, woven intricately in the tapestry of my soul."
And we still get a relevant image:
Set up the environment. No API keys needed!
TEXT_MODEL_PROVIDER=LlamaIndex
TEXT_MODEL_URI=http://localhost:11434
TEXT_TEXT_MODEL=mistral
IMAGE_MODEL_PROVIDER=Stable Diffusion
IMAGE_MODEL_URI=http://localhost:7860
TEXT_IMAGE_MODEL=txt2img
IMAGE_IMAGE_MODEL=img2img
IMAGE_CFG_SCALE=8
IMAGE_CFG_SCALE_TRUE=24
IMAGE_DENOISING_STRENGTH=0.8
IMAGE_DENOISING_STRENGTH_TRUE=0.56
IMAGE_BATCH_SIZE=2
DELAY=200
RENDER=true
VERBOSE=true
GREETING=false
CACHE=true
MAX_STORAGE_KEY_LENGTH=32
LOG_PREFIX=<Ragdoll>
STORAGE_URI=./.tmp
- Download Ollama
Linux: curl -fsSL https://ollama.com/install.sh | sh
Windows & Mac: ollama.com/download
- Run the CLI
ollama start
- Find a model you like here and run it in your Terminal:
ollama run mistral
The Ollama (Mistral) API is now listening on http://localhost:11434/
-
Have Python 3 already installed
-
Navigate to the desired directory and
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
- Run the web UI
Linux & Mac: Run ./webui.sh --api --xformers --lowvram.
Windows: Run ./webui-user.bat --api --lowvram from Windows Explorer as normal, non-administrator, user.
Note: --lowvram is an optional flag, if running on a great machine (16GB+ vram) you can omit this.
The Stable Diffusion API is now listening on http://localhost:7860/
npm start
Ragdoll is now running in your terminal.
TEXT_TEXT_MODEL
The name of the text-to-text model you want to use (this should be running on http://localhost:11434). Example: mistral.
IMAGE_MODEL_URI
Example: http://localhost:7860.
DELAY
Delay between requests (in ms), for rate limiting, artificial delays, etc.
VERBOSE
Set to true to show all logs. Enable VERBOSE to see the generated prompts in your console, for example, in this case the query was "how many blood elves have you killed?":
<Ragdoll> Text (mistral) Prompt: Re-write the following message in the first-person, as if you are Arthas, in a style that is inspiring but grim, from the year 1200 A.D., using as few characters as possible (never exceed 500), in a tone that is slightly resentful, omitting any references to Earth or real-world society: Arthas killed Sylvanas Windrunner, King Anasterian Sunstrider, and Dar'Khan Drathir, who were blood elves. So, Arthas has killed three blood elves.
<Ragdoll> Text (mistral) responded with "I, Arthas, vanquished Sylvanas Windrunner, King Anasterian Sunstrider, and Dar'Khan Drathir, noble blood elves. Three lives claimed by my hand.".
<Ragdoll> Waiting 2 seconds...
<Ragdoll> Image (txt2img) Prompt: Render the following in the style of Blizzard's World of Warcraft concept art in high resolution like a finely-tuned video game model including each detail and anatomically correct features (if any): I, Arthas, vanquished Sylvanas Windrunner, King Anasterian Sunstrider, and Dar'Khan Drathir, noble blood elves. Three lives claimed by my hand.
CACHE
Set to true to cache inputs, llamaindex queries, LLM prompts, responses, & images.
The transformed input/prompt is what's cached, not the literal user input. For example, the questions "who are you", "explain who you are", and "who is arthas?" all transform to the same query ("Who is Arthas?"). The LLM responses are cached too, so you'll get the same answer when asking similar questions (but without having to request the LLM again).
MAX_STORAGE_KEY_LENGTH
How long storage keys can be. The keys are derived from queries/prompts, but there are key/value limits in localStorage and some prompts can be very long. An alternative to this config would be to make the developer provide a key (similar to React) each time remember is called, but that isn't supported right now.
STORAGE_URI
Path to a temp folder used for cache (default is ./.tmp).
Pass this config object to Ragdoll when you instantiate a new persona.
const myPersona = await Ragdoll({
cache,
greeting,
knowledgeURI,
name,
artStyle,
writingStyle,
writingTone,
query
});
// If there's a greeting or initial query
// an answer is already available
const { text: textAnswer } = myPersona.answer;
console.log(textAnswer);
// Now you can chat freely using
// the `chat` method
const { text: textAnswer2 } = await myPersona.chat('who are you?');
console.log(textAnswer2);- Llama2 Chat LLMs (70B, 13B, and 7B parameters)
- MistralAI Chat LLMs
- Fireworks Chat LLMs
- Stable Diffusion txt2img
- Stable Diffusion img2img
Support planned for image-to-animation, image-to-cgi, and image-to-film models.
Support planned for audio-to-music and audio-to-sfx models.
Support planned for a text-to-code model.
To ensure integrity, optionally integrate lifecycle middleware at 2 stages:
- LLM query: Run the formatted prompt through another transformer (like OpenAI GPT-4)
- Transformed response: Run the final image prompt through a different image model (like Leonardo Kino XL)
Instructions coming soon.