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

fix: Fix open/close files tests and upgrade letta client (#3535)

Browse Source
This commit is contained in:
Matthew Zhou
2025-07-24 10:52:38 -07:00
committed by GitHub
parent 1997c0bcf8
commit 32bdbe6b2a
3 changed files with 76 additions and 528 deletions
Download Patch File Download Diff File Expand all files Collapse all files

592
poetry.lock generated
View File

File diff suppressed because it is too large Load Diff

2
pyproject.toml
View File

@@ -72,7 +72,7 @@ llama-index = "^0.12.2"
llama-index-embeddings-openai = "^0.3.1"
e2b-code-interpreter = {version = "^1.0.3", optional = true}
anthropic = "^0.49.0"
letta_client = "^0.1.219"
letta_client = "^0.1.220"
openai = "^1.60.0"
opentelemetry-api = "1.30.0"
opentelemetry-sdk = "1.30.0"

10
tests/test_sources.py
View File

@@ -995,7 +995,7 @@ def test_agent_open_file(disable_pinecone, client: LettaSDKClient, agent_state:
file_metadata = upload_file_and_wait(client, source.id, file_path)
# Basic test open_file function
closed_files = client.agents.open_file(agent_id=agent_state.id, file_id=file_metadata.id)
closed_files = client.agents.files.open(agent_id=agent_state.id, file_id=file_metadata.id)
assert len(closed_files) == 0
@@ -1012,10 +1012,10 @@ def test_agent_close_file(disable_pinecone, client: LettaSDKClient, agent_state:
file_metadata = upload_file_and_wait(client, source.id, file_path)
# First open the file
client.agents.open_file(agent_id=agent_state.id, file_id=file_metadata.id)
client.agents.files.open(agent_id=agent_state.id, file_id=file_metadata.id)
# Test close_file function
client.agents.close_file(agent_id=agent_state.id, file_id=file_metadata.id)
client.agents.files.close(agent_id=agent_state.id, file_id=file_metadata.id)
# Result can be None or any type based on the signature
# Just verify the function executes without error
@@ -1037,10 +1037,10 @@ def test_agent_close_all_open_files(disable_pinecone, client: LettaSDKClient, ag
file_metadata = upload_file_and_wait(client, source.id, file_path)
file_metadatas.append(file_metadata)
# Open each file
client.agents.open_file(agent_id=agent_state.id, file_id=file_metadata.id)
client.agents.files.open(agent_id=agent_state.id, file_id=file_metadata.id)
# Test close_all_open_files function
result = client.agents.close_all_open_files(agent_id=agent_state.id)
result = client.agents.files.close_all(agent_id=agent_state.id)
# Verify result is a list of strings
assert isinstance(result, list), f"Expected list, got {type(result)}"