Skip to content

Language server for text spell and grammar check with various tools.

License

Notifications You must be signed in to change notification settings

hangyav/textLSP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

textLSP

Language server for text spell and grammar check with various AI tools.

This tool is in early development.

textLSP

Features

LSP features

  • Diagnostics:
    • spelling or grammatical errors
  • Code actions:
    • Fix suggestions
    • Analyze paragraph with a selected passive analyzer (if the analyzer does not check on save or change)
      Showcase
    • Only on the first character of the first line: analyze the whole document if it was not fully checked yet
      Showcase
    • Custom actions defined by a given analyzer
      E.g. OpenAI text generation
  • Context based word suggestion
    Showcase

Analyzers

Local tools

The following tools run on the local system:

  • LanguageTool: Mainly for development purposes, see ltex-ls for a more mature implementation.
  • Ollama: Run LLMs efficiently on your local machine. It supports diagnostics, code actions and prompt based text generation.
    • Ollama needs to be installed manually first.
    • Various LLMs are supported, such as Llama 3, Gemma or Mixtra. Suggested model is Phi3, due to its speed, size and accuracy.
  • hf_checker: Huggingface text2text-generation pipeline based analyser. See the flan-t5-large-grammar-synthesis model for an example.
    Models
    • pszemraj/grammar-synthesis-small
    • pszemraj/grammar-synthesis-large
    • pszemraj/flan-t5-large-grammar-synthesis
    • pszemraj/flan-t5-xl-grammar-synthesis
    • pszemraj/bart-base-grammar-synthesis
  • hf_instruction_checker: Huggingface text2text-generation pipeline based analyser using instruction tuned models. See the Grammarly's CoEdIT model for an example. Supports error checking and text generation, such as paraphrasing, through the %HF% magic command (see the OpenAI analyser below).
    Models
    • grammarly/coedit-large
    • grammarly/coedit-xl
    • grammarly/coedit-xl-composite
    • grammarly/coedit-xxl
    • jbochi/coedit-base
    • jbochi/coedit-small
    • jbochi/candle-coedit-quantized
  • hf_completion: Huggingface fill-mask pipeline based text completion.
  • Gramformer: Neural network based system.

Tools using remote services

DISCLAIMER: THE RELATED APIS REQUIRE REGISTRATION AND ARE NOT FREE TO USE! USE THESE ANALYZERS ON YOUR OWN RESPONSIBILITY! THE AUTHORS OF TEXTLSP DO NOT ASSUME ANY RESPONSIBILITY FOR THE COSTS INCURRED!

The following tools use remote text APIs. Due to potential costs turning off automatic analysis if suggested.

  • OpenAI: Supports text correction as well as text generation through a magic command in the text file.
    • A custom URL can be set to use an OpenAI-compatible server. See the example configuration below.

      Generation showcase
  • GrammarBot: The GrammarBot API provides spelling and grammar checking.

Supported File Types

  • latex
  • org
  • markdown
  • any other file types as plain text

Setup

Install

pip install textLSP

For the latest version:

pip install git+https://github.com/hangyav/textLSP

Additional dependencies

Some analyzers need additional dependencies!

  • hf_checker, hf_instruction_checker and hf_completion:
pip install textLSP[transformers]
  • Gramformer needs to be installed manually:
pip install git+https://github.com/PrithivirajDamodaran/Gramformer.git

Running

Simply run:

textlsp

Since some analyzers are computation intensive, consider running it on a server using the TCP interface:

textlsp --address 0.0.0.0 --port 1234

or simply over ssh (with ssh key) if the client doesn't support it:

ssh <server> textlsp

Configuration

Using textLSP within an editor depends on the editor of choice. For a few examples how to set up language servers in general in some of the popular editors see here or take a look at the related documentation of your editor.

By default, all analyzers are disabled in textLSP, they have to be turned on in the settings. Example configuration in lua for nvim (other editors should be set up accordingly):

textLSP = {
    analysers = {
        languagetool = {
            enabled = true,
            check_text = {
                on_open = true,
                on_save = true,
                on_change = false,
            }
        },
        ollama = {
          enabled = true,
          check_text = {
            on_open = false,
            on_save = true,
            on_change = false,
          },
          model = "phi3:3.8b-instruct",  -- smaller but faster model
          -- model = "phi3:14b-instruct",  -- more accurate
          max_token = 50,
        },
        gramformer = {
            -- gramformer dependency needs to be installed manually
            enabled = false,
            gpu = false,
            check_text = {
                on_open = false,
                on_save = true,
                on_change = false,
            }
        },
        hf_checker = {
            enabled = false,
            gpu = false,
            quantize=32,
            model='pszemraj/flan-t5-large-grammar-synthesis',
            min_length=40,
            check_text = {
                on_open = false,
                on_save = true,
                on_change = false,
            }
        },
        hf_instruction_checker = {
            enabled = false,
            gpu = false,
            quantize=32,
            model='grammarly/coedit-large',
            min_length=40,
            check_text = {
                on_open = false,
                on_save = true,
                on_change = false,
            }
        },
        hf_completion = {
            enabled = false,
            gpu = false,
            quantize=32,
            model='bert-base-multilingual-cased',
            topk=5,
        },
        openai = {
            enabled = false,
            api_key = '<MY_API_KEY>',
            -- url = '<CUSTOM_URL>'  -- optional to use an OpenAI-compatible server
            check_text = {
                on_open = false,
                on_save = false,
                on_change = false,
            },
            model = 'gpt-3.5-turbo',
            max_token = 16,
        },
        grammarbot = {
            enabled = false,
            api_key = '<MY_API_KEY>',
            -- longer texts are split, this parameter sets the maximum number of splits per analysis
            input_max_requests = 1,
            check_text = {
                on_open = false,
                on_save = false,
                on_change = false,
            }
        },
    },
    documents = {
        -- the language of the documents, could be set to `auto` of `auto:<fallback>`
        -- to detect automatically, default: auto:en
        language = "auto:en",
        -- do not autodetect documents with fewer characters
        min_length_language_detect = 20,
        org = {
            org_todo_keywords = {
                'TODO',
                'IN_PROGRESS',
                'DONE'
            },
        },
        txt = {
            parse = true,
        },
    },
}