INNER CODE UNIT · Python
__init__
shakedzy/companion · python/chatbot.py:38
def __init__(self, config: Config, memory: Memory):
self.client = config.openai.client
self._memory = memory
self._model = config.model.name
self._temperature = config.model.temperature
self._language = config.language.learning
lang = iso6391_to_language_name(config.language.learning)
user_lang = iso6391_to_language_name(config.language.native)
self._memory.add("system", dedent(SYSTEM_PROMPT.format(
teacher_name=config.bot.name, user_name=config.user.name, language=lang, user_language=user_lang,
level=config.language.level, user_gender=config.user.gender, bot_gender=config.bot.gender
)))
def get_response(self, is_initial_message=False) -> Generator:
"""
send previous messages (stored in `self._memory`) to GPT and receive a response.
The response is streamed, therefore a Generator is returned