32 lines
996 B
Python
32 lines
996 B
Python
from conftest import create_test_module
|
|
from letta_client import UnprocessableEntityError
|
|
|
|
from letta.constants import CORE_MEMORY_HUMAN_CHAR_LIMIT, CORE_MEMORY_PERSONA_CHAR_LIMIT
|
|
|
|
BLOCKS_CREATE_PARAMS = [
|
|
("human_block", {"label": "human", "value": "test"}, {"limit": CORE_MEMORY_HUMAN_CHAR_LIMIT}, None),
|
|
("persona_block", {"label": "persona", "value": "test1"}, {"limit": CORE_MEMORY_PERSONA_CHAR_LIMIT}, None),
|
|
]
|
|
|
|
BLOCKS_UPDATE_PARAMS = [
|
|
("human_block", {"value": "test2"}, {}, None),
|
|
("persona_block", {"value": "testing testing testing", "limit": 10}, {}, UnprocessableEntityError),
|
|
]
|
|
|
|
BLOCKS_LIST_PARAMS = [
|
|
({}, 2),
|
|
({"label": "human"}, 1),
|
|
({"label": "persona"}, 1),
|
|
]
|
|
|
|
# Create all test module components at once
|
|
globals().update(
|
|
create_test_module(
|
|
resource_name="blocks",
|
|
id_param_name="block_id",
|
|
create_params=BLOCKS_CREATE_PARAMS,
|
|
update_params=BLOCKS_UPDATE_PARAMS,
|
|
list_params=BLOCKS_LIST_PARAMS,
|
|
)
|
|
)
|