diff --git a/memgpt/client/client.py b/memgpt/client/client.py index e02b7c8219..17ea6478ae 100644 --- a/memgpt/client/client.py +++ b/memgpt/client/client.py @@ -924,7 +924,7 @@ def load_data(self, connector: DataConnector, source_name: str): self.server.load_data(user_id=self.user_id, connector=connector, source_name=source_name) def create_source(self, name: str): - self.server.create_source(user_id=self.user_id, name=name) + return self.server.create_source(user_id=self.user_id, name=name) def attach_source_to_agent(self, source_id: uuid.UUID, agent_id: uuid.UUID): self.server.attach_source_to_agent(user_id=self.user_id, source_id=source_id, agent_id=agent_id)