Merge pull request #8 from IQAIcom/fix-add-timeout-to-env-for-sampling-messages

add timeout to env for sampling messages

Author: Royal-lobster

.changeset/happy-suits-say.md

@@ -0,0 +1,5 @@
+---
+"@iqai/mcp-discord": patch
+---
+
+Adds timeout setting in env for sampling messages

src/config.ts

@@ -32,6 +32,14 @@ const envSchema = z.object({
     .optional()
     .default(true)
     .describe('Only respond to messages that mention the bot'),
+  SAMPLING_DEFAULT_TIMEOUT: z
+    .number()
+    .optional()
+    .describe('A timeout (in milliseconds) for this request'),
+  SAMPLING_REACTION_TIMEOUT: z
+    .number()
+    .optional()
+    .describe('A timeout for sending reactions to discord'),
   BLOCK_DMS: z
     .stringbool()
     .optional()

src/sampling.ts

@@ -187,7 +187,10 @@ export class SamplingHandler {
             maxTokens: 10,
           },
         },
-        z.any()
+        z.any(),
+        {
+          timeout: env.SAMPLING_REACTION_TIMEOUT,
+        }
       );
 
       const response =
@@ -225,7 +228,8 @@ export class SamplingHandler {
           maxTokens: 200,
         },
       },
-      z.any()
+      z.any(),
+      { timeout: env.SAMPLING_DEFAULT_TIMEOUT }
     );
   }