SlashGPT

SlashGPT is a playground for developers to make quick prototypes of LLM agents (or apps with Natural Language UI).

Here are the design goals:

Extremely easy to create a new LLM agent. You just need to add a new manifest file (in Json or YAML).
Instantly switch among agents, by just typing "/{agent_name}"
Extensible enough so that it is possible to implement most of LLM agents without writing any code.
It is possible to integrate ChatGPT plugins as agents without writing any code.
It enables broker agent (or dispatcher), which routes user's message to an appropriate agent.
It is able to run generated Python code like Code Interpreter (see "code" agent).

Note

If you want to try it out immediately, please try the Google Google Colaboratory version.

Initialization for developer

Install the required packages:

If you want to use all the functions, install this. pip install -r requirements/full.txt
If you want to use only the general features, install this pip install -r requirements.txt

Create .env file, and specify your OpenAI key as follows:

OPENAI_API_KEY=...
You need to specify other variables to use following features (optional).
- PINECONE_API_KEY, PINECONE_ENVIRONMENT: required to use embedding vector db.
- GOOGLE_PALM_KEY: required to switch to "palm" model with /llm command.
- REPLICATE_API_TOKEN: required to switch to "llama2" model with /llm command.
- CODEBOX_API_KEY: set this to "local" to use CodeBox's LocalBox instead of IPython
- SLASH_GPT_ENV_WOLFRAM_API_KEY: required to use "walfram" agent.
- SLASH_GPT_ENV_OPENWEATHER_API_KEY: required to use "weather" agent.
- SLASH_GPT_ENV_NOTEABLE_API_KEY: required to use "noteable" agent.
- SLASH_GPT_ENV_WEBPILOT_UID: required to use "webpilot" agent (any unique UUID is fine)
- ALCHEMY_API_KEY: required to use "web3" agent.

API Documentation

pdoc src/slashgpt

https://snakajima.github.io/SlashGPT/

Execution

Type ./SlashGPT.py
When you see "You({agent_name}):", type a message to the agent OR type a slash command starting with "/".
It activate "dispatcher" agent first, which is able to dispatch queries to appropriate agents.
Type "/help" to see the list of system commands and available agents.

Execution on Docker

Build docker image docker build -t slashgpt .
Run SlashGPT on Docker docker run -it slashgpt ./SlashGPT.py

Outputs

Each conversation will be store as a json file under the "output/{context}" folder, where the context is "GTP" for general chat, and the app id for a specialized chat.
Please notice that the "output" folder is ignored by git.
Code Interpreter agents will generate Jupyter notebook in "output/notebooks" folder.

Code Interpreter Agents

Some of agents are built to mimic the behaviors of ChatGPT code interpreter with various LLMs.

code: GPT3.5
code_palm: PaLM2 (GOOGLE_PALM_KEY key is required)
code_llama: LlaMA (REPLICATE_API_TOKEN is required)

code (GPT3.5) works just like OpenAI's Code Interpreter. It is able to execute the output of generated code appropriately.

code_palm2 (PaLM2) and code_llma (LlaMA) are not able to execute the output of generated code (they often enter into an infinite loop). Therefore, we stop the conversation after the output ("skip_function_result": true in the manifest), and the user needs to explicitly ask it to run the generated code.

For the runtime, it uses IPython by default, but it uses CodeBox if you specify CODEBOX_API_KEY key. IPython displays images as popups, but does not write them into the notebook. CodeBox is able to write them into the notebook.

Sample queries.

Draw sine curve
List first 50 prime numbers
graph common moving average
Draw histogram
Graph 4 year stock price of apple and tesla using yfinance

Manifest files

Create a new manifest file, {agent_name}.json/yml in "manifests" folder with following properties:

title (string, required): Title for the user to see
about (string, optional): About the manifest (URL, email, github id, or twitter id)
description (string, optional): Human/LLM readable description of this agent
prompt (array of strings, required): The system prompts which define the agent (required)
form (string): format string to extend user's query (e.g. "Write python code to {question}").
result_form (string): format string to extend function call result.
skip_function_result (boolean): skip the chat completion right after the function call.
notebook (boolean): create a new notebook at the beginning of each session (for code_palm2)
bot (string, optional): Agent name. The default is Agent({agent_name}).
you (string, optional): User name. The default is You({agent_name}).
sample or smaple... (string, optional): Sample question (type "/sample" to submit it as the user message)
intro (array of strings, optional): Introduction statements (will be randomly selected)
model (string or dict, optional): LLM model (such as "gpt-4-613", the default is "gpt-3-turbo")
temperature (number, optional): Temperature (the default is 0.7)
stream (boolean, optional): Enable LLM output streaming (not yet implemented)
logprobs (number, optional): Number of "next probable tokens" + associated log probabilities to return alongside the output
num_completions (number, optional): Number of different completions to request from the model per prompt
list (array of string, optional): {random} will put one of them randomly into the prompt
embeddings (object, optional):
- name (string, optional): index name of the embedding vector database
resource (string, optional): location of the resource file. Use {resource} to paste it into the prompt
functions (string or list, optional): string - location of the function definitions, list - function definitions
function_call (string, optional): the name of tne function LLM should call
module (string, optional): location of the Python script to be loaded for function calls
actions (object, optional): Template-based function processor (see details below)

Name of that file becomes the slash command. (the slash command of "foo.json" is "/foo")

Actions

It defines template-based function implementations (including mockups), alternative to writing Python code using the "module" property.

It supports four different methods.

1. Formatted string (type="message_template").

Use this method to develop the front-end of a system before the backend become ready.

message (format string, required): chat message to be added
manifest (format string, optional): manifest file name to be loaded for chained action

Here is an example (home).

  "actions": {
    "fill_bath": { "type": "message_template", "message":"Success. I started filling the bath tab." },
    "set_temperature": { "type": "message_template", "message":"Success. I set the temperature to {temperature} for {location}" },
    "start_sprinkler": { "type": "message_template", "message":"Success. I started the sprinkler for {location}" },
    "take_picture": { "type": "message_template", "message":"Success. I took a picture of {location}" },
    "play_music": { "type": "message_template", "message":"Success. I started playing {music} in {location}" },
    "control_light": { "type": "message_template", "message":"Success. The light switch of {location} is now {switch}." }
  }

2. REST calls (type="rest")

Use this method to call REST APIs (equivalent to ChatGPT's plugin system).

url (string, required): Python-style format string, which references to function arguments.
method (string, optional): Specify "POST" if we need to use HTTP-POST. The body will contain a JSON representation of function parameters.

Here is an example (currency).

  "actions": {
    "convert": {
      "type": "rest",
      "url": "https://today-currency-converter.oiconma.repl.co/currency-converter?from={from}&to={to}&amount={amount}"
    }
  }

3. GraphQL calls (type="graphQL")

Use this method to call GraphQL APIs.

url (string, required): GraphQL server url.

Here is an example (spacex).

  "actions": {
    "convert": {
      "type": "graphQL",
      "url": "https://spacex-production.up.railway.app/graphql"
    }
  }

4. data URL (type="data_url")

This method allows a developer to generate a text data (typically in JSON, but not limited to), and turn it into a data URL.

template (string, required): The location of the template file.
mime_type (string, required): The mime type of the data.
message (string, required): Python-style format string, which references to the data-URL as {url}.

Here is an example for "make_event" function (cal).

  "actions": {
    "make_event": {
      "type": "data_url",
      "template": "./resources/calendar.ics",
      "mime_type": "text/calendar",
      "message": "The event was scheduled. Here is the invitation link: '{url}'"
    }
  }

The contents of calendar.ics file.

BEGIN:VCALENDAR
VERSION:2.0
PRODID:-//My Calendar//NONSGML v1.0//EN
BEGIN:VEVENT
DTSTART:{DTSTART}
DTEND:{DTEND}
SUMMARY:{SUMMARY}
DESCRIPTION:{DESCRIPTION}
LOCATION:{LOCATION}
END:VEVENT
END:VCALENDAR

The definition of "make_event" function.

{
  "name": "make_event",
  "description": "Create a calendar event in iCalendar format",
  "parameters": {
    "type": "object",
    "properties": {
      "SUMMARY": {
        "type": "string",
        "description": "a short, one-line description of the event"
      },
      "DESCRIPTION": {
        "type": "string",
        "description": "a more complete description of the calendar",
        "maxLength": 400
      },
      "DTSTART": {
        "type": "string",
        "format": "date-time",
        "description": "the date and time in UTC that the event begins such as 19980119T020000Z"
      },
      "DTEND": {
        "type": "string",
        "format": "date-time",
        "description": "the date and time in UTC that the event ends such as 19980119T030000Z"
      },
      "LOCATION": {
        "type": "string",
        "description": "the intended venue with address for the event."
      }
    },
    "required": ["SUMMARY", "DTSTART", "DTEND", "DESCRIPTION", "LOCATION"]
  }
}

5. Emit (type="emit")

It will emit an event to the caller (of ChatSession:call_loop).

The "emit_method" determines the action and the "emit_data" defines parameters.

ChatApplication implements the "switch_session" method, which takes "message", "agent" and "memory" properties.

message(str, optional): the initial user message to be given to the new agent.
agent(str): the agent key
memory(dict, optional): the short-term memory to be passed to the new agent.

Here is an example (dispatcher):

  "actions": {
    "categorize": {
      "type": "emit",
      "emit_method": "switch_session",
      "emit_data": {
        "message": "{question}",
        "agent": "{category}"
      }
    }
  },

Standard Test Sequence

Automated.

./SlashGPT.py
/autotest

This is the standard manual test sequence.

# Test REST API
Input: /sample currency
Expected Output: 1 USD is equivalent to 146.31 JPY.
Input: /switch main

# Test REST API with appkey header (WEBPILOT_UID in .env is required)
Input: /sample webpilot
Expected Output: Title: France riorts ...\nSummary: The fourth night ...
Input: /switch main

# Test GraphQL
Input: /sample spacex
Expected Output: The CEO of SpaceX is Elon Musk.
Input: /switch main

# Test DataURL
Input: /sample cal
Expected Output: I have scheduled a meeting with Tim Cook on July 4th at 8:00 PM UTC for 30 minutes. The meeting will be held at Tim Cook's office. I have sent the invitation to Tim Cook at tim@apple.com.
Input: /switch main

# Test Code Interpreter
Input: /code
Input: /sample_stock
Expected Output: <Marketcap history of Apple and Tesla>

Unittest

Run test

make test

Code formatter and linter

Lint

make lint

Format

make format

mfunaki/SlashGPT