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

fix: remove letta-free embeddings from testing (#6870)

Browse Source
This commit is contained in:
Sarah Wooders
2025-12-14 18:55:40 -08:00
committed by Caren Thomas
parent 33d39f4643
commit 0c0ba5d03d
23 changed files with 236 additions and 168 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_sources.py
View File

@@ -1243,14 +1243,12 @@ def test_file_processing_timeout_logic():
assert not (recent_time < timeout_threshold), "Recent file should not trigger timeout"
def test_letta_free_embedding(disable_pinecone, disable_turbopuffer, client: LettaSDKClient):
"""Test creating a source with letta/letta-free embedding and uploading a file"""
# create a source with letta-free embedding
source = client.folders.create(name="test_letta_free_source", embedding="letta/letta-free")
def test_openai_embedding(disable_pinecone, disable_turbopuffer, client: LettaSDKClient):
"""Test creating a source with OpenAI embeddings and uploading a file"""
source = client.folders.create(name="test_openai_embed_source", embedding="openai/text-embedding-3-small")
# verify source was created with correct embedding
assert source.name == "test_letta_free_source"
# assert source.embedding_config.embedding_model == "letta-free"
assert source.name == "test_openai_embed_source"
# upload test.txt file
file_path = "tests/data/test.txt"