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

fix: ask for embedding config for letta load directory ... (#2184)

Browse Source
This commit is contained in:
Sarah Wooders
2024-12-09 15:11:31 -08:00
committed by GitHub
parent 6c4f8d5d56
commit 1c13e223c3
3 changed files with 23 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
letta/client/client.py
View File

@@ -1319,6 +1319,7 @@ class RESTClient(AbstractClient):
Returns:
source (Source): Created source
"""
assert embedding_config or self._default_embedding_config, f"Must specify embedding_config for source"
source_create = SourceCreate(name=name, embedding_config=embedding_config or self._default_embedding_config)
payload = source_create.model_dump()
response = requests.post(f"{self.base_url}/{self.api_prefix}/sources", json=payload, headers=self.headers)
@@ -2896,6 +2897,7 @@ class LocalClient(AbstractClient):
Returns:
source (Source): Created source
"""
assert embedding_config or self._default_embedding_config, f"Must specify embedding_config for source"
source = Source(
name=name, embedding_config=embedding_config or self._default_embedding_config, organization_id=self.user.organization_id
)