feat: add (dummy) create tool route (#1139)
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
import uuid
|
||||
from functools import partial
|
||||
from typing import List
|
||||
from typing import List, Optional, Literal
|
||||
|
||||
from fastapi import APIRouter, Depends, Body
|
||||
from pydantic import BaseModel, Field
|
||||
@@ -17,6 +17,17 @@ class ListToolsResponse(BaseModel):
|
||||
tools: List[ToolModel] = Field(..., description="List of tools (functions).")
|
||||
|
||||
|
||||
class CreateToolRequest(BaseModel):
|
||||
name: str = Field(..., description="The name of the function.")
|
||||
source_code: str = Field(..., description="The source code of the function.")
|
||||
source_type: Optional[Literal["python"]] = Field(None, description="The type of the source code.")
|
||||
tags: Optional[List[str]] = Field(None, description="Metadata tags.")
|
||||
|
||||
|
||||
class CreateToolResponse(BaseModel):
|
||||
tool: ToolModel = Field(..., description="Information about the newly created tool.")
|
||||
|
||||
|
||||
def setup_tools_index_router(server: SyncServer, interface: QueuingInterface, password: str):
|
||||
get_current_user_with_server = partial(partial(get_current_user, server), password)
|
||||
|
||||
@@ -32,4 +43,18 @@ def setup_tools_index_router(server: SyncServer, interface: QueuingInterface, pa
|
||||
tools = server.ms.list_tools(user_id=user_id)
|
||||
return ListToolsResponse(tools=tools)
|
||||
|
||||
@router.post("/tools", tags=["tools"], response_model=ListToolsResponse)
|
||||
async def create_tool(
|
||||
request: CreateToolRequest = Body(...),
|
||||
user_id: uuid.UUID = Depends(get_current_user_with_server),
|
||||
):
|
||||
"""
|
||||
Create a new tool (dummy route)
|
||||
"""
|
||||
return ToolModel(
|
||||
name="dummy_tool",
|
||||
json_schema={},
|
||||
tags=[],
|
||||
)
|
||||
|
||||
return router
|
||||
|
||||
Reference in New Issue
Block a user