minor improvements

This commit is contained in:
ned
2023-03-25 14:53:00 +01:00
parent 5d2c288789
commit 1a28b6b683
5 changed files with 73 additions and 77 deletions

View File

@@ -36,16 +36,8 @@ def main():
'n_choices': int(os.environ.get('N_CHOICES', 1)),
'temperature': float(os.environ.get('TEMPERATURE', 1.0)),
'image_size': os.environ.get('IMAGE_SIZE', '512x512'),
# 'gpt-3.5-turbo' or 'gpt-3.5-turbo-0301'
'model': os.environ.get('OPENAI_MODEL', 'gpt-3.5-turbo'),
# Number between -2.0 and 2.0. Positive values penalize new tokens based on whether
# they appear in the text so far, increasing the model's likelihood to talk about new topics.
'presence_penalty': int(os.environ.get('PRESENCE_PENALTY', 0)),
# Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing
# frequency in the text so far, decreasing the model's likelihood to repeat the same line verbatim.
'frequency_penalty': int(os.environ.get('FREQUENCY_PENALTY', 0)),
}
@@ -62,6 +54,7 @@ def main():
'image_prices': [float(i) for i in os.environ.get('IMAGE_PRICES',"0.016,0.018,0.02").split(",")],
'transcription_price': float(os.environ.get('TOKEN_PRICE', 0.002)),
}
# Setup and run ChatGPT and Telegram bot
openai_helper = OpenAIHelper(config=openai_config)
telegram_bot = ChatGPT3TelegramBot(config=telegram_config, openai=openai_helper)