BotForge/quickbot_agent
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add logging

Browse Source
This commit is contained in:
Alexander Kalinovsky
2025-08-14 20:39:55 +03:00
parent aad8bf9c72
commit 94eaf1b162
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/quickbot_agent/config.py
View File

@@ -8,7 +8,7 @@ class Config(BaseSettings):
OPENAI_API_KEY: str OPENAI_API_KEY: str
OPENAI_MODEL: str = "gpt-4o-2024-11-20" OPENAI_MODEL: str = "gpt-4o-2024-11-20"
MESSAGE_HISTORY_DEPTH: int = 10 MESSAGE_HISTORY_DEPTH: int = 20
config = Config() config = Config()

12
src/quickbot_agent/handlers/messages.py
View File

@@ -15,10 +15,13 @@ from ..utils import get_message_log, add_message_log, generate_gpt_tools_schemas
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
import ujson as json import ujson as json
from ..config import config from ..config import config
from logging import getLogger
if TYPE_CHECKING: if TYPE_CHECKING:
from ..main import AgentPlugin from ..main import AgentPlugin
logger = getLogger(__name__)
router = Router() router = Router()
@@ -69,6 +72,9 @@ async def handle_openai_cycle(
else: else:
gpt_tools_metadata = plugin._gpt_tools_metadata 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( response = await client.chat.completions.create(
model=config.OPENAI_MODEL, model=config.OPENAI_MODEL,
messages=messages, messages=messages,
@@ -318,11 +324,9 @@ async def get_messages(
) )
messages = [ messages = [
{ {"role": "system", "content": system_prompt},
"role": "system",
"content": system_prompt,
}
] ]
for message_log_item in message_log: for message_log_item in message_log:
messages.append(json.loads(message_log_item.content)) messages.append(json.loads(message_log_item.content))