DEBO-V1 / modules /gpt_modules.py
Ridealist's picture
refactor: ๋ถˆํ•„์š”ํ•œ ์ฝ”๋“œ ์ œ๊ฑฐ, DB ๊ด€๋ จ ์ฝ”๋“œ ์ฃผ์„ ์ฒ˜๋ฆฌ
3708b8e
import openai
import streamlit as st
# from dotenv import dotenv_values
# config = dotenv_values(".env")
# if config:
# openai.organization = config.get('OPENAI_ORGANIZATION')
# openai.api_key = config.get('OPENAI_API_KEY')
# else:
# openai.organization = st.secrets['OPENAI_ORGANIZATION']
# openai.api_key = st.secrets['OPENAI_API_KEY']
def gpt_call(api_key, prompt, role="user"):
openai.api_key = api_key
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": role, "content": prompt},
]
)
output_text = response["choices"][0]["message"]["content"]
return output_text
def gpt_call_context(api_key, messages):
openai.api_key = api_key
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=messages
)
output_text = response["choices"][0]["message"]["content"]
return output_text