From 699820cecd1a76022eadc7804cbf34b1535b3601 Mon Sep 17 00:00:00 2001 From: Ari Webb Date: Mon, 17 Nov 2025 18:06:04 -0800 Subject: [PATCH] fix: managers test (#6232) fix managers test Co-authored-by: Ari Webb --- tests/managers/conftest.py | 5 ++++ tests/managers/test_agent_manager.py | 29 ++++++++++++++++++++++++ tests/managers/test_agent_tag_manager.py | 6 +++++ tests/managers/test_archive_manager.py | 15 ++++++++++++ tests/managers/test_block_manager.py | 1 + tests/managers/test_group_manager.py | 1 + tests/managers/test_identity_manager.py | 2 ++ tests/managers/test_source_manager.py | 1 + tests/managers/test_tool_manager.py | 1 + 9 files changed, 61 insertions(+) diff --git a/tests/managers/conftest.py b/tests/managers/conftest.py index 7e06c67f..15e3f564 100644 --- a/tests/managers/conftest.py +++ b/tests/managers/conftest.py @@ -332,6 +332,7 @@ async def sarah_agent(server: SyncServer, default_user, default_organization): agent_state = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="sarah_agent", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -348,6 +349,7 @@ async def charles_agent(server: SyncServer, default_user, default_organization): agent_state = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="charles_agent", + agent_type="memgpt_v2_agent", memory_blocks=[CreateBlock(label="human", value="Charles"), CreateBlock(label="persona", value="I am a helpful assistant")], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -684,6 +686,7 @@ async def agent_with_tags(server: SyncServer, default_user): agent1 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent1", + agent_type="memgpt_v2_agent", tags=["primary_agent", "benefit_1"], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -696,6 +699,7 @@ async def agent_with_tags(server: SyncServer, default_user): agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent2", + agent_type="memgpt_v2_agent", tags=["primary_agent", "benefit_2"], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -708,6 +712,7 @@ async def agent_with_tags(server: SyncServer, default_user): agent3 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent3", + agent_type="memgpt_v2_agent", tags=["primary_agent", "benefit_1", "benefit_2"], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), diff --git a/tests/managers/test_agent_manager.py b/tests/managers/test_agent_manager.py index 9c870fbf..d463673b 100644 --- a/tests/managers/test_agent_manager.py +++ b/tests/managers/test_agent_manager.py @@ -166,6 +166,7 @@ async def test_create_agent_include_base_tools(server: SyncServer, default_user) create_agent_request = CreateAgent( name="test_default_source_agent", + agent_type="memgpt_v2_agent", system="test system", memory_blocks=memory_blocks, llm_config=LLMConfig.default_config("gpt-4o-mini"), @@ -196,6 +197,7 @@ async def test_create_agent_base_tool_rules_excluded_providers(server: SyncServe # Test with excluded provider (openai) create_agent_request = CreateAgent( name="test_excluded_provider_agent", + agent_type="memgpt_v2_agent", system="test system", memory_blocks=memory_blocks, llm_config=LLMConfig.default_config("gpt-4o-mini"), # This has model_endpoint_type="openai" @@ -225,6 +227,7 @@ async def test_create_agent_base_tool_rules_non_excluded_providers(server: SyncS # Test with non-excluded provider (together) create_agent_request = CreateAgent( name="test_non_excluded_provider_agent", + agent_type="memgpt_v2_agent", system="test system", memory_blocks=memory_blocks, llm_config=LLMConfig( @@ -321,6 +324,7 @@ async def test_create_agent_with_default_source(server: SyncServer, default_user create_agent_request = CreateAgent( name="test_default_source_agent", + agent_type="memgpt_v2_agent", system="test system", memory_blocks=memory_blocks, llm_config=LLMConfig.default_config("gpt-4o-mini"), @@ -355,6 +359,7 @@ async def test_create_agent_with_default_source(server: SyncServer, default_user # Test with include_default_source=False create_agent_request_no_source = CreateAgent( name="test_no_default_source_agent", + agent_type="memgpt_v2_agent", system="test system", memory_blocks=memory_blocks, llm_config=LLMConfig.default_config("gpt-4o-mini"), @@ -413,6 +418,7 @@ async def test_get_context_window_basic( async def test_create_agent_passed_in_initial_messages(server: SyncServer, default_user, default_block): memory_blocks = [CreateBlock(label="human", value="BananaBoy"), CreateBlock(label="persona", value="I am a helpful assistant")] create_agent_request = CreateAgent( + agent_type="memgpt_v2_agent", system="test system", memory_blocks=memory_blocks, llm_config=LLMConfig.default_config("gpt-4o-mini"), @@ -442,6 +448,7 @@ async def test_create_agent_passed_in_initial_messages(server: SyncServer, defau async def test_create_agent_default_initial_message(server: SyncServer, default_user, default_block): memory_blocks = [CreateBlock(label="human", value="BananaBoy"), CreateBlock(label="persona", value="I am a helpful assistant")] create_agent_request = CreateAgent( + agent_type="memgpt_v2_agent", system="test system", memory_blocks=memory_blocks, llm_config=LLMConfig.default_config("gpt-4o-mini"), @@ -472,6 +479,7 @@ async def test_create_agent_with_json_in_system_message(server: SyncServer, defa "{'query1' : 'reason1', 'query2' : 'reason2'}" ) create_agent_request = CreateAgent( + agent_type="memgpt_v2_agent", system=system_prompt, llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -527,6 +535,7 @@ async def test_agent_file_defaults_based_on_context_window(server: SyncServer, d llm_config_small.context_window = 8000 create_agent_request = CreateAgent( name="test_agent_small_context", + agent_type="memgpt_v2_agent", llm_config=llm_config_small, embedding_config=EmbeddingConfig.default_config(provider="openai"), block_ids=[default_block.id], @@ -547,6 +556,7 @@ async def test_agent_file_defaults_based_on_context_window(server: SyncServer, d llm_config_medium.context_window = 32000 create_agent_request = CreateAgent( name="test_agent_medium_context", + agent_type="memgpt_v2_agent", llm_config=llm_config_medium, embedding_config=EmbeddingConfig.default_config(provider="openai"), block_ids=[default_block.id], @@ -567,6 +577,7 @@ async def test_agent_file_defaults_based_on_context_window(server: SyncServer, d llm_config_large.context_window = 128000 create_agent_request = CreateAgent( name="test_agent_large_context", + agent_type="memgpt_v2_agent", llm_config=llm_config_large, embedding_config=EmbeddingConfig.default_config(provider="openai"), block_ids=[default_block.id], @@ -591,6 +602,7 @@ async def test_agent_file_defaults_explicit_values(server: SyncServer, default_u llm_config_explicit.context_window = 32000 # would normally get defaults of 5 and 30k create_agent_request = CreateAgent( name="test_agent_explicit_values", + agent_type="memgpt_v2_agent", llm_config=llm_config_explicit, embedding_config=EmbeddingConfig.default_config(provider="openai"), block_ids=[default_block.id], @@ -775,6 +787,7 @@ async def test_list_agents_ascending(server: SyncServer, default_user): agent1 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_oldest", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -789,6 +802,7 @@ async def test_list_agents_ascending(server: SyncServer, default_user): agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_newest", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -808,6 +822,7 @@ async def test_list_agents_descending(server: SyncServer, default_user): agent1 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_oldest", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -822,6 +837,7 @@ async def test_list_agents_descending(server: SyncServer, default_user): agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_newest", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -841,6 +857,7 @@ async def test_list_agents_by_last_stop_reason(server: SyncServer, default_user) agent1 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_requires_approval", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -858,6 +875,7 @@ async def test_list_agents_by_last_stop_reason(server: SyncServer, default_user) agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_error", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -875,6 +893,7 @@ async def test_list_agents_by_last_stop_reason(server: SyncServer, default_user) agent3 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_no_stop_reason", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -908,6 +927,7 @@ async def test_count_agents_with_filters(server: SyncServer, default_user): agent1 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_requires_approval", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -925,6 +945,7 @@ async def test_count_agents_with_filters(server: SyncServer, default_user): agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_error", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -942,6 +963,7 @@ async def test_count_agents_with_filters(server: SyncServer, default_user): agent3 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_completed", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -959,6 +981,7 @@ async def test_count_agents_with_filters(server: SyncServer, default_user): agent4 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_no_stop_reason", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), memory_blocks=[], @@ -1019,6 +1042,7 @@ async def test_list_agents_ordering_and_pagination(server: SyncServer, default_u agent = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name=name, + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -1095,6 +1119,7 @@ async def test_agent_environment_variables_encrypt_on_create(server: SyncServer, # Create agent with secrets agent_create = CreateAgent( name="test-agent-with-secrets", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=DEFAULT_EMBEDDING_CONFIG, include_base_tools=False, @@ -1146,6 +1171,7 @@ async def test_agent_environment_variables_decrypt_on_read(server: SyncServer, d # Create agent with secrets agent_create = CreateAgent( name="test-agent-read-secrets", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=DEFAULT_EMBEDDING_CONFIG, include_base_tools=False, @@ -1192,6 +1218,7 @@ async def test_agent_environment_variables_update_encryption(server: SyncServer, # Create agent with initial secrets agent_create = CreateAgent( name="test-agent-update-secrets", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=DEFAULT_EMBEDDING_CONFIG, include_base_tools=False, @@ -1607,6 +1634,7 @@ async def test_agent_state_relationship_loads(server: SyncServer, default_user, create_agent_request = CreateAgent( name="test_default_source_agent", + agent_type="memgpt_v2_agent", system="test system", memory_blocks=memory_blocks, llm_config=LLMConfig.default_config("gpt-4o-mini"), @@ -1702,6 +1730,7 @@ async def test_create_template_agent_with_files_from_sources(server: SyncServer, # Create agent using InternalTemplateAgentCreate with the source create_agent_request = InternalTemplateAgentCreate( name="test_template_agent_with_files", + agent_type="memgpt_v2_agent", system="test system", memory_blocks=memory_blocks, llm_config=LLMConfig.default_config("gpt-4o-mini"), diff --git a/tests/managers/test_agent_tag_manager.py b/tests/managers/test_agent_tag_manager.py index 6eaf8809..de67136c 100644 --- a/tests/managers/test_agent_tag_manager.py +++ b/tests/managers/test_agent_tag_manager.py @@ -231,6 +231,7 @@ async def test_list_agents_by_tags_pagination(server: SyncServer, default_user, agent1 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent1", + agent_type="memgpt_v2_agent", tags=["pagination_test", "tag1"], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -247,6 +248,7 @@ async def test_list_agents_by_tags_pagination(server: SyncServer, default_user, agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent2", + agent_type="memgpt_v2_agent", tags=["pagination_test", "tag2"], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -289,6 +291,7 @@ async def test_list_agents_query_text_pagination(server: SyncServer, default_use agent1 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="Search Agent One", + agent_type="memgpt_v2_agent", memory_blocks=[], description="This is a search agent for testing", llm_config=LLMConfig.default_config("gpt-4o-mini"), @@ -304,6 +307,7 @@ async def test_list_agents_query_text_pagination(server: SyncServer, default_use agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="Search Agent Two", + agent_type="memgpt_v2_agent", memory_blocks=[], description="Another search agent for testing", llm_config=LLMConfig.default_config("gpt-4o-mini"), @@ -319,6 +323,7 @@ async def test_list_agents_query_text_pagination(server: SyncServer, default_use agent3 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="Different Agent", + agent_type="memgpt_v2_agent", memory_blocks=[], description="This is a different agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), @@ -379,6 +384,7 @@ async def test_list_tags(server: SyncServer, default_user, default_organization) actor=default_user, agent_create=CreateAgent( name="tag_agent_" + str(i), + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), diff --git a/tests/managers/test_archive_manager.py b/tests/managers/test_archive_manager.py index ece11330..33241f08 100644 --- a/tests/managers/test_archive_manager.py +++ b/tests/managers/test_archive_manager.py @@ -137,6 +137,7 @@ async def test_archive_manager_get_agents_for_archive_async(server: SyncServer, agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="test_agent_2", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -175,6 +176,7 @@ async def test_archive_manager_race_condition_handling(server: SyncServer, defau agent = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="test_agent_race_condition", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -431,6 +433,7 @@ async def test_archive_manager_attach_agent_to_archive_async(server: SyncServer, agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="test_attach_agent", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -486,6 +489,7 @@ async def test_archive_manager_detach_agent_from_archive_async(server: SyncServe agent1 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="test_detach_agent_1", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -497,6 +501,7 @@ async def test_archive_manager_detach_agent_from_archive_async(server: SyncServe agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="test_detach_agent_2", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -560,6 +565,7 @@ async def test_archive_manager_attach_detach_idempotency(server: SyncServer, def agent = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="idempotency_test_agent", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -588,6 +594,7 @@ async def test_archive_manager_attach_detach_idempotency(server: SyncServer, def non_existent_agent = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="never_attached_agent", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -625,6 +632,7 @@ async def test_archive_manager_detach_with_multiple_archives(server: SyncServer, agent1 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="multi_test_agent_1", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -636,6 +644,7 @@ async def test_archive_manager_detach_with_multiple_archives(server: SyncServer, agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="multi_test_agent_2", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -691,6 +700,7 @@ async def test_archive_manager_detach_deleted_agent(server: SyncServer, default_ agent = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent_to_be_deleted", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -735,6 +745,7 @@ async def test_archive_manager_cascade_delete_on_archive_deletion(server: SyncSe agent1 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="cascade_test_agent_1", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -746,6 +757,7 @@ async def test_archive_manager_cascade_delete_on_archive_deletion(server: SyncSe agent2 = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="cascade_test_agent_2", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -816,6 +828,7 @@ async def test_archive_manager_list_agents_with_pagination(server: SyncServer, d agent = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name=f"pagination_test_agent_{i}", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -872,6 +885,7 @@ async def test_archive_manager_get_default_archive_for_agent_async(server: SyncS agent = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="test_default_archive_agent", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -941,6 +955,7 @@ async def test_archive_manager_get_agents_with_include_parameter(server: SyncSer agent = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="test_include_agent", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), diff --git a/tests/managers/test_block_manager.py b/tests/managers/test_block_manager.py index c3432847..f4007f24 100644 --- a/tests/managers/test_block_manager.py +++ b/tests/managers/test_block_manager.py @@ -232,6 +232,7 @@ async def test_refresh_memory_async(server: SyncServer, default_user): agent = await server.agent_manager.create_agent_async( CreateAgent( name="test", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), include_base_tools=False, diff --git a/tests/managers/test_group_manager.py b/tests/managers/test_group_manager.py index 08bf3d60..47a94df3 100644 --- a/tests/managers/test_group_manager.py +++ b/tests/managers/test_group_manager.py @@ -118,6 +118,7 @@ async def test_create_internal_template_objects(server: SyncServer, default_user agent = await server.agent_manager.create_agent_async( InternalTemplateAgentCreate( name="template-agent", + agent_type="memgpt_v2_agent", base_template_id=base_template_id, template_id=template_id, deployment_id=deployment_id, diff --git a/tests/managers/test_identity_manager.py b/tests/managers/test_identity_manager.py index 7b060af6..9534e306 100644 --- a/tests/managers/test_identity_manager.py +++ b/tests/managers/test_identity_manager.py @@ -235,6 +235,7 @@ async def test_get_set_agents_for_identities(server: SyncServer, sarah_agent, ch agent_with_identity = await server.create_agent_async( CreateAgent( + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), @@ -245,6 +246,7 @@ async def test_get_set_agents_for_identities(server: SyncServer, sarah_agent, ch ) agent_without_identity = await server.create_agent_async( CreateAgent( + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), diff --git a/tests/managers/test_source_manager.py b/tests/managers/test_source_manager.py index 5d13fdbf..b3a2418e 100644 --- a/tests/managers/test_source_manager.py +++ b/tests/managers/test_source_manager.py @@ -247,6 +247,7 @@ async def test_get_agents_for_source_id_pagination(server: SyncServer, default_s agent = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name=f"Test Agent {i}", + agent_type="memgpt_v2_agent", memory_blocks=[], llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), diff --git a/tests/managers/test_tool_manager.py b/tests/managers/test_tool_manager.py index be87e01b..46e1a56e 100644 --- a/tests/managers/test_tool_manager.py +++ b/tests/managers/test_tool_manager.py @@ -538,6 +538,7 @@ async def test_attach_tool_with_default_requires_approval_on_creation(server: Sy agent = await server.agent_manager.create_agent_async( agent_create=CreateAgent( name="agent11", + agent_type="memgpt_v2_agent", llm_config=LLMConfig.default_config("gpt-4o-mini"), embedding_config=EmbeddingConfig.default_config(provider="openai"), tools=[bash_tool.name],