Upload tool_schemas.py

agent/schemas/tool_schemas.py

@@ -1,6 +1,6 @@
 """
-NeuraPrompt Agent — Tool Input Schemas (v7.6)
-Added GitHub MCP connector schemas
+NeuraPrompt Agent — Tool Input Schemas (v9.0)
+Added: document_tools, media_tools schemas
 """
 
 from pydantic import BaseModel, Field
@@ -8,7 +8,6 @@ from typing import Optional, List, Dict, Any
 
 
 class BaseToolInput(BaseModel):
-    """Base class for all tool inputs"""
     pass
 
 
@@ -26,7 +25,7 @@ class CreateFileInput(BaseToolInput):
     filename: str = Field(..., description="Name of the file to create")
     content: str = Field(..., description="Content of the file")
     file_type: str = Field(default="text", description="Type of file")
-    extra_files: Optional[List[Dict[str, str]]] = Field(default=None, description="Additional files to create")
+    extra_files: Optional[List[Dict[str, str]]] = Field(default=None)
 
 
 class WebSearchInput(BaseToolInput):
@@ -39,95 +38,175 @@ class FetchUrlInput(BaseToolInput):
 
 class AnalyzeImageInput(BaseToolInput):
     image_b64: str = Field(..., description="Base64 encoded image")
-    prompt: str = Field(default="Describe this image in detail", description="Analysis prompt")
+    prompt: str = Field(default="Describe this image in detail")
 
 
 class FinishInput(BaseToolInput):
     final_answer: str = Field(..., description="Final response to the user")
 
 
+# ==================== DOCUMENT TOOL SCHEMAS ====================
+
+class CreatePDFInput(BaseToolInput):
+    title: str = Field(..., description="PDF document title")
+    content: str = Field(..., description="Body text content (use \\n for new lines)")
+    filename: Optional[str] = Field(default=None, description="Optional output filename")
+
+
+class CreateDOCXInput(BaseToolInput):
+    title: str = Field(..., description="Document title")
+    content: str = Field(..., description="Body text content (use \\n for new lines)")
+    filename: Optional[str] = Field(default=None, description="Optional output filename")
+    sections: Optional[List[Dict[str, str]]] = Field(
+        default=None,
+        description="Optional structured sections: [{heading: str, text: str}]"
+    )
+
+
+class CreateExcelInput(BaseToolInput):
+    sheet_name: str = Field(..., description="Name of the worksheet")
+    headers: List[str] = Field(..., description="Column header names")
+    rows: List[List[Any]] = Field(..., description="List of rows, each row is a list of values")
+    filename: Optional[str] = Field(default=None, description="Optional output filename")
+
+
+class CreateCSVInput(BaseToolInput):
+    headers: List[str] = Field(..., description="Column header names")
+    rows: List[List[Any]] = Field(..., description="List of data rows")
+    filename: Optional[str] = Field(default=None, description="Optional output filename")
+
+
+class CreateTextFileInput(BaseToolInput):
+    content: str = Field(..., description="File content")
+    filename: Optional[str] = Field(default=None, description="Optional output filename")
+    file_format: str = Field(default="txt", description="File format: txt or md")
+
+
+# ==================== MEDIA TOOL SCHEMAS ====================
+
+class CreateBarChartInput(BaseToolInput):
+    labels: List[str] = Field(..., description="Bar labels")
+    values: List[float] = Field(..., description="Bar values")
+    title: str = Field(default="Chart", description="Chart title")
+    x_label: str = Field(default="", description="X axis label")
+    y_label: str = Field(default="", description="Y axis label")
+    filename: Optional[str] = Field(default=None, description="Optional output filename")
+
+
+class CreateLineChartInput(BaseToolInput):
+    x_values: List[Any] = Field(..., description="X axis values")
+    y_values: List[float] = Field(..., description="Y axis values")
+    title: str = Field(default="Line Chart", description="Chart title")
+    x_label: str = Field(default="", description="X axis label")
+    y_label: str = Field(default="", description="Y axis label")
+    filename: Optional[str] = Field(default=None, description="Optional output filename")
+
+
+class CreatePieChartInput(BaseToolInput):
+    labels: List[str] = Field(..., description="Slice labels")
+    values: List[float] = Field(..., description="Slice values")
+    title: str = Field(default="Pie Chart", description="Chart title")
+    filename: Optional[str] = Field(default=None, description="Optional output filename")
+
+
+class CreateQRCodeInput(BaseToolInput):
+    data: str = Field(..., description="Text or URL to encode in the QR code")
+    filename: Optional[str] = Field(default=None, description="Optional output filename")
+
+
+class ResizeImageInput(BaseToolInput):
+    image_b64: str = Field(..., description="Base64 encoded image")
+    width: int = Field(..., description="Target width in pixels")
+    height: int = Field(..., description="Target height in pixels")
+    filename: Optional[str] = Field(default=None, description="Optional output filename")
+
+
+class CreateZipInput(BaseToolInput):
+    file_paths: List[str] = Field(..., description="List of filenames in agent_outputs folder to zip")
+    zip_filename: Optional[str] = Field(default=None, description="Optional output zip filename")
+
+
 # ==================== GITHUB MCP SCHEMAS ====================
 
 class GitHubListReposInput(BaseToolInput):
     user_id: str = Field(..., description="User ID of the connected GitHub account")
-    type_filter: str = Field(default="owner", description="Filter: all, owner, member, public, private")
-    sort: str = Field(default="updated", description="Sort by: created, updated, pushed, full_name")
-    per_page: int = Field(default=10, description="Number of repos to return (max 100)")
+    type_filter: str = Field(default="owner")
+    sort: str = Field(default="updated")
+    per_page: int = Field(default=10)
 
 
 class GitHubCreateRepoInput(BaseToolInput):
     user_id: str = Field(..., description="User ID of the connected GitHub account")
     name: str = Field(..., description="Repository name")
-    description: str = Field(default="", description="Repository description")
-    private: bool = Field(default=False, description="Create as private repository")
-    auto_init: bool = Field(default=True, description="Initialize with README")
+    description: str = Field(default="")
+    private: bool = Field(default=False)
+    auto_init: bool = Field(default=True)
 
 
 class GitHubGetRepoInput(BaseToolInput):
-    user_id: str = Field(..., description="User ID of the connected GitHub account")
-    owner: str = Field(..., description="Repository owner (username or org)")
-    repo: str = Field(..., description="Repository name")
+    user_id: str = Field(...)
+    owner: str = Field(...)
+    repo: str = Field(...)
 
 
 class GitHubListIssuesInput(BaseToolInput):
-    user_id: str = Field(..., description="User ID of the connected GitHub account")
-    owner: str = Field(..., description="Repository owner")
-    repo: str = Field(..., description="Repository name")
-    state: str = Field(default="open", description="Issue state: open, closed, all")
-    per_page: int = Field(default=10, description="Number of issues to return")
+    user_id: str = Field(...)
+    owner: str = Field(...)
+    repo: str = Field(...)
+    state: str = Field(default="open")
+    per_page: int = Field(default=10)
 
 
 class GitHubCreateIssueInput(BaseToolInput):
-    user_id: str = Field(..., description="User ID of the connected GitHub account")
-    owner: str = Field(..., description="Repository owner")
-    repo: str = Field(..., description="Repository name")
-    title: str = Field(..., description="Issue title")
-    body: str = Field(default="", description="Issue body/description")
-    labels: Optional[List[str]] = Field(default=None, description="List of label names")
+    user_id: str = Field(...)
+    owner: str = Field(...)
+    repo: str = Field(...)
+    title: str = Field(...)
+    body: str = Field(default="")
+    labels: Optional[List[str]] = Field(default=None)
 
 
 class GitHubReadFileInput(BaseToolInput):
-    user_id: str = Field(..., description="User ID of the connected GitHub account")
-    owner: str = Field(..., description="Repository owner")
-    repo: str = Field(..., description="Repository name")
-    path: str = Field(..., description="File path within the repo (e.g. 'src/main.py')")
-    ref: str = Field(default="main", description="Branch, tag, or commit SHA")
+    user_id: str = Field(...)
+    owner: str = Field(...)
+    repo: str = Field(...)
+    path: str = Field(...)
+    ref: str = Field(default="main")
 
 
 class GitHubWriteFileInput(BaseToolInput):
-    user_id: str = Field(..., description="User ID of the connected GitHub account")
-    owner: str = Field(..., description="Repository owner")
-    repo: str = Field(..., description="Repository name")
-    path: str = Field(..., description="File path within the repo")
-    content: str = Field(..., description="File content to write")
-    message: str = Field(..., description="Git commit message")
-    branch: str = Field(default="main", description="Branch to commit to")
+    user_id: str = Field(...)
+    owner: str = Field(...)
+    repo: str = Field(...)
+    path: str = Field(...)
+    content: str = Field(...)
+    message: str = Field(...)
+    branch: str = Field(default="main")
 
 
 class GitHubCreateBranchInput(BaseToolInput):
-    user_id: str = Field(..., description="User ID of the connected GitHub account")
-    owner: str = Field(..., description="Repository owner")
-    repo: str = Field(..., description="Repository name")
-    branch: str = Field(..., description="New branch name")
-    from_branch: str = Field(default="main", description="Source branch to branch from")
+    user_id: str = Field(...)
+    owner: str = Field(...)
+    repo: str = Field(...)
+    branch: str = Field(...)
+    from_branch: str = Field(default="main")
 
 
 class GitHubCreatePullRequestInput(BaseToolInput):
-    user_id: str = Field(..., description="User ID of the connected GitHub account")
-    owner: str = Field(..., description="Repository owner")
-    repo: str = Field(..., description="Repository name")
-    title: str = Field(..., description="PR title")
-    head: str = Field(..., description="Branch containing changes")
-    base: str = Field(..., description="Branch to merge into")
-    body: str = Field(default="", description="PR description/body")
+    user_id: str = Field(...)
+    owner: str = Field(...)
+    repo: str = Field(...)
+    title: str = Field(...)
+    head: str = Field(...)
+    base: str = Field(...)
+    body: str = Field(default="")
 
 
 class GitHubSearchCodeInput(BaseToolInput):
-    user_id: str = Field(..., description="User ID of the connected GitHub account")
-    query: str = Field(..., description="Search query (e.g. 'language:python flask')")
-    per_page: int = Field(default=10, description="Number of results to return")
+    user_id: str = Field(...)
+    query: str = Field(...)
+    per_page: int = Field(default=10)
 
 
 class GitHubGetUserProfileInput(BaseToolInput):
-    user_id: str = Field(..., description="User ID of the connected GitHub account")
-    
+    user_id: str = Field(...)