NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

logger changes

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-02-27 16:13:51 -05:00
parent 64d3ee1bd6
commit 35587cbdc0
3 changed files with 63 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

19
atroposlib/envs/server_handling/managed_server.py
View file

@ -8,6 +8,7 @@ This wrapper maintains a tree structure of sequences, where:
"""
import os
import logging
import time
import uuid
import warnings
@ -23,6 +24,8 @@ from pydantic import BaseModel
from atroposlib.envs.server_handling.server_baseline import APIServer
logger = logging.getLogger(__name__)
class SequenceNode(BaseModel):
"""
@ -292,16 +295,14 @@ class ManagedServer:
if self._debug_requests_enabled():
msg_count = len(messages)
prompt_preview = prompt.replace("\n", "\\n")[:600]
print(
f"[ATROPOS_REQ_DEBUG] chat_completion messages={msg_count} "
f"n={completion_kwargs.get('n')} max_tokens={completion_kwargs.get('max_tokens')} "
f"temperature={completion_kwargs.get('temperature')}",
flush=True,
)
print(
f"[ATROPOS_REQ_DEBUG] prompt_preview={prompt_preview!r}",
flush=True,
logger.debug(
"[ATROPOS_REQ_DEBUG] chat_completion messages=%s n=%s max_tokens=%s temperature=%s",
msg_count,
completion_kwargs.get("n"),
completion_kwargs.get("max_tokens"),
completion_kwargs.get("temperature"),
)
logger.debug("[ATROPOS_REQ_DEBUG] prompt_preview=%r", prompt_preview)
# Set model name if not provided
if "model" not in completion_kwargs: