Update OpenAI API usage to new version

This commit is contained in:
2024-08-19 10:36:03 -05:00
parent 79b72ee4d0
commit 3c76c33a00
+10 -9
View File
@@ -9,12 +9,14 @@ from telegram.ext import Application, CommandHandler, MessageHandler, filters, C
from dotenv import load_dotenv
from tools.base_tool import BaseTool
from tools.metrics_tool import MetricsTool
import openai
from openai import OpenAI
# Load environment variables
load_dotenv()
openai.api_key = os.environ.get("OPENAI_API_KEY")
client = OpenAI(
api_key=os.environ.get("OPENAI_API_KEY"),
)
# Set up logging to console and file
logging.basicConfig(level=logging.WARNING, handlers=[
@@ -168,17 +170,16 @@ def get_openai_response(messages):
for function in functions
]
response = openai.ChatCompletion.create(
model="gpt-4o", # Using "gpt-4o" as specified
response = client.chat.completions.create(
model="gpt-4", # Changed from "gpt-4o" to "gpt-4"
messages=[{"role": "system", "content": system_prompt}] + messages,
functions=openai_functions,
function_call="auto",
max_tokens=4096 # Using 4096 as specified for "gpt-4o"
)
message = response['choices'][0]['message']
content = message.get('content', '')
function_call = message.get('function_call')
message = response.choices[0].message
content = message.content
function_call = message.function_call
if function_call:
return {
@@ -193,7 +194,7 @@ def get_openai_response(messages):
return None
async def status(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
await update.message.reply_text("Currently using gpt-4o")
await update.message.reply_text("Currently using gpt-4")
async def abort_processing(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
query = update.callback_query