kopia lustrzana https://codeberg.org/pluja/openai-telegram-bot
add config for temperature and model to env
rodzic
f4d3e3de2e
commit
3248221e9a
|
@ -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
|
7
main.py
7
main.py
|
@ -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
|
||||
|
|
Ładowanie…
Reference in New Issue