pw.xpacks.llm.splitters

Splits a given string or a list of strings into chunks based on token count.

This splitter tokenizes the input texts and splits them into smaller parts (“chunks”) ensuring that each chunk has a token count between min_tokens and max_tokens. It also attempts to break chunks at sensible points such as punctuation marks.

All arguments set default which may be overridden in the UDF call

• Parameters
  • min_tokens (int) – minimum tokens in a chunk of text.
  • max_tokens (int) – maximum size of a chunk in tokens.
  • encoding_name (str) – name of the encoding from tiktoken.

Example:

from pathway.xpacks.llm.splitters import TokenCountSplitter
import pathway as pw
t  = pw.debug.table_from_markdown(
    '''| text
1| cooltext'''
)
splitter = TokenCountSplitter(min_tokens=1, max_tokens=1)
t += t.select(chunks = splitter(pw.this.text))
pw.debug.compute_and_print(t, include_id=False)

text     | chunks
cooltext | (('cool', pw.Json({})), ('text', pw.Json({})))

__call__(text, **kwargs)

sourceSplit given strings into smaller chunks.

• Parameters
  • messages (-) – Column with texts to be split
  • **kwargs (-) – override for defaults set in the constructor