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:
|
chatbot:
|
||||||
build: .
|
build: .
|
||||||
environment:
|
environment:
|
||||||
- OPENAI_API_KEY=YourApiKey
|
- OPENAI_API_KEY=XXX
|
||||||
|
- OPENAI_MODEL=gpt-3.5-turbo
|
||||||
- CHATGPT_SYSTEM_PROMPT=You are a helpful assistant.
|
- CHATGPT_SYSTEM_PROMPT=You are a helpful assistant.
|
||||||
|
- CHATGPT_TEMPERATURE=1.0
|
||||||
- CHATGPT_MAX_USER_CONTEXT=5
|
- CHATGPT_MAX_USER_CONTEXT=5
|
||||||
- BOT_TOKEN=XXXXXX:AAAAAAAAAAAA-AAA-AAAAAA
|
- BOT_TOKEN=XXX
|
||||||
- BOT_ALLOWED_USERS=5xxxxxxxx2,6xxxxxxxx1
|
- BOT_ALLOWED_USERS=USER_ID_1,USER_ID_2
|
7
main.py
7
main.py
|
@ -36,6 +36,8 @@ users = {
|
||||||
}
|
}
|
||||||
ALLOWED_USERS=[]
|
ALLOWED_USERS=[]
|
||||||
SYSTEM_PROMPT=os.environ.get("CHATGPT_SYSTEM_PROMPT")
|
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"))
|
MAX_USER_CONTEXT=int(os.environ.get("CHATGPT_MAX_USER_CONTEXT"))
|
||||||
|
|
||||||
def restricted(func):
|
def restricted(func):
|
||||||
|
@ -165,9 +167,10 @@ async def chat(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
|
||||||
# Interact with ChatGPT API and stream the response
|
# Interact with ChatGPT API and stream the response
|
||||||
response = openai.ChatCompletion.create(
|
response = openai.ChatCompletion.create(
|
||||||
model="gpt-3.5-turbo",
|
model=MODEL,
|
||||||
messages=[{"role": "system", "content": SYSTEM_PROMPT}] + user_context,
|
messages=[{"role": "system", "content": SYSTEM_PROMPT}] + user_context,
|
||||||
stream=True
|
stream=True,
|
||||||
|
temperature=float(TEMPERATURE)
|
||||||
)
|
)
|
||||||
|
|
||||||
# Initialize variables for streaming
|
# Initialize variables for streaming
|
||||||
|
|
Ładowanie…
Reference in New Issue