Merge pull request #87 from 2501-ai/ISS-168-discord-message-and-loop-threshold

Updated join discord message and increased loop threshold

Author: zhuk-aa

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@2501-ai/cli",
-  "version": "0.0.51",
+  "version": "0.0.52",
   "description": "",
   "main": "dist/index.js",
   "bin": {

package.json

@@ -1,14 +1,15 @@
 import axios from 'axios';
 import fs from 'fs';
-import { addAgent, readConfig } from '../utils/conf';
+import { terminal } from 'terminal-kit';
+import { addAgent, readConfig, setValue } from '../utils/conf';
 
 import Logger from '../utils/logger';
 
 import { API_HOST, API_VERSION } from '../constants';
 import { isDirUnsafe } from '../helpers/security';
 import { Configuration } from '../utils/types';
 import { createAgent } from '../helpers/api';
-import { JOIN_DISCORD_MESSAGE } from '../utils/messaging';
+import { DISCORD_LINK } from '../utils/messaging';
 
 axios.defaults.baseURL = `${API_HOST}${API_VERSION}`;
 axios.defaults.timeout = 120 * 1000;
@@ -84,11 +85,26 @@ async function getWorkspacePath(options?: InitCommandOptions): Promise<string> {
 export async function initCommand(options?: InitCommandOptions) {
   try {
     const workspace = await getWorkspacePath(options);
+    const config = readConfig();
+
+    if (!config?.join_discord_shown) {
+      const term = terminal;
+
+      term('\n');
+      term.gray('🔗 Join our Discord\n');
+      term
+        .gray('│ ')
+        .gray(
+          'Connect with the 2501 team and community for updates, support, and insights:\n'
+        );
+      term.gray('│ ').gray.underline(`${DISCORD_LINK}\n`);
+
+      setValue('join_discord_shown', true);
+    }
 
     logger.start('Creating agent');
     const configKey = options?.config || 'CODING_AGENT';
     const configuration = await getConfiguration(configKey);
-    const config = readConfig();
 
     const createResponse = await createAgent(
       workspace,
@@ -108,8 +124,6 @@ export async function initCommand(options?: InitCommandOptions) {
     });
 
     logger.stop(`Agent ${createResponse.id} created`);
-
-    Logger.log(JOIN_DISCORD_MESSAGE);
   } catch (e: unknown) {
     logger.handleError(e as Error, (e as Error).message);
   }

src/commands/init.ts

@@ -12,7 +12,7 @@ import { jobSubscriptionCommand } from './commands/jobs';
 import { authMiddleware } from './middleware/auth';
 import { isLatestVersion } from './utils/versioning';
 import Logger from './utils/logger';
-import { JOIN_DISCORD_MESSAGE } from './utils/messaging';
+import { DISCORD_LINK } from './utils/messaging';
 
 process.on('SIGINT', () => {
   console.log('Process interrupted with Ctrl+C');
@@ -35,7 +35,7 @@ program
                                                   
         ---- AI Autonomous Systems ----
         
-${JOIN_DISCORD_MESSAGE}
+Join our Discord server: ${DISCORD_LINK}
   `
   )
   // eslint-disable-next-line @typescript-eslint/no-var-requires

src/index.ts

@@ -33,7 +33,12 @@ Before we begin, you need to configure your API key. (You can get your API key b
     'string'
   );
   if (res) {
-    writeConfig({ api_key: res, workspace_disabled: false, agents: [] });
+    writeConfig({
+      api_key: res,
+      workspace_disabled: false,
+      agents: [],
+      join_discord_shown: false,
+    });
 
     logger.log(
       `The CLI will now create a workspace and synchronize your files with the 2501 platform before creating your first agent.`

src/middleware/auth.ts

@@ -1,7 +1,7 @@
 import { createHash } from 'crypto';
 import { FunctionAction } from './types';
 
-const REPETITION_THRESHOLD = 5;
+const REPETITION_THRESHOLD = 15;
 
 const actionHashes = new Map<string, number>();
 

src/utils/loopDetection.ts

@@ -1,3 +1,3 @@
 export const CHUNK_MESSAGE_CLEAR = 'CHUNK_CLEAR';
 
-export const JOIN_DISCORD_MESSAGE = `Join our Discord server: \u001b[34m\u001b[4mhttps://discord.gg/uuCma4eHBF\u001b[0m`;
+export const DISCORD_LINK = `https://discord.gg/uuCma4eHBF`;

src/utils/messaging.ts

@@ -101,6 +101,7 @@ export type LocalConfig = {
   engine?: EngineType;
   stream?: boolean;
   agents: AgentConfig[];
+  join_discord_shown: boolean;
 };
 
 export type AgentCallbackType = (...args: unknown[]) => Promise<void>;