diff --git a/llama-index-integrations/tools/llama-index-tools-notion/llama_index/tools/notion/base.py b/llama-index-integrations/tools/llama-index-tools-notion/llama_index/tools/notion/base.py index ed5a92f0cdbf8..88eea4b13c3fe 100644 --- a/llama-index-integrations/tools/llama-index-tools-notion/llama_index/tools/notion/base.py +++ b/llama-index-integrations/tools/llama-index-tools-notion/llama_index/tools/notion/base.py @@ -54,15 +54,15 @@ def get_fn_schema_from_fn_name( raise ValueError(f"Invalid function name: {fn_name}") def load_data( - self, page_ids: Optional[List[str]] = None, database_id: Optional[str] = None + self, page_ids: Optional[List[str]] = None, database_ids: Optional[List[str]] = None, ) -> str: - """Loads content from a set of page ids or a database id. + """Loads content from a set of page ids or database ids. - Don't use this endpoint if you don't know the page ids or database id. + Don't use this endpoint if you don't know the page ids or database ids. """ page_ids = page_ids or [] - docs = self.reader.load_data(page_ids=page_ids, database_id=database_id) + docs = self.reader.load_data(page_ids=page_ids, database_ids=database_ids) return "\n".join([doc.get_content() for doc in docs]) def search_data(