mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-02-12 07:45:14 -05:00
113e87a23c
I'm getting circular import issues because there is a lot of cross-importing between `backend.data`, `backend.blocks`, and other modules. This change reduces block-related cross-imports and thus risk of breaking circular imports. ### Changes 🏗️ - Strip down `backend.data.block` - Move `Block` base class and related class/enum defs to `backend.blocks._base` - Move `is_block_auth_configured` to `backend.blocks._utils` - Move `get_blocks()`, `get_io_block_ids()` etc. to `backend.blocks` (`__init__.py`) - Update imports everywhere - Remove unused and poorly typed `Block.create()` - Change usages from `block_cls.create()` to `block_cls()` - Improve typing of `load_all_blocks` and `get_blocks` - Move cross-import of `backend.api.features.library.model` from `backend/data/__init__.py` to `backend/data/integrations.py` - Remove deprecated attribute `NodeModel.webhook` - Re-generate OpenAPI spec and fix frontend usage - Eliminate module-level `backend.blocks` import from `blocks/agent.py` - Eliminate module-level `backend.data.execution` and `backend.executor.manager` imports from `blocks/helpers/review.py` - Replace `BlockInput` with `GraphInput` for graph inputs ### Checklist 📋 #### For code changes: - [x] I have clearly listed my changes in the PR description - [x] I have made a test plan - [x] I have tested my changes according to the test plan: - CI static type-checking + tests should be sufficient for this
78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
"""Text encoding block for converting special characters to escape sequences."""
|
|
|
|
import codecs
|
|
|
|
from backend.blocks._base import (
|
|
Block,
|
|
BlockCategory,
|
|
BlockOutput,
|
|
BlockSchemaInput,
|
|
BlockSchemaOutput,
|
|
)
|
|
from backend.data.model import SchemaField
|
|
|
|
|
|
class TextEncoderBlock(Block):
|
|
"""
|
|
Encodes a string by converting special characters into escape sequences.
|
|
|
|
This block is the inverse of TextDecoderBlock. It takes text containing
|
|
special characters (like newlines, tabs, etc.) and converts them into
|
|
their escape sequence representations (e.g., newline becomes \\n).
|
|
"""
|
|
|
|
class Input(BlockSchemaInput):
|
|
"""Input schema for TextEncoderBlock."""
|
|
|
|
text: str = SchemaField(
|
|
description="A string containing special characters to be encoded",
|
|
placeholder="Your text with newlines and quotes to encode",
|
|
)
|
|
|
|
class Output(BlockSchemaOutput):
|
|
"""Output schema for TextEncoderBlock."""
|
|
|
|
encoded_text: str = SchemaField(
|
|
description="The encoded text with special characters converted to escape sequences"
|
|
)
|
|
error: str = SchemaField(description="Error message if encoding fails")
|
|
|
|
def __init__(self):
|
|
super().__init__(
|
|
id="5185f32e-4b65-4ecf-8fbb-873f003f09d6",
|
|
description="Encodes a string by converting special characters into escape sequences",
|
|
categories={BlockCategory.TEXT},
|
|
input_schema=TextEncoderBlock.Input,
|
|
output_schema=TextEncoderBlock.Output,
|
|
test_input={
|
|
"text": """Hello
|
|
World!
|
|
This is a "quoted" string."""
|
|
},
|
|
test_output=[
|
|
(
|
|
"encoded_text",
|
|
"""Hello\\nWorld!\\nThis is a "quoted" string.""",
|
|
)
|
|
],
|
|
)
|
|
|
|
async def run(self, input_data: Input, **kwargs) -> BlockOutput:
|
|
"""
|
|
Encode the input text by converting special characters to escape sequences.
|
|
|
|
Args:
|
|
input_data: The input containing the text to encode.
|
|
**kwargs: Additional keyword arguments (unused).
|
|
|
|
Yields:
|
|
The encoded text with escape sequences, or an error message if encoding fails.
|
|
"""
|
|
try:
|
|
encoded_text = codecs.encode(input_data.text, "unicode_escape").decode(
|
|
"utf-8"
|
|
)
|
|
yield "encoded_text", encoded_text
|
|
except Exception as e:
|
|
yield "error", f"Encoding error: {str(e)}"
|