2ba0460f19
* feat: introduce file extract capability powered by MoonshotAI * fix: correct indentation in default configuration file * fix: add error handling for file extract application in InternalAgentSubStage * fix: update file name handling in InternalAgentSubStage to correctly associate file names with extracted content * feat: add condition settings for local agent runner in default configuration * fix: enhance file naming logic in File component and update prompt handling in InternalAgentSubStage
24 lines
697 B
Python
24 lines
697 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", # type: ignore
|
|
)
|
|
return (await client.files.content(file_id=file_object.id)).text
|