feat: message orm migration (#2144)
Co-authored-by: Mindy Long <mindy@letta.com> Co-authored-by: Sarah Wooders <sarahwooders@gmail.com> Co-authored-by: Matt Zhou <mattzh1314@gmail.com>
This commit is contained in:
@@ -58,21 +58,6 @@ def clear_tables():
|
||||
Sandbox.connect(sandbox.sandbox_id).kill()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_e2b_api_key_none():
|
||||
# Store the original value of e2b_api_key
|
||||
original_api_key = tool_settings.e2b_api_key
|
||||
|
||||
# Set e2b_api_key to None
|
||||
tool_settings.e2b_api_key = None
|
||||
|
||||
# Yield control to the test
|
||||
yield
|
||||
|
||||
# Restore the original value of e2b_api_key
|
||||
tool_settings.e2b_api_key = original_api_key
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def check_e2b_key_is_set():
|
||||
original_api_key = tool_settings.e2b_api_key
|
||||
|
||||
Reference in New Issue
Block a user