87 lines
2.7 KiB
Python
87 lines
2.7 KiB
Python
"""Feishu Document Tool -- read document content via Feishu/Lark API.
|
|
|
|
Provides ``feishu_doc_read`` for reading document content as plain text.
|
|
Uses the same lazy-import + BaseRequest pattern as feishu_comment.py.
|
|
"""
|
|
|
|
from tools.feishu_lark import ( # noqa: F401 (set_client/get_client are imported by feishu_comment)
|
|
build_request,
|
|
_check_feishu,
|
|
get_client,
|
|
raw_body,
|
|
set_client,
|
|
)
|
|
from tools.registry import registry, tool_error, tool_result
|
|
|
|
_RAW_CONTENT_URI = "/open-apis/docx/v1/documents/:document_id/raw_content"
|
|
|
|
FEISHU_DOC_READ_SCHEMA = {
|
|
"name": "feishu_doc_read",
|
|
"description": (
|
|
"Read the full content of a Feishu/Lark document as plain text. "
|
|
"Useful when you need more context beyond the quoted text in a comment."
|
|
),
|
|
"parameters": {
|
|
"type": "object",
|
|
"properties": {
|
|
"doc_token": {
|
|
"type": "string",
|
|
"description": "The document token (from the document URL or comment context).",
|
|
},
|
|
},
|
|
"required": ["doc_token"],
|
|
},
|
|
}
|
|
|
|
|
|
def _handle_feishu_doc_read(args: dict, **kwargs) -> str:
|
|
doc_token = args.get("doc_token", "").strip()
|
|
if not doc_token:
|
|
return tool_error("doc_token is required")
|
|
|
|
client = get_client()
|
|
if client is None:
|
|
return tool_error("Feishu client not available (not in a Feishu comment context)")
|
|
|
|
try:
|
|
request = build_request("GET", _RAW_CONTENT_URI, paths={"document_id": doc_token})
|
|
except ImportError:
|
|
return tool_error("lark_oapi not installed")
|
|
|
|
# Tool handlers run synchronously in a worker thread (no running event
|
|
# loop), so call the blocking lark client directly.
|
|
response = client.request(request)
|
|
|
|
code = getattr(response, "code", None)
|
|
if code != 0:
|
|
msg = getattr(response, "msg", "unknown error")
|
|
return tool_error(f"Failed to read document: code={code} msg={msg}")
|
|
|
|
body = raw_body(response)
|
|
if body is not None:
|
|
try:
|
|
return tool_result(success=True, content=body.get("data", {}).get("content", ""))
|
|
except AttributeError:
|
|
pass
|
|
|
|
# Fallback: try the typed response.data
|
|
data = getattr(response, "data", None)
|
|
if data:
|
|
content = data.get("content", "") if isinstance(data, dict) else getattr(data, "content", str(data))
|
|
return tool_result(success=True, content=content)
|
|
|
|
return tool_error("No content returned from document API")
|
|
|
|
|
|
registry.register(
|
|
name="feishu_doc_read",
|
|
toolset="feishu_doc",
|
|
schema=FEISHU_DOC_READ_SCHEMA,
|
|
handler=_handle_feishu_doc_read,
|
|
check_fn=_check_feishu,
|
|
requires_env=[],
|
|
is_async=False,
|
|
description="Read Feishu document content",
|
|
emoji="\U0001f4c4",
|
|
)
|