Skip to content

Commit

Permalink
fix: correct the input params of "load_data" in NotionPageReader
Browse files Browse the repository at this point in the history
  • Loading branch information
minglu7 committed Jan 16, 2025
1 parent 0fca89a commit 7c8c59f
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 7c8c59f

Please sign in to comment.