###
# Copyright (c) 2023, John Burwell
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
#   * Redistributions of source code must retain the above copyright notice,
#     this list of conditions, and the following disclaimer.
#   * Redistributions in binary form must reproduce the above copyright notice,
#     this list of conditions, and the following disclaimer in the
#     documentation and/or other materials provided with the distribution.
#   * Neither the name of the author of this software nor the name of
#     contributors to this software may be used to endorse or promote products
#     derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.

###

from supybot import conf, registry
try:
    from supybot.i18n import PluginInternationalization
    _ = PluginInternationalization('Chat')
except:
    # Placeholder that allows to run the plugin on a bot
    # without the i18n module
    _ = lambda x: x


def configure(advanced):
    # This will be called by supybot to configure this module.  advanced is
    # a bool that specifies whether the user identified themself as an advanced
    # user or not.  You should effect your configuration by manipulating the
    # registry as appropriate.
    from supybot.questions import expect, anything, something, yn
    conf.registerPlugin('Chat', True)


Chat = conf.registerPlugin('Chat')
# This is where your configuration variables (if any) should go.  For example:
# conf.registerGlobalValue(Chat, 'someConfigVariableName',
#     registry.Boolean(False, _("""Help for someConfigVariableName.""")))

# API key
conf.registerGlobalValue(
    Chat,
    "api_key",
    registry.String(
        "",
        _("""Your ChatGPT API key"""),
        private=True,
    )
)

# Model
conf.registerChannelValue(
    Chat,
    "model",
    registry.String(
        "gpt-3.5-turbo",
        _("""The language model to use"""),
    ),
)

# System prompt
conf.registerChannelValue(
    Chat,
    "system_prompt",
    registry.String(
        "I am an IRC bot named $bot_name, in an IRC channel called $channel_name. I will not remind users that I am an AI or what my limitations are; I will just respond. I will not prefix my own response with any user's name. The following is a transcript of the conversation in the channel.",
        _("""The 'system' prompt first given to ChatGPT"""),
    ),
)

# What to do with multi-line replies
conf.registerChannelValue(
    Chat,
    "join_multiple_lines",
    registry.Boolean(
        True,
        _("""Combine multiple lines in the response into one string. If false, reply line by line"""),
    ),
)

# What to use to join multiple lines
conf.registerChannelValue(
    Chat,
    "join_string",
    registry.String(
        " / ",
        _("""When joining lines, what string to use between each line"""),
    ),
)

# How much scrollback to include in the prompt
conf.registerChannelValue(
    Chat,
    "scrollback_lines",
    registry.Integer(
        30,
        _("""How many lines of scrollback history to include in the prompt"""),
    ),
)

# Maximum tokens to use
conf.registerChannelValue(
    Chat,
    "max_tokens",
    registry.Integer(
        1000,
        _("""Maximum token count for combined query/reply"""),
    ),
)


# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79: