github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix: formatting

Browse Source
This commit is contained in:
Nicholas Tindle
2025-04-10 18:11:07 -05:00
parent 74f91c6563
commit a08ca2af1c
2 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
autogpt_platform/backend/backend/executor/manager.py
View File

@@ -97,38 +97,48 @@ class LogMetadata:
msg = self._wrap(msg, **extra)
truncated_extra = self._truncate_large_data(extra)
truncated_metadata = self._truncate_large_data(self.metadata)
logger.info(msg, extra={"json_fields": {**truncated_metadata, **truncated_extra}})
logger.info(
msg, extra={"json_fields": {**truncated_metadata, **truncated_extra}}
)
def warning(self, msg: str, **extra):
msg = self._wrap(msg, **extra)
truncated_extra = self._truncate_large_data(extra)
truncated_metadata = self._truncate_large_data(self.metadata)
logger.warning(msg, extra={"json_fields": {**truncated_metadata, **truncated_extra}})
logger.warning(
msg, extra={"json_fields": {**truncated_metadata, **truncated_extra}}
)
def error(self, msg: str, **extra):
msg = self._wrap(msg, **extra)
truncated_extra = self._truncate_large_data(extra)
truncated_metadata = self._truncate_large_data(self.metadata)
logger.error(msg, extra={"json_fields": {**truncated_metadata, **truncated_extra}})
logger.error(
msg, extra={"json_fields": {**truncated_metadata, **truncated_extra}}
)
def debug(self, msg: str, **extra):
msg = self._wrap(msg, **extra)
truncated_extra = self._truncate_large_data(extra)
truncated_metadata = self._truncate_large_data(self.metadata)
logger.debug(msg, extra={"json_fields": {**truncated_metadata, **truncated_extra}})
logger.debug(
msg, extra={"json_fields": {**truncated_metadata, **truncated_extra}}
)
def exception(self, msg: str, **extra):
msg = self._wrap(msg, **extra)
truncated_extra = self._truncate_large_data(extra)
truncated_metadata = self._truncate_large_data(self.metadata)
logger.exception(msg, extra={"json_fields": {**truncated_metadata, **truncated_extra}})
logger.exception(
msg, extra={"json_fields": {**truncated_metadata, **truncated_extra}}
)
def _wrap(self, msg: str, **extra):
extra_msg = str(extra or "")
if len(extra_msg) > 1000:
extra_msg = extra_msg[:1000] + "..."
return f"{self.prefix} {msg} {extra_msg}"
def _truncate_large_data(self, data, max_size=10000):
if isinstance(data, dict):
return {k: self._truncate_large_data(v, max_size) for k, v in data.items()}

4
autogpt_platform/backend/test/test_data_creator.py
View File

@@ -10,16 +10,12 @@ from prisma.types import (
AgentGraphCreateInput,
AgentNodeCreateInput,
AgentNodeLinkCreateInput,
AgentPresetCreateInput,
AnalyticsDetailsCreateInput,
AnalyticsMetricsCreateInput,
APIKeyCreateInput,
CreditTransactionCreateInput,
LibraryAgentCreateInput,
ProfileCreateInput,
StoreListingCreateInput,
StoreListingReviewCreateInput,
StoreListingVersionCreateInput,
UserCreateInput,
)