From 7e58f9d31721431a6d5e6b2e9e8e686bc3cec098 Mon Sep 17 00:00:00 2001 From: cthomas Date: Mon, 26 May 2025 13:25:22 -0700 Subject: [PATCH] feat(asyncify): actor load in source upload (#2437) --- letta/server/server.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/letta/server/server.py b/letta/server/server.py index 10a56e8a..b74fcee0 100644 --- a/letta/server/server.py +++ b/letta/server/server.py @@ -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]: