add config for temperature and model to env

main
pluja 2023-03-22 13:47:29 +01:00
rodzic f4d3e3de2e
commit 3248221e9a
2 zmienionych plików z 10 dodań i 5 usunięć

Wyświetl plik

@ -3,8 +3,10 @@ services:
chatbot:
build: .
environment:
- OPENAI_API_KEY=YourApiKey
- OPENAI_API_KEY=XXX
- OPENAI_MODEL=gpt-3.5-turbo
- CHATGPT_SYSTEM_PROMPT=You are a helpful assistant.
- CHATGPT_TEMPERATURE=1.0
- CHATGPT_MAX_USER_CONTEXT=5
- BOT_TOKEN=XXXXXX:AAAAAAAAAAAA-AAA-AAAAAA
- BOT_ALLOWED_USERS=5xxxxxxxx2,6xxxxxxxx1
- BOT_TOKEN=XXX
- BOT_ALLOWED_USERS=USER_ID_1,USER_ID_2

Wyświetl plik

@ -36,6 +36,8 @@ users = {
}
ALLOWED_USERS=[]
SYSTEM_PROMPT=os.environ.get("CHATGPT_SYSTEM_PROMPT")
TEMPERATURE=os.environ.get("CHATGPT_TEMPERATURE")
MODEL=os.environ.get("OPENAI_MODEL")
MAX_USER_CONTEXT=int(os.environ.get("CHATGPT_MAX_USER_CONTEXT"))
def restricted(func):
@ -165,9 +167,10 @@ async def chat(update: Update, context: ContextTypes.DEFAULT_TYPE):
# Interact with ChatGPT API and stream the response
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
model=MODEL,
messages=[{"role": "system", "content": SYSTEM_PROMPT}] + user_context,
stream=True
stream=True,
temperature=float(TEMPERATURE)
)
# Initialize variables for streaming