Project

baran

0.03
The project is in a healthy, maintained state
Text Splitter for Large Language Model Datasets.
2005
2006
2007
2008
2009
2010
2011
2012
2013
2014
2015
2016
2017
2018
2019
2020
2021
2022
2023
2024
 Dependencies
 Project Readme

Baran

v dt license

Text Splitter for Large Language Model datasets.

To avoid token constraints and improve the accuracy of vector search in the Large Language Model, it is necessary to divide the document. This gem supports splitting the text in the specified manner.

Installation

Install the gem and add to the application's Gemfile by executing:

$ bundle add baran

If bundler is not being used to manage dependencies, install the gem by executing:

$ gem install baran

Usage

Default Paramters

  • chunk_size: 1024 (characters)
  • chunk_overlap: 64 (characters)

Character Text Splitter

Splitting by the specified character.

splitter = Baran::CharacterTextSplitter.new(
    chunk_size: 1024,
    chunk_overlap: 64,
    separator: "\n\n"
)
splitter.chunks(text, metadata: { ... })
# => [{ cursor: 0, text: "...", metadata: { ... } }, ...]

Recursive Character Text Splitter

Splitting by the specified characters recursively.

splitter = Baran::RecursiveCharacterTextSplitter.new(
    separators: ["\n\n", "\n", " ", ""]
)
splitter.chunks(text, metadata: { ... })
# => [{ cursor: 0, text: "...", metadata: { ... } }, ...]

Markdown Text Splitter

Splitting by the Markdown descriptions.

splitter = Baran::MarkdownSplitter.new
splitter.chunks(markdown, metadata: { ... })
# => [{ cursor: 0, text: "...", metadata: { ... } }, ...]

Split with the following priority.

[
    "\n# ", # h1
    "\n## ", # h2
    "\n### ", # h3
    "\n#### ", # h4
    "\n##### ", # h5
    "\n###### ", # h6
    "```\n\n", # code block
    "\n\n***\n\n", # horizontal rule
    "\n\n---\n\n", # horizontal rule
    "\n\n___\n\n", # horizontal rule
    "\n\n", # new line
    "\n", # new line
    " ", # space
    "" # empty
]

Development

After checking out the repo, run bin/setup to install dependencies. Then, run bundle exec rake to run the tests. You can also run bin/console for an interactive prompt that will allow you to experiment.

Contributing

Bug reports and pull requests are welcome on GitHub at https://github.com/kawakamimoeki/baran. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the code of conduct.

License

The gem is available as open source under the terms of the MIT License.

Code of Conduct

Everyone interacting in the Baran project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the code of conduct.