diff --git a/src/quickbot_agent/config.py b/src/quickbot_agent/config.py index 8e1be21..7d7eddc 100644 --- a/src/quickbot_agent/config.py +++ b/src/quickbot_agent/config.py @@ -8,7 +8,7 @@ class Config(BaseSettings): OPENAI_API_KEY: str OPENAI_MODEL: str = "gpt-4o-2024-11-20" - MESSAGE_HISTORY_DEPTH: int = 10 + MESSAGE_HISTORY_DEPTH: int = 20 config = Config() diff --git a/src/quickbot_agent/handlers/messages.py b/src/quickbot_agent/handlers/messages.py index 98dc00d..7473998 100644 --- a/src/quickbot_agent/handlers/messages.py +++ b/src/quickbot_agent/handlers/messages.py @@ -15,10 +15,13 @@ from ..utils import get_message_log, add_message_log, generate_gpt_tools_schemas from typing import TYPE_CHECKING import ujson as json from ..config import config +from logging import getLogger if TYPE_CHECKING: from ..main import AgentPlugin +logger = getLogger(__name__) + router = Router() @@ -69,6 +72,9 @@ async def handle_openai_cycle( else: gpt_tools_metadata = plugin._gpt_tools_metadata + logger.debug(f"Messages: {messages}") + logger.debug(f"GPT tools metadata: {gpt_tools_metadata}") + response = await client.chat.completions.create( model=config.OPENAI_MODEL, messages=messages, @@ -318,11 +324,9 @@ async def get_messages( ) messages = [ - { - "role": "system", - "content": system_prompt, - } + {"role": "system", "content": system_prompt}, ] + for message_log_item in message_log: messages.append(json.loads(message_log_item.content))