@z_ai/mcp-server 0.1.1-beta.0 → 0.1.1-beta.1

package/build/tools/data-viz.js

@@ -41,7 +41,12 @@ class DataVizAnalysisService extends BaseImageAnalysisService {
 export function registerDataVizAnalysisTool(server) {
     const service = new DataVizAnalysisService();
     const retryableAnalyze = withRetry(service.analyzeDataViz.bind(service), 2, 1000);
-    server.tool('analyze_data_visualization', 'Analyze data visualizations, charts, graphs, and dashboards to extract insights and trends.', {
+    server.tool('analyze_data_visualization', `Analyze data visualizations, charts, graphs, and dashboards to extract insights and trends.
+
+Use this tool ONLY when the user has a data visualization image and wants to understand the data patterns or metrics.
+This tool specializes in interpreting visual data representations.
+
+Do NOT use for: UI mockups, error messages, or technical architecture diagrams.`, {
         image_source: z.string().describe('Local file path or remote URL to the image'),
         prompt: z.string().describe('What insights or information you want to extract from this visualization.'),
         analysis_focus: z.string().optional().describe('Optional: specify what to focus on (e.g., \'trends\', \'anomalies\', \'comparisons\', \'performance metrics\'). Leave empty for comprehensive analysis.')

package/build/tools/diagram-analysis.js

@@ -41,7 +41,12 @@ class DiagramAnalysisService extends BaseImageAnalysisService {
 export function registerDiagramAnalysisTool(server) {
     const service = new DiagramAnalysisService();
     const retryableAnalyze = withRetry(service.analyzeDiagram.bind(service), 2, 1000);
-    server.tool('understand_technical_diagram', 'Analyze and explain technical diagrams including architecture diagrams, flowcharts, UML, ER diagrams, and system design diagrams.', {
+    server.tool('understand_technical_diagram', `Analyze and explain technical diagrams including architecture diagrams, flowcharts, UML, ER diagrams, and system design diagrams.
+
+Use this tool ONLY when the user has a technical diagram and wants to understand its structure or components.
+This tool specializes in interpreting visual technical documentation.
+
+Do NOT use for: UI screenshots, error messages, or data visualizations/charts.`, {
         image_source: z.string().describe('Local file path or remote URL to the image'),
         prompt: z.string().describe('What you want to understand or extract from this diagram.'),
         diagram_type: z.string().optional().describe('Optional: specify the diagram type if known (e.g., \'architecture\', \'flowchart\', \'uml\', \'er-diagram\', \'sequence\'). Leave empty for auto-detection.')

package/build/tools/error-diagnosis.js

@@ -41,7 +41,12 @@ class ErrorDiagnosisService extends BaseImageAnalysisService {
 export function registerErrorDiagnosisTool(server) {
     const service = new ErrorDiagnosisService();
     const retryableDiagnose = withRetry(service.diagnoseError.bind(service), 2, 1000);
-    server.tool('diagnose_error_screenshot', 'Diagnose and analyze error messages, stack traces, and exception screenshots.', {
+    server.tool('diagnose_error_screenshot', `Diagnose and analyze error messages, stack traces, and exception screenshots.
+
+Use this tool ONLY when the user has an error screenshot and needs help understanding or fixing it.
+This tool specializes in error analysis and provides actionable solutions.
+
+Do NOT use for: code extraction, UI analysis, or diagram understanding.`, {
         image_source: z.string().describe('Local file path or remote URL to the image'),
         prompt: z.string().describe('Description of what you need help with regarding this error. Include any relevant context about when it occurred.'),
         context: z.string().optional().describe('Optional: additional context about when the error occurred (e.g., \'during npm install\', \'when running the app\', \'after deployment\'). Helps with more accurate diagnosis.')

package/build/tools/general-image.js

@@ -34,7 +34,12 @@ class GeneralImageAnalysisService extends BaseImageAnalysisService {
 export function registerGeneralImageAnalysisTool(server) {
     const service = new GeneralImageAnalysisService();
     const retryableAnalyze = withRetry(service.analyzeImage.bind(service), 2, 1000);
-    server.tool('analyze_image', 'General-purpose image analysis for scenarios not covered by specialized tools.', {
+    server.tool('analyze_image', `General-purpose image analysis for scenarios not covered by specialized tools.
+
+Use this tool as a FALLBACK when none of the other specialized tools (ui_to_artifact, extract_text_from_screenshot,
+diagnose_error_screenshot, understand_technical_diagram, analyze_data_visualization, ui_diff_check) fit the user's need.
+
+This tool provides flexible image understanding for any visual content.`, {
         image_source: z.string().describe('Local file path or remote URL to the image'),
         prompt: z.string().describe('Detailed description of what you want to analyze, extract, or understand from the image. Be specific about your requirements.')
     }, async (params) => {

package/build/tools/text-extraction.js

@@ -41,7 +41,12 @@ class TextExtractionService extends BaseImageAnalysisService {
 export function registerTextExtractionTool(server) {
     const service = new TextExtractionService();
     const retryableExtract = withRetry(service.extractText.bind(service), 2, 1000);
-    server.tool('extract_text_from_screenshot', 'Extract and recognize text from screenshots using advanced OCR capabilities.', {
+    server.tool('extract_text_from_screenshot', `Extract and recognize text from screenshots using advanced OCR capabilities.
+
+Use this tool ONLY when the user has a screenshot containing text and wants to extract it.
+This tool specializes in OCR for code, terminal output, documentation, and general text extraction.
+
+Do NOT use for: UI design conversion, error diagnosis, or diagram understanding.`, {
         image_source: z.string().describe('Local file path or remote URL to the image'),
         prompt: z.string().describe('Instructions for text extraction. Specify what type of text to extract and any formatting requirements.'),
         programming_language: z.string().optional().describe('Optional: specify the programming language if the screenshot contains code (e.g., \'python\', \'javascript\', \'java\'). Leave empty for auto-detection or non-code text.')

package/build/tools/ui-diff.js

@@ -46,7 +46,12 @@ ${userPrompt}`;
 export function registerUiDiffCheckTool(server) {
     const service = new UiDiffCheckService();
     const retryableCompare = withRetry(service.compareUiScreenshots.bind(service), 2, 1000);
-    server.tool('ui_diff_check', 'Compare two UI screenshots to identify visual differences and implementation discrepancies.', {
+    server.tool('ui_diff_check', `Compare two UI screenshots to identify visual differences and implementation discrepancies.
+
+Use this tool ONLY when the user wants to compare an expected/reference UI with an actual implementation.
+This tool is specialized for UI quality assurance and design-to-implementation verification.
+
+Do NOT use for: general image comparison, error diagnosis, or analyzing single UIs.`, {
         expected_image_source: z.string().describe('Local file path or remote URL to the image'),
         actual_image_source: z.string().describe('Local file path or remote URL to the image'),
         prompt: z.string().describe('Instructions for the comparison. Specify what aspects to focus on or what level of detail is needed.')

package/build/tools/ui-to-artifact.js

@@ -42,7 +42,15 @@ class UiToArtifactService extends BaseImageAnalysisService {
 export function registerUiToArtifactTool(server) {
     const service = new UiToArtifactService();
     const retryableConvert = withRetry(service.convertUiToArtifact.bind(service), 2, 1000);
-    server.tool('ui_to_artifact', 'Convert UI screenshots into various artifacts: code, prompts, design specifications, or descriptions.', {
+    server.tool('ui_to_artifact', `Convert UI screenshots into various artifacts: code, prompts, design specifications, or descriptions.
+
+Use this tool ONLY when the user wants to:
+- Generate frontend code from UI design (output_type='code')
+- Create AI prompts for UI generation (output_type='prompt')
+- Extract design specifications (output_type='spec')
+- Get natural language description of the UI (output_type='description')
+
+Do NOT use for: screenshots containing text/code to extract, error messages, diagrams, or data visualizations.`, {
         image_source: z.string().describe('Local file path or remote URL to the image'),
         output_type: z.enum(['code', 'prompt', 'spec', 'description']).describe("Type of output to generate. Options: 'code' (generate frontend code), 'prompt' (generate AI prompt for recreating this UI), 'spec' (generate design specification document), 'description' (natural language description of the UI)."),
         prompt: z.string().describe('Detailed instructions describing what to generate from this UI image. Should clearly state the desired output and any specific requirements.')

package/build/tools/video-analysis.js

@@ -78,7 +78,16 @@ export function registerVideoAnalysisTool(server) {
     const retryableAnalyze = withRetry(analysisService.analyzeVideo.bind(analysisService), 2, // Maximum 2 retries
     1000 // 1 second delay
     );
-    server.tool('analyze_video', 'Analyze a video using advanced AI vision models with comprehensive understanding capabilities. Supports both local files and remote URL. Maximum local file size: 8MB', {
+    server.tool('analyze_video', `Analyze video content using advanced AI vision models.
+
+Use this tool when the user wants to:
+- Understand what happens in a video
+- Extract key moments or actions from video
+- Analyze video content, scenes, or sequences
+- Get descriptions of video footage
+- Identify objects, people, or activities in video
+
+Supports both local files and remote URL. Maximum file size: 8MB. Supports MP4, MOV, M4V formats.`, {
         video_source: z.string().describe('Local file path or remote URL to the video (supports MP4, MOV, M4V)'),
         prompt: z.string().describe('Detailed text prompt describing what to analyze, extract, or understand from the video')
     }, async (params) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@z_ai/mcp-server",
-  "version": "0.1.1-beta.0",
+  "version": "0.1.1-beta.1",
   "description": "MCP Server for Z.AI - A Model Context Protocol server that provides AI capabilities",
   "main": "build/index.js",
   "type": "module",