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

feat(asyncify): actor load in source upload (#2437)

Browse Source
This commit is contained in:
cthomas
2025-05-26 13:25:22 -07:00
committed by GitHub
parent 6913e7d986
commit 7e58f9d317
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
letta/server/server.py
View File

@@ -1384,13 +1384,13 @@ class SyncServer(Server):
# TODO: this should be implemented as a batch job or at least async, since it may take a long time
# load data from a data source into the document store
user = self.user_manager.get_user_by_id(user_id=user_id)
source = await self.source_manager.get_source_by_name(source_name=source_name, actor=user)
actor = await self.user_manager.get_actor_by_id_async(actor_id=user_id)
source = await self.source_manager.get_source_by_name(source_name=source_name, actor=actor)
if source is None:
raise ValueError(f"Data source {source_name} does not exist for user {user_id}")
# load data into the document store
passage_count, document_count = await load_data(connector, source, self.passage_manager, self.source_manager, actor=user)
passage_count, document_count = await load_data(connector, source, self.passage_manager, self.source_manager, actor=actor)
return passage_count, document_count
def list_data_source_passages(self, user_id: str, source_id: str) -> List[Passage]: