diff options
-rw-r--r-- | pgcli/pgstyle.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/pgcli/pgstyle.py b/pgcli/pgstyle.py index 2ee3e200..82290370 100644 --- a/pgcli/pgstyle.py +++ b/pgcli/pgstyle.py @@ -38,6 +38,8 @@ TOKEN_TO_PROMPT_STYLE = { Token.Literal.String: "literal.string", Token.Literal.Number: "literal.number", Token.Keyword: "keyword", + Token.Prompt: "prompt", + Token.Continuation: "continuation", } # reverse dict for cli_helpers, because they still expect Pygments tokens. |