Resetting file to before chess attempting

This commit is contained in:
AllfatherHatt 2025-02-22 02:02:26 +01:00
parent 72171e9ed0
commit 4337e7be10

View File

@ -8,9 +8,6 @@ import traceback
from collections import Counter from collections import Counter
from redbot.core import Config, commands from redbot.core import Config, commands
from openai import OpenAIError from openai import OpenAIError
from .chess_addon import ChessHandler
chess_handler = ChessHandler()
class ReginaldCog(commands.Cog): class ReginaldCog(commands.Cog):
def __init__(self, bot): def __init__(self, bot):
@ -102,7 +99,7 @@ class ReginaldCog(commands.Cog):
formatted_messages += [{"role": "user", "content": f"{entry['user']}: {entry['content']}"} for entry in memory] formatted_messages += [{"role": "user", "content": f"{entry['user']}: {entry['content']}"} for entry in memory]
formatted_messages.append({"role": "user", "content": f"{user_name}: {prompt}"}) formatted_messages.append({"role": "user", "content": f"{user_name}: {prompt}"})
response_text = await self.generate_response(api_key, formatted_messages, ctx) response_text = await self.generate_response(api_key, formatted_messages)
# ✅ Extract potential long-term facts from Reginald's response # ✅ Extract potential long-term facts from Reginald's response
potential_fact = self.extract_fact_from_response(response_text) potential_fact = self.extract_fact_from_response(response_text)
@ -274,92 +271,24 @@ class ReginaldCog(commands.Cog):
return None # No strong fact found return None # No strong fact found
async def generate_response(self, api_key, messages, ctx): async def generate_response(self, api_key, messages):
"""Handles AI responses and function calling for chess interactions."""
model = await self.config.openai_model() model = await self.config.openai_model()
try: try:
client = openai.AsyncClient(api_key=api_key) client = openai.AsyncClient(api_key=api_key)
response = await client.chat.completions.create( response = await client.chat.completions.create(
model=model, model=model,
messages=messages, messages=messages,
max_tokens=1500, # Balanced token limit to allow function execution & flavor text max_tokens=4112,
temperature=0.7, temperature=0.7,
presence_penalty=0.5, presence_penalty=0.5,
frequency_penalty=0.5, frequency_penalty=0.5
functions=[
{
"name": "set_board",
"description": "Sets up the chessboard to a given FEN string.",
"parameters": {
"type": "object",
"properties": {
"fen": {"type": "string", "description": "The FEN string representing the board state."}
},
"required": ["fen"]
}
},
{
"name": "make_move",
"description": "Executes a chess move for the current game.",
"parameters": {
"type": "object",
"properties": {
"move": {"type": "string", "description": "The move in SAN format (e.g., 'e2e4')."}
},
"required": ["move"]
}
},
{
"name": "reset_board",
"description": "Resets the chessboard to the default starting position.",
"parameters": {}
},
{
"name": "resign",
"description": "Resigns from the current chess game.",
"parameters": {}
},
{
"name": "get_board_state_text",
"description": "Retrieves the current board state as a FEN string.",
"parameters": {
"type": "object",
"properties": {
"user_id": {"type": "string", "description": "The user's unique ID."}
},
"required": ["user_id"]
}
}
]
) )
response_text = response.choices[0].message.content.strip()
if response_text.startswith("Reginald:"):
response_text = response_text[len("Reginald:"):].strip()
return response_text
response_data = response.choices[0].message except OpenAIError as e:
# 🟢 Check if OpenAI returned a function call
if hasattr(response_data, "function_call") and response_data.function_call:
function_call = response_data.function_call
function_name = function_call.name
function_args = json.loads(function_call.arguments) # Convert JSON string to dict
# 🟢 Call the appropriate function
if function_name == "set_board":
return chess_handler.set_board(ctx.author.id, function_args["fen"])
elif function_name == "make_move":
return chess_handler.make_move(ctx.author.id, function_args["move"])
elif function_name == "reset_board":
return chess_handler.reset_board(ctx.author.id)
elif function_name == "resign":
return chess_handler.resign(ctx.author.id)
elif function_name == "get_board_state_text":
return chess_handler.get_fen(ctx.author.id) # Returns FEN string of the board
# 🟢 If no function was called, return AI-generated response with flavor text
return response_data.get("content", "I'm afraid I have nothing to say.")
except openai.OpenAIError as e:
error_message = f"OpenAI Error: {e}" error_message = f"OpenAI Error: {e}"
reginald_responses = [ reginald_responses = [
f"Regrettably, I must inform you that I have encountered a bureaucratic obstruction:\n\n```{error_message}```", f"Regrettably, I must inform you that I have encountered a bureaucratic obstruction:\n\n```{error_message}```",
@ -622,4 +551,4 @@ class ReginaldCog(commands.Cog):
async def setup(bot): async def setup(bot):
"""✅ Correct async cog setup for Redbot""" """✅ Correct async cog setup for Redbot"""
await bot.add_cog(ReginaldCog(bot)) await bot.add_cog(ReginaldCog(bot))