71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
import os
|
||
from fastapi import APIRouter, Body
|
||
from pydantic import BaseModel
|
||
from typing import List, Dict, Any, Optional
|
||
|
||
from src.agents import agent_manager
|
||
|
||
tool = APIRouter(prefix="/tool")
|
||
|
||
|
||
class Tool(BaseModel):
|
||
name: str
|
||
title: str
|
||
description: str
|
||
url: str
|
||
method: Optional[str] = "POST"
|
||
params: Optional[Dict[str, Any]] = None
|
||
metadata: Optional[Dict[str, Any]] = None
|
||
|
||
@tool.get("/", response_model=List[Tool])
|
||
async def route_index():
|
||
tools = [
|
||
Tool(
|
||
name="text-chunking",
|
||
title="文本分块",
|
||
description="将文本分块以更好地理解。可以输入文本或者上传文件。",
|
||
url="/tools/text-chunking",
|
||
method="POST",
|
||
),
|
||
Tool(
|
||
name="pdf2txt",
|
||
title="PDF转文本",
|
||
description="将PDF文件转换为文本文件。",
|
||
url="/tools/pdf2txt",
|
||
method="POST",
|
||
),
|
||
Tool(
|
||
name="agent",
|
||
title="智能体（Dev）",
|
||
description="智能体演练平台",
|
||
url="/tools/agent",
|
||
)
|
||
]
|
||
|
||
for agent in agent_manager.agents.values():
|
||
tools.append(
|
||
Tool(
|
||
name=agent.name,
|
||
title=agent.name,
|
||
description=agent.description,
|
||
url=f"/agent/{agent.name}",
|
||
method="POST",
|
||
metadata=agent.config_schema.to_dict(),
|
||
)
|
||
)
|
||
|
||
return tools
|
||
|
||
@tool.post("/text-chunking")
|
||
async def text_chunking(text: str = Body(...), params: Dict[str, Any] = Body(...)):
|
||
from src.core.indexing import chunk
|
||
nodes = chunk(text, params=params)
|
||
return {"nodes": [node.to_dict() for node in nodes]}
|
||
|
||
@tool.post("/pdf2txt")
|
||
async def handle_pdf2txt(file: str = Body(...)):
|
||
from src.plugins import ocr
|
||
text = ocr.process_pdf(file)
|
||
return {"text": text}
|
||
|