Revert "feat: provider-specific model configuration " (#5873)
Revert "feat: provider-specific model configuration (#5774)" This reverts commit 34a334949a3ef72cd49ff0ca3da9e85d16daa57c.
This commit is contained in:
committed by
Caren Thomas
parent
d3ab51b822
commit
cfeed463a9
@@ -587,6 +587,37 @@ def test_agent_creation(client: Letta):
|
||||
client.agents.delete(agent_id=agent.id)
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------------------------------------------------
|
||||
# Agent sources
|
||||
# --------------------------------------------------------------------------------------------------------------------
|
||||
def test_attach_detach_agent_source(client: Letta, agent: AgentState):
|
||||
"""Test that we can attach and detach a source from an agent"""
|
||||
|
||||
# Create a source
|
||||
source = client.sources.create(
|
||||
name="test_source",
|
||||
embedding="openai/text-embedding-3-small",
|
||||
)
|
||||
initial_sources = client.agents.sources.list(agent_id=agent.id)
|
||||
assert source.id not in [s.id for s in initial_sources]
|
||||
|
||||
# Attach source
|
||||
client.agents.sources.attach(agent_id=agent.id, source_id=source.id)
|
||||
|
||||
# Verify source is attached
|
||||
final_sources = client.agents.sources.list(agent_id=agent.id)
|
||||
assert source.id in [s.id for s in final_sources]
|
||||
|
||||
# Detach source
|
||||
client.agents.sources.detach(agent_id=agent.id, source_id=source.id)
|
||||
|
||||
# Verify source is detached
|
||||
final_sources = client.agents.sources.list(agent_id=agent.id)
|
||||
assert source.id not in [s.id for s in final_sources]
|
||||
|
||||
client.sources.delete(source.id)
|
||||
|
||||
|
||||
# --------------------------------------------------------------------------------------------------------------------
|
||||
# Agent Initial Message Sequence
|
||||
# --------------------------------------------------------------------------------------------------------------------
|
||||
|
||||
Reference in New Issue
Block a user