Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: managers test (#6232)

Browse Source 
fix managers test

Co-authored-by: Ari Webb <ari@letta.com>
This commit is contained in:
Ari Webb
2025-11-17 18:06:04 -08:00
committed by Caren Thomas
parent b0779f7969
commit 699820cecd
9 changed files with 61 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/managers/conftest.py
View File

@@ -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"),

29
tests/managers/test_agent_manager.py
View File

@@ -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"),

6
tests/managers/test_agent_tag_manager.py
View File

@@ -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"),

15
tests/managers/test_archive_manager.py
View File

@@ -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"),

1
tests/managers/test_block_manager.py
View File

@@ -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,

1
tests/managers/test_group_manager.py
View File

@@ -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,

2
tests/managers/test_identity_manager.py
View File

@@ -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"),

1
tests/managers/test_source_manager.py
View File

@@ -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"),

1
tests/managers/test_tool_manager.py
View File

@@ -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],