mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-16 09:40:30 +08:00
24 lines
681 B
Python
24 lines
681 B
Python
from pathlib import Path
|
|
|
|
from openai import AsyncOpenAI
|
|
|
|
|
|
async def extract_file_moonshotai(file_path: str, api_key: str) -> str:
|
|
"""Extract text from a file using Moonshot AI API"""
|
|
"""
|
|
Args:
|
|
file_path: The path to the file to extract text from
|
|
api_key: The API key to use to extract text from the file
|
|
Returns:
|
|
The text extracted from the file
|
|
"""
|
|
client = AsyncOpenAI(
|
|
api_key=api_key,
|
|
base_url="https://api.moonshot.cn/v1",
|
|
)
|
|
file_object = await client.files.create(
|
|
file=Path(file_path),
|
|
purpose="file-extract",
|
|
)
|
|
return (await client.files.content(file_id=file_object.id)).text
|