AskDavis/nate-discord
1
0
Fork 0
mirror of https://github.com/AskDavis/nate-discord.git synced 2026-01-04 20:59:48 -08:00
Code Issues Packages Projects Releases Wiki Activity

debug in docker

Browse Source
This commit is contained in:
William Lane
2023-04-03 16:22:48 -07:00
parent b5b7048784
commit 7461f994e7
5 changed files with 18 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
app/mabel/connect_openai.py
View File

@@ -2,8 +2,12 @@ from dotenv import load_dotenv
import openai
import os
import json
load_dotenv()
openai.api_key=os.getenv('OPENAI_KEY')
import logging
#load_dotenv()
openai_key = openai.api_key=os.getenv('OPENAI_KEY')
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
SYSTEM_DIRECTIVES = []
DIALOGUE_STACK = []
@@ -22,14 +26,17 @@ async def chatgpt_response(prompt):
DIALOGUE_STACK = DIALOGUE_STACK[-6:]
messages = SYSTEM_DIRECTIVES + DIALOGUE_STACK
if "mabel" in prompt.lower() or "Mabel" in prompt.lower():
if "mabel" in prompt.lower():
logger.info("PROMPT: " + prompt.lower())
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
model="gpt-4",
messages=messages,
max_tokens=1800,
)
DIALOGUE_STACK.append({"role": "assistant", "content": response['choices'][0]['message']['content']})
return response ['choices'][0]['message']['content']
response_content = response['choices'][0]['message']['content']
logger.info("RESPONSE: " + response_content)
DIALOGUE_STACK.append({"role": "assistant", "content": response_content })
return response_content
else:
DIALOGUE_STACK.append({"role": "user", "content": prompt})
return