mirror of
https://github.com/NovaOSS/LingoSynth.git
synced 2024-11-25 21:33:58 +01:00
26 lines
564 B
Python
26 lines
564 B
Python
import os
|
|
import openai
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
if os.getenv('OPENAI_API_BASE'):
|
|
openai.api_base = os.getenv('OPENAI_API_BASE')
|
|
|
|
openai.api_key = os.getenv('OPENAI_API_KEY')
|
|
|
|
MODEL = os.getenv('OPENAI_CHAT_MODEL') or 'gpt-3.5-turbo'
|
|
|
|
def generate(messages: str):
|
|
"""Generates a new message based on the given messages."""
|
|
|
|
return openai.ChatCompletion.create(
|
|
model=MODEL,
|
|
messages=messages,
|
|
temperature=0.9,
|
|
frequency_penalty=0.2,
|
|
presence_penalty=0.6,
|
|
|
|
).choices[0].message.content
|