diff --git a/.gitattributes b/.gitattributes index 9f080463..e92c4834 100644 --- a/.gitattributes +++ b/.gitattributes @@ -1 +1,2 @@ *.min.lua linguist-vendored +*documentation.html diff --git a/.gitignore b/.gitignore index b3adf65a..033958a9 100644 --- a/.gitignore +++ b/.gitignore @@ -17,7 +17,7 @@ userscripts/* !userscripts/examples !userscripts/kaipreset_*.lua !userscripts/Readme.* -!userscripts/api_documentation.md +!userscripts/api_documentation.* softprompts/* # Ignore PyCharm project files. diff --git a/userscripts/api_documentation.html b/userscripts/api_documentation.html new file mode 100644 index 00000000..f504f821 --- /dev/null +++ b/userscripts/api_documentation.html @@ -0,0 +1,720 @@ + + + + + + + API + + + + +
+
+ + + +
+
+
+
+

kobold

+
global kobold: KoboldLib
+
+

Methods:

+ +

Fields:

+ +

kobold.decode()

+
function KoboldLib.decode(tok: integer|table<integer, integer>) -> string
+
+

Callable from: anywhere

+

Decodes the given token or list of tokens using the current tokenizer. If kobold.backend is 'readonly' or 'api', the tokenizer used is the GPT-2 tokenizer, otherwise the model’s tokenizer is used. This function is the inverse of kobold.encode().

+
print(kobold.decode({15496, 2159}))  -- 'Hello World'
+
+

Parameters:

+ +

Returns:

+ +

kobold.encode()

+
function KoboldLib.encode(str: string) -> table<integer, integer>
+
+

Callable from: anywhere

+

Encodes the given string using the current tokenizer into an array of tokens. If kobold.backend is 'readonly' or 'api', the tokenizer used is the GPT-2 tokenizer, otherwise the model’s tokenizer is used. This function is the inverse of kobold.decode().

+
local tokens = kobold.encode("Hello World")
+print(#tokens)  -- 2
+print(tokens[1])  -- 15496
+print(tokens[2])  -- 2159
+
+

Parameters:

+ +

Returns:

+ +

kobold.get_config_file()

+
function KoboldLib.get_config_file(clear?: boolean) -> file*
+
+

Callable from: anywhere

+

Returns a file handle representing your script’s configuration file, which is usually the file in the userscripts folder with the same filename as your script but with “.conf” appended at the end. This function throws an error on failure.

+

If the configuration file does not exist when this function is called, the configuration file will first be created as a new empty file.

+

If KoboldAI does not possess an open file handle to the configuration file, this function opens the file in w+b mode if the clear parameter is a truthy value, otherwise the file is opened in r+b mode. These are mostly the same – the file is opened in binary read-write mode and then seeked to the start of the file – except the former mode deletes the contents of the file prior to opening it and the latter mode does not.

+

If KoboldAI does possess an open file handle to the configuration file, that open file handle is returned without seeking or deleting the contents of the file. You can check if KoboldAI possesses an open file handle to the configuration file by using kobold.is_config_file_open.

+
local example_config = "return 'Hello World'"
+local cfg
+do
+    -- If config file is empty, write example config
+    local f <close> = kobold.get_config_file()
+    f:seek("set")
+    if f:read(1) == nil then f:write(example_config) end
+    f:seek("set")
+
+    -- Read config
+    local err
+    cfg, err = load(f:read("a"))
+    if err ~= nil then error(err) end
+    cfg = cfg()
+end
+
+

Parameters:

+ +

Returns:

+ +

kobold.halt_generation()

+
function KoboldLib.halt_generation() -> nil
+
+

Callable from: anywhere

+

If called from an input modifier, prevents the user’s input from being sent to the model and skips directly to the output modifier.

+

If called from a generation modifier, stops generation after the current token is generated and then skips to the output modifier. In other words, if, when you call this function, kobold.generated has n columns, it will have exactly n+1 columns when the output modifier is called.

+

If called from an output modifier, has no effect.

+

kobold.restart_generation()

+
function KoboldLib.restart_generation(sequence?: integer) -> nil
+
+

Callable from: output modifier

+

After the current output is sent to the GUI, starts another generation using the empty string as the submission.

+

Whatever ends up being the output selected by the user or by the sequence parameter will be saved in kobold.feedback when the new generation begins.

+

Parameters:

+ +

kobold.authorsnote

+
field KoboldLib.authorsnote: string
+
+

Readable from: anywhere
+Writable from: anywhere (triggers regeneration when written to from generation modifier)

+

The author’s note as set from the “Memory” button in the GUI.

+

Modifying this field from inside of a generation modifier triggers a regeneration, which means that the context is recomputed after modification and generation begins again with the new context and previously generated tokens. This incurs a small performance penalty and should not be performed in excess.

+

kobold.authorsnotetemplate

+
field KoboldLib.authorsnotetemplate: string
+
+

Readable from: anywhere
+Writable from: anywhere (triggers regeneration when written to from generation modifier)

+

The author’s note template as set from the “Memory” button in the GUI.

+

Modifying this field from inside of a generation modifier triggers a regeneration, which means that the context is recomputed after modification and generation begins again with the new context and previously generated tokens. This incurs a small performance penalty and should not be performed in excess.

+

kobold.custmodpth

+
field KoboldLib.custmodpth: string
+
+

Readable from: anywhere
+Writable from: nowhere

+

Path to a directory that the user chose either via the file dialog that appears when KoboldAI asks you to choose the path to your custom model or via the --path command-line flag.

+

If the user loaded a built-in model from the menu, this is instead the model ID of the model on Hugging Face’s model hub, such as “KoboldAI/GPT-Neo-2.7B-Picard” or “hakurei/lit-6B”.

+

kobold.feedback

+
field KoboldLib.feedback: string?
+
+

Readable from: anywhere
+Writable from: nowhere

+

If this is a repeat generation caused by kobold.restart_generation(), this will be a string containing the previous output. If not, this will be nil.

+

kobold.generated

+
field KoboldLib.generated: table<integer, table<integer, integer>>?
+
+

Readable from: generation modifier and output modifier, but only if kobold.modelbackend is not 'api' or 'readonly'
+Writable from: generation modifier

+

Two-dimensional array of tokens generated thus far. Each row represents one sequence, each column one token.

+

kobold.generated_cols

+
field KoboldLib.generated_cols: integer
+
+

Readable from: anywhere
+Writable from: nowhere

+

Number of columns in kobold.generated. In other words, the number of tokens generated thus far, which is equal to the number of times that the generation modifier has been called, not including the current time if this is being read from a generation modifier.

+

If kobold.modelbackend is 'api' or 'readonly', this returns 0 instead.

+

kobold.generated_rows

+
field KoboldLib.generated_rows: integer
+
+

Readable from: anywhere
+Writable from: nowhere

+

Number of rows in kobold.generated, equal to kobold.settings.numseqs.

+

kobold.is_config_file_open

+
field KoboldLib.is_config_file_open: boolean
+
+

Readable from: anywhere
+Writable from: nowhere

+

Whether or not KoboldAI possesses an open file handle to your script’s configuration file. See kobold.get_config_file() for more details.

+

kobold.logits

+
field KoboldLib.logits: table<integer, table<integer, number>>?
+
+

Readable from: generation modifier, but only if kobold.modelbackend is not 'api' or 'readonly'
+Writable from: generation modifier

+

Two-dimensional array of logits prior to being filtered by top-p sampling, etc. Each row represents one sequence, each column one of the tokens in the model’s vocabulary. The ith column represents the logit score of token i-1, so if you want to access the logit score of token 18435 (" Hello" with a leading space), you need to access column 18436. You may alter this two-dimensional array to encourage or deter certain tokens from appearing in the output in a stochastic manner.

+

Don’t modify this table unnecessarily unless you know what you are doing! The bias example scripts show how to use this feature properly.

+

kobold.logits_cols

+
field KoboldLib.logits_cols: integer
+
+

Readable from: anywhere
+Writable from: nowhere

+

Number of columns in kobold.logits, equal to the vocabulary size of the current model. Most models based on GPT-2 (e.g. GPT-Neo and GPT-J) have a vocabulary size of 50257. GPT-J models in particular have a vocabulary size of 50400 instead, although GPT-J models aren’t trained to use the rightmost 143 tokens of the logits array.

+

If kobold.modelbackend is 'api' or 'readonly', this returns 0 instead.

+

kobold.logits_rows

+
field KoboldLib.logits_rows: integer
+
+

Readable from: anywhere
+Writable from: nowhere

+

Number of rows in kobold.generated, equal to kobold.settings.numseqs. a local KoboldAI install.

+

kobold.memory

+
field KoboldLib.memory: string
+
+

Readable from: anywhere
+Writable from: anywhere (triggers regeneration when written to from generation modifier)

+

The memory as set from the “Memory” button in the GUI.

+

Modifying this field from inside of a generation modifier triggers a regeneration, which means that the context is recomputed after modification and generation begins again with the new context and previously generated tokens. This incurs a small performance penalty and should not be performed in excess.

+

kobold.modelbackend

+
field KoboldLib.modelbackend: string
+
+

Readable from: anywhere
+Writable from: nowhere

+

One of the following values:

+ +

kobold.modeltype

+
field KoboldLib.modeltype: string
+
+

Readable from: anywhere
+Writable from: nowhere

+

One of the following values:

+ +

kobold.num_outputs

+
field KoboldLib.num_outputs: integer
+
+

Readable from: anywhere
+Writable from: nowhere

+

Number of rows in kobold.outputs. This is equal to kobold.settings.numseqs unless you’re using a non-Colab third-party API such as OpenAI or InferKit, in which case this is 1. If you decide to write to kobold.settings.numseqs from an output modifier, this value remains unchanged.

+

kobold.outputs

+
field KoboldLib.outputs: table<integer, string>
+
+

Readable from: output modifier
+Writable from: output modifier

+

Model output before applying output formatting. One row per “Gens Per Action”, unless you’re using OpenAI or InferKit, in which case this always has exactly one row.

+

kobold.settings

+
field KoboldLib.settings: KoboldSettings
+
+

Readable from: anywhere
+Writable from: anywhere (does not affect other scripts when written to since each script has its own copy of this object)

+

Contains most of the settings. They have the same names as in gensettings.py, so the top-p value is kobold.settings.settopp.

+

All the settings can be read from anywhere and written from anywhere, except kobold.settings.numseqs which can only be written to from an input modifier or output modifier.

+

Modifying certain fields from inside of a generation modifier triggers a regeneration, which means that the context is recomputed after modification and generation begins again with the new context and previously generated tokens. This incurs a small performance penalty and should not be performed in excess. Currently, only the following fields and their aliases cause this to occur:

+ +

kobold.story

+

Readable from: anywhere
+Writable from: nowhere

+
field KoboldLib.story: KoboldStory
+
+

Contains the chunks of the current story. Don’t use pairs or ipairs to iterate over the story chunks, use kobold.story:forward_iter() or kobold.story:reverse_iter(), which guarantee amortized worst-case iteration time complexity linear to the number of chunks in the story regardless of what the highest chunk number is.

+

You can index this object to get a story chunk (as a KoboldStoryChunk object) by its number, which is an integer. The prompt chunk, if it exists, is guaranteed to be chunk 0. Aside from that, the chunk numbers are not guaranteed to be contiguous or ordered in any way.

+
local prompt_chunk = kobold.story[0]  -- KoboldStoryChunk object referring to the prompt chunk
+
+

Methods:

+ +

kobold.story:forward_iter()

+

Callable from: anywhere

+
function KoboldStory:forward_iter() -> fun(): KoboldStoryChunk, table, nil
+
+

Returns a stateful iterator that efficiently iterates through story chunks from top to bottom.

+
for chunk in kobold.story:forward_iter() do
+    print(chunk.num, chunk.content)
+end
+
+

kobold.story:reverse_iter()

+

Callable from: anywhere

+
function KoboldStory:reverse_iter() -> fun(): KoboldStoryChunk, table, nil
+
+

Returns a stateful iterator that efficiently iterates through story chunks from bottom to top.

+
for chunk in kobold.story:reverse_iter() do
+    print(chunk.num, chunk.content)
+end
+
+

KoboldStoryChunk

+

Represents a story chunk.

+

Fields:

+ +

KoboldStoryChunk.content

+

Readable from: anywhere
+Writable from: anywhere (triggers regeneration when written to from generation modifier)

+
field KoboldStoryChunk.content: string
+
+

The text inside of the story chunk.

+

Modifying this field from inside of a generation modifier triggers a regeneration, which means that the context is recomputed after modification and generation begins again with the new context and previously generated tokens. This incurs a small performance penalty and should not be performed in excess.

+

KoboldStoryChunk.num

+

Readable from: anywhere
+Writable from: nowhere

+
field KoboldStoryChunk.num: integer
+
+

The number of the story chunk. Chunk 0 is guaranteed to be the prompt chunk if it exists; no guarantees can be made about the numbers of other chunks.

+

kobold.submission

+

Readable from: anywhere
+Writable from: input modifier

+
field kobold.submission: string
+
+

The user-submitted text after being formatted by input formatting. If this is a repeated generation incurred by kobold.restart_generation(), then this is the empty string.

+

kobold.worldinfo

+
field KoboldLib.worldinfo: KoboldWorldInfo
+
+

Represents the world info entries.

+

Indexing this object at index i returns the ith world info entry from the top in amortized constant worst-case time as a KoboldWorldInfoEntry. This includes world info entries that are inside folders.

+
local entry = kobold.worldinfo[5]  -- Retrieves fifth entry from top as a KoboldWorldInfoEntry
+
+

You can use ipairs or a numeric loop to iterate from top to bottom:

+
for index, entry in ipairs(kobold.worldinfo):
+    print(index, entry.content)
+end
+
+
for index = 1, #kobold.worldinfo do:
+    print(index, kobold.worldinfo[index].content)
+end
+
+

Methods:

+ +

Fields:

+ +

kobold.worldinfo:compute_context()

+

Callable from: anywhere

+
function KoboldWorldInfo:compute_context(submission: string, entries?: KoboldWorldInfoEntry|table<any, KoboldWorldInfoEntry>, kwargs?: table<string, any>) -> string
+
+

Computes the context that would be sent to the generator with the user’s current settings if submission were the user’s input after being formatted by input formatting. The context would include memory at the top, followed by active world info entries, followed by some story chunks with the author’s note somewhere, followed by submission.

+

Parameters

+ +

Returns

+

string: Computed context.

+

kobold.worldinfo:finduid()

+

Callable from: anywhere

+
function KoboldWorldInfo:finduid(u: integer) -> KoboldWorldInfoEntry?
+
+

Returns the world info entry with the given UID in amortized constant worst-case time, or nil if not found.

+

Parameters

+ +

Returns

+ +

kobold.worldinfo:is_valid()

+

Callable from: anywhere

+
function KoboldWorldInfo:is_valid() -> boolean
+
+

This always returns true.

+

kobold.worldinfo.folders

+

Readable from: anywhere
+Writable from: nowhere

+
field KoboldWorldInfo.folders: KoboldWorldInfoFolderSelector
+
+

Can be indexed in amortized constant worst-case time and iterated over and has a finduid method just like kobold.worldinfo, but gets folders (as KoboldWorldInfoFolder objects) instead.

+
local folder = kobold.worldinfo.folders[5]  -- Retrieves fifth folder from top as a KoboldWorldInfoFolder
+
+
for index, folder in ipairs(kobold.worldinfo.folders):
+    print(index, folder.name)
+end
+
+
for index = 1, #kobold.worldinfo.folders do:
+    print(index, kobold.worldinfo.folders[index].name)
+end
+
+

Methods

+ +

KoboldWorldInfoEntry

+

Represents a world info entry.

+

Methods:

+ +

Fields:

+ +

KoboldWorldInfoEntry:compute_context()

+

Callable from: anywhere

+
function KoboldWorldInfoEntry:compute_context(submission: string, kwargs?: table<string, any>) -> string
+
+

The same as calling kobold.worldinfo:compute_context() with this world info entry as the argument.

+

Parameters

+ +

Returns

+

string: Computed context.

+

KoboldWorldInfoEntry:is_valid()

+

Callable from: anywhere

+
function KoboldWorldInfoEntry:is_valid() -> boolean
+
+

Returns true if this world info entry still exists (i.e. wasn’t deleted), otherwise returns false.

+

KoboldWorldInfoEntry.comment

+

Readable from: anywhere
+Writable from: anywhere

+
field KoboldWorldInfoEntry.comment: string
+
+

The world info entry’s comment that appears in its topmost text box.

+

KoboldWorldInfoEntry.constant

+

Readable from: anywhere
+Writable from: anywhere (triggers regeneration when written to from generation modifier)

+
field KoboldWorldInfoEntry.constant: boolean
+
+

Whether or not this world info entry is constant. Constant world info entries are always included in the context regardless of whether or not its keys match the story chunks in the context.

+

Modifying this field from inside of a generation modifier triggers a regeneration, which means that the context is recomputed after modification and generation begins again with the new context and previously generated tokens. This incurs a small performance penalty and should not be performed in excess.

+

KoboldWorldInfoEntry.content

+

Readable from: anywhere
+Writable from: anywhere (triggers regeneration when written to from generation modifier)

+
field KoboldWorldInfoEntry.content: string
+
+

The text in the “What To Remember” text box that gets included in the context when the world info entry is active.

+

Modifying this field from inside of a generation modifier triggers a regeneration, which means that the context is recomputed after modification and generation begins again with the new context and previously generated tokens. This incurs a small performance penalty and should not be performed in excess.

+

KoboldWorldInfoEntry.folder

+

Readable from: anywhere
+Writable from: nowhere

+
field KoboldWorldInfoEntryfolder: integer?
+
+

UID of the folder the world info entry is in, or nil if it’s outside of a folder.

+

KoboldWorldInfoEntry.key

+

Readable from: anywhere
+Writable from: anywhere (triggers regeneration when written to from generation modifier)

+
field KoboldWorldInfoEntry.key: string
+
+

For non-selective world info entries, this is the world info entry’s comma-separated list of keys. For selective world info entries, this is the comma-separated list of primary keys.

+

Modifying this field from inside of a generation modifier triggers a regeneration, which means that the context is recomputed after modification and generation begins again with the new context and previously generated tokens. This incurs a small performance penalty and should not be performed in excess.

+

KoboldWorldInfoEntry.keysecondary

+

Readable from: anywhere
+Writable from: anywhere (triggers regeneration when written to from generation modifier)

+
field KoboldWorldInfoEntry.keysecondary: string
+
+

For non-selective world info entries, the value of this field is undefined and writing to it has no effect. For selective world info entries, this is the comma-separated list of secondary keys.

+

Modifying this field from inside of a generation modifier triggers a regeneration, which means that the context is recomputed after modification and generation begins again with the new context and previously generated tokens. This incurs a small performance penalty and should not be performed in excess.

+

KoboldWorldInfoEntry.num

+

Readable from: anywhere
+Writable from: nowhere

+
field KoboldWorldInfoEntry.num: integer
+
+

This is 0 if the entry is the first one from the top, 1 if second from the top, and so on.

+

KoboldWorldInfoEntry.selective

+

Readable from: anywhere
+Writable from: anywhere (triggers regeneration when written to from generation modifier)

+
field KoboldWorldInfoEntry.selective: boolean
+
+

Whether or not the world info entry is selective. Selective entries have both primary and secondary keys.

+

Modifying this field from inside of a generation modifier triggers a regeneration, which means that the context is recomputed after modification and generation begins again with the new context and previously generated tokens. This incurs a small performance penalty and should not be performed in excess.

+

KoboldWorldInfoEntry.uid

+

Readable from: anywhere
+Writable from: nowhere

+
field KoboldWorldInfoEntry.uid: integer
+
+

UID of the world info entry.

+

KoboldWorldInfoFolderSelector:finduid()

+

Callable from: anywhere

+
function KoboldWorldInfoFolderSelector:finduid(u: integer) -> KoboldWorldInfoFolder?
+
+

Returns the world info folder with the given UID in amortized constant worst-case time, or nil if not found.

+

Parameters

+ +

Returns

+ +

KoboldWorldInfoFolderSelector:is_valid()

+

Callable from: anywhere

+
function KoboldWorldInfoFolderSelector:is_valid() -> boolean
+
+

This always returns true.

+

KoboldWorldInfoFolder

+

Represents a world info folder.

+

Methods:

+ +

Fields:

+ +

KoboldWorldInfoFolder:compute_context()

+

Callable from: anywhere

+
function KoboldWorldInfoFolder:compute_context(submission: string, entries?: KoboldWorldInfoEntry|table<any, KoboldWorldInfoEntry>, kwargs?: table<string, any>) -> string
+
+

Computes the context that would be sent to the generator with the user’s current settings if submission were the user’s input after being formatted by input formatting. The context would include memory at the top, followed by active world info entries, followed by some story chunks with the author’s note somewhere, followed by submission.

+

Unlike kobold.worldinfo:compute_context(), this function doesn’t include world info keys outside of the folder.

+

Parameters

+ +

Returns

+

string: Computed context.

+

KoboldWorldInfoFolder:finduid()

+

Callable from: anywhere

+
function KoboldWorldInfoFolder:finduid(u: integer) -> KoboldWorldInfoEntry?
+
+

Returns the world info entry inside of the folder with the given UID in amortized constant worst-case time, or nil if not found.

+

Parameters

+ +

Returns

+ +

KoboldWorldInfoFolder:is_valid()

+

Callable from: anywhere

+
function KoboldWorldInfoFolder:is_valid() -> boolean
+
+

Returns whether or not the folder still exists (i.e. wasn’t deleted).

+

KoboldWorldInfoFolder.name

+

Readable from: anywhere
+Writable from: anywhere

+
field KoboldWorldInfoFolder.name: string
+
+

Name of the world info folder as defined in its one text box.

+

KoboldWorldInfoFolder.uid

+

Readable from: anywhere
+Writable from: nowhere

+
field KoboldWorldInfoFolder.uid: integer
+
+

UID of the world info folder.

+ +
+
+ + +