feat: update server to support response modes and domain filtering

Author: ricklamers

package.json

@@ -1,6 +1,6 @@
 {
   "name": "groq-compound-mcp-server",
-  "version": "0.1.5",
+  "version": "0.1.7",
   "description": "MCP server for interacting with Groq models",
   "main": "build/index.js",
   "type": "module",

src/server.ts

@@ -3,22 +3,27 @@ import { z } from "zod";
 import Groq from "groq-sdk";
 
 const ModelEnum = z.enum(["compound-beta", "compound-beta-mini"]);
+const ModeEnum = z.enum(["minimal", "verbose"]);
 
 // Define Tool Schemas
 export const realtimeToolArgsSchema = z.object({
   question: z.string().describe("The question to ask the model, especially if it requires real-time information (e.g., current news, recent events)."),
   model: ModelEnum.optional()
     .default("compound-beta")
     .describe("The model to use (compound-beta or compound-beta-mini). Defaults to compound-beta. Use compound-beta-mini for quick answers."),
-  verbose: z.boolean().optional().default(false).describe("Optional (default: false). If true, includes executed tools in the response. This is very verbose and should only be used when the user asks for it or when the user query cannot be answered without it (always first try without it)."),
+  mode: ModeEnum.optional().default("minimal").describe("Response mode ('minimal' or 'verbose'). Defaults to 'minimal'. 'verbose' includes executed tools in the response. This is very verbose and should only be used when the user asks for it or when the user query cannot be answered without it (always first try without it)."),
+  include_domains: z.array(z.string()).optional().describe("List of domains to specifically include in the search."),
+  exclude_domains: z.array(z.string()).optional().describe("List of domains to exclude from the search."),
 });
 
 export const replToolArgsSchema = z.object({
     question: z.string().describe("The question to ask the model, especially one that benefits from Python REPL interaction (e.g., for intermediate calculations or code execution)."),
     model: ModelEnum.optional()
       .default("compound-beta")
       .describe("The model to use (compound-beta or compound-beta-mini). Defaults to compound-beta. Use compound-beta-mini for quick answers."),
-    verbose: z.boolean().optional().default(false).describe("Optional (default: false). If true, includes executed tools in the response. This is very verbose and should only be used when the user asks for it or when the user query cannot be answered without it (always first try without it)."),
+    mode: ModeEnum.optional().default("minimal").describe("Response mode ('minimal' or 'verbose'). Defaults to 'minimal'. 'verbose' includes executed tools in the response. This is very verbose and should only be used when the user asks for it or when the user query cannot be answered without it (always first try without it)."),
+    include_domains: z.array(z.string()).optional().describe("List of domains to specifically include in the search."),
+    exclude_domains: z.array(z.string()).optional().describe("List of domains to exclude from the search."),
   });
 
 // Type alias for the arguments
@@ -36,15 +41,18 @@ async function executeGroqQuery(args: ToolArgs) {
           content: args.question,
         },
       ],
-      model: args.model
+      model: args.model,
+      // Add include_domains and exclude_domains if they exist in args
+      ...(args.include_domains && { include_domains: args.include_domains }),
+      ...(args.exclude_domains && { exclude_domains: args.exclude_domains }),
     });
 
     const choice = chatCompletion.choices[0]?.message;
     const responseTextContent = choice?.content || "No response from model.";
     let finalResponseText = responseTextContent;
 
     // Check if verbose flag is true
-    if ('verbose' in args && args.verbose) {
+    if ('mode' in args && args.mode === 'verbose') {
       // Use the logic to get executed_tools
       const executedTools = (choice as any)?.executed_tools ?? null;
       const responsePayload = {