mirror of
https://github.com/hanzoai/runtime.git
synced 2026-10-30 18:23:13 +00:00
zeekay
ea80dc3638
- Add main Dockerfile for production builds - Add Dockerfile.runtime for secure code execution - Add Dockerfile.runtime.fixed with improved configuration - Include .dockerignore for build optimization - Add integration documentation and publishing guides - Update API and CLI components for better runtime support - Enhance sandbox and container management features
113 lines
3.1 KiB
Go
113 lines
3.1 KiB
Go
// Copyright 2025 Hanzo Industries Inc.
|
|
// SPDX-License-Identifier: AGPL-3.0
|
|
|
|
package tools
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"path/filepath"
|
|
|
|
"github.com/hanzoai/runtime/cli/apiclient"
|
|
"github.com/mark3labs/mcp-go/mcp"
|
|
)
|
|
|
|
type FileDownloadArgs struct {
|
|
Id *string `json:"id,omitempty"`
|
|
FilePath *string `json:"filePath,omitempty"`
|
|
}
|
|
|
|
type Content struct {
|
|
Type string `json:"type"`
|
|
Text string `json:"text,omitempty"`
|
|
Data string `json:"data,omitempty"`
|
|
}
|
|
|
|
func GetFileDownloadTool() mcp.Tool {
|
|
return mcp.NewTool("file_download",
|
|
mcp.WithDescription("Download a file from the Runtime sandbox. Returns the file content either as text or as a base64 encoded image. Handles special cases like matplotlib plots stored as JSON with embedded base64 images."),
|
|
mcp.WithString("filePath", mcp.Required(), mcp.Description("Path to the file to download.")),
|
|
mcp.WithString("id", mcp.Required(), mcp.Description("ID of the sandbox to download the file from.")),
|
|
)
|
|
}
|
|
|
|
func FileDownload(ctx context.Context, request mcp.CallToolRequest, args FileDownloadArgs) (*mcp.CallToolResult, error) {
|
|
apiClient, err := apiclient.GetApiClient(nil, runtimeMCPHeaders)
|
|
if err != nil {
|
|
return &mcp.CallToolResult{IsError: true}, err
|
|
}
|
|
|
|
if args.Id == nil || *args.Id == "" {
|
|
return &mcp.CallToolResult{IsError: true}, fmt.Errorf("sandbox ID is required")
|
|
}
|
|
|
|
if args.FilePath == nil || *args.FilePath == "" {
|
|
return &mcp.CallToolResult{IsError: true}, fmt.Errorf("filePath parameter is required")
|
|
}
|
|
|
|
// Download the file
|
|
file, _, err := apiClient.ToolboxAPI.DownloadFile(ctx, *args.Id).Path(*args.FilePath).Execute()
|
|
if err != nil {
|
|
return &mcp.CallToolResult{IsError: true}, fmt.Errorf("error downloading file: %v", err)
|
|
}
|
|
defer file.Close()
|
|
|
|
// Read file content
|
|
content, err := io.ReadAll(file)
|
|
if err != nil {
|
|
return &mcp.CallToolResult{IsError: true}, fmt.Errorf("error reading file content: %v", err)
|
|
}
|
|
|
|
// Process file content based on file type
|
|
ext := filepath.Ext(*args.FilePath)
|
|
var result []Content
|
|
|
|
switch ext {
|
|
case ".png", ".jpg", ".jpeg", ".gif":
|
|
// For image files, return as base64 encoded data
|
|
result = []Content{{
|
|
Type: "image",
|
|
Data: string(content),
|
|
}}
|
|
case ".json":
|
|
// For JSON files, try to parse and handle special cases like matplotlib plots
|
|
var jsonData map[string]interface{}
|
|
if err := json.Unmarshal(content, &jsonData); err != nil {
|
|
// If not valid JSON, return as text
|
|
result = []Content{{
|
|
Type: "text",
|
|
Text: string(content),
|
|
}}
|
|
} else {
|
|
// Check if it's a matplotlib plot
|
|
if _, ok := jsonData["data"]; ok {
|
|
result = []Content{{
|
|
Type: "image",
|
|
Data: jsonData["data"].(string),
|
|
}}
|
|
} else {
|
|
result = []Content{{
|
|
Type: "text",
|
|
Text: string(content),
|
|
}}
|
|
}
|
|
}
|
|
default:
|
|
// For all other files, return as text
|
|
result = []Content{{
|
|
Type: "text",
|
|
Text: string(content),
|
|
}}
|
|
}
|
|
|
|
// Convert result to JSON
|
|
resultJSON, err := json.Marshal(result)
|
|
if err != nil {
|
|
return &mcp.CallToolResult{IsError: true}, fmt.Errorf("error marshaling result: %v", err)
|
|
}
|
|
|
|
return mcp.NewToolResultText(string(resultJSON)), nil
|
|
}
|