소스 검색

feat(notion): Notion Database extracts Rows content `in row order` and appends `Row Page URL` (#22646)

Co-authored-by: Aurelius Huang <cm.huang@aftership.com>
tags/1.7.2
Aurelius Huang 3 달 전
부모
커밋
ffddabde43
No account linked to committer's email address
1개의 변경된 파일4개의 추가작업 그리고 1개의 파일을 삭제
  1. 4
    1
      api/core/rag/extractor/notion_extractor.py

+ 4
- 1
api/core/rag/extractor/notion_extractor.py 파일 보기

@@ -1,5 +1,6 @@
import json
import logging
import operator
from typing import Any, Optional, cast

import requests
@@ -130,13 +131,15 @@ class NotionExtractor(BaseExtractor):
data[property_name] = value
row_dict = {k: v for k, v in data.items() if v}
row_content = ""
for key, value in row_dict.items():
for key, value in sorted(row_dict.items(), key=operator.itemgetter(0)):
if isinstance(value, dict):
value_dict = {k: v for k, v in value.items() if v}
value_content = "".join(f"{k}:{v} " for k, v in value_dict.items())
row_content = row_content + f"{key}:{value_content}\n"
else:
row_content = row_content + f"{key}:{value}\n"
if "url" in result:
row_content = row_content + f"Row Page URL:{result.get('url', '')}\n"
database_content.append(row_content)

has_more = response_data.get("has_more", False)

Loading…
취소
저장