@alwaysai/device-agent 0.0.9 → 0.0.11-internal

package/src/subcommands/app/app.ts

@@ -24,6 +24,7 @@ import {
   updateModels,
 } from '../../application-control';
 import { AgentConfigFile } from '../../infrastructure/agent-config';
+import { logger } from '../../util/logger';
 
 export const listAppsCliLeaf = CliLeaf({
   name: 'list',
@@ -47,7 +48,7 @@ export const listAppReleasesCliLeaf = CliLeaf({
   async action(_, opts) {
     const { project } = opts;
     const releaseHistory = await listAppReleases({ projectId: project });
-    console.log(releaseHistory);
+    logger.info(releaseHistory);
   },
 });
 
@@ -66,7 +67,7 @@ export const listAppLatestReleaseCliLeaf = CliLeaf({
     if (latestReleaseHash === undefined) {
       throw new CliTerseError('This application has not been published yet');
     }
-    console.log(latestReleaseHash);
+    logger.info(latestReleaseHash);
   },
 });
 
@@ -107,7 +108,7 @@ export const getAppStatusCliLeaf = CliLeaf({
   async action(_, opts) {
     const { project } = opts;
     const appStatus = await getAppStatus({ projectId: project });
-    console.log(appStatus);
+    logger.info(appStatus);
   },
 });
 
@@ -143,7 +144,7 @@ export const getAppLogsCliLeaf = CliLeaf({
     const readable = await getAppLogs({ projectId: project });
     readable.setEncoding('utf8');
     for await (const chunk of readable) {
-      console.log(chunk);
+      logger.info(chunk);
     }
   },
 });
@@ -307,7 +308,7 @@ export const getAllEnvsCLiLeaf = CliLeaf({
   },
   async action(_, opts) {
     const { project } = opts;
-    console.log(await getAllEnvs({ projectId: project }));
+    logger.info(await getAllEnvs({ projectId: project }));
   },
 });
 

package/src/subcommands/device/device.ts

@@ -8,6 +8,7 @@ import {
   getTargetHardwareUuid,
   writeCertificateAndToken,
 } from '../../infrastructure/certificates-and-tokens';
+import { logger } from '../../util/logger';
 
 export const initCliLeaf = CliLeaf({
   name: 'init',
@@ -24,7 +25,7 @@ export const initCliLeaf = CliLeaf({
   },
   async action(_, opts) {
     const { name, description } = opts;
-    console.log('Initializing device');
+    logger.info('Initializing device');
     await checkUserIsLoggedInComponent({ yes: true });
     if (!(await checkPaidPlan())) {
       throw new Error(`This action only supported for Enterprise alwaysAI accounts!`);
@@ -44,7 +45,7 @@ export const initCliLeaf = CliLeaf({
     };
     const response = await addDevice(deviceToAdd, 'production');
     await writeCertificateAndToken({ deviceUuid: response.deviceUUID });
-    console.log(`Initialized device as ${response.deviceUUID}`);
+    logger.info(`Initialized device as ${response.deviceUUID}`);
   },
 });
 

package/src/subcommands/get-model-package.ts

@@ -2,6 +2,7 @@ import { CliLeaf, CliNumberInput, CliStringInput } from '@alwaysai/alwayscli';
 import { appInstallModel } from 'alwaysai/lib/core/app';
 import { CliRpcClient } from 'alwaysai/lib/infrastructure';
 import { JsSpawner } from 'alwaysai/lib/util';
+import { logger } from '../util/logger';
 
 export const getModelPackageCliLeaf = CliLeaf({
   name: 'get-model-package',
@@ -27,6 +28,6 @@ export const getModelPackageCliLeaf = CliLeaf({
     const version =
       opts.version || (await CliRpcClient().getModelVersion({ id })).version;
     await appInstallModel(spawner, id, version);
-    console.log(`Completed downloading ${id} to ${path}`);
+    logger.info(`Completed downloading ${id} to ${path}`);
   },
 });

package/src/util/logger.ts

@@ -0,0 +1,28 @@
+import * as winston from 'winston';
+import 'winston-daily-rotate-file';
+import * as path from 'path';
+import { AAI_DIR } from 'alwaysai/lib/constants';
+import { ALWAYSAI_LOG_LEVEL, ALWAYSAI_LOG_TO_CONSOLE } from '../environment';
+
+const LOG_LEVEL = ALWAYSAI_LOG_LEVEL || 'info';
+const transports = ALWAYSAI_LOG_TO_CONSOLE
+  ? [new winston.transports.Console({ level: LOG_LEVEL })]
+  : [
+      new winston.transports.DailyRotateFile({
+        filename: path.join(AAI_DIR, 'agent-logs', 'agent-logs.txt'),
+        maxSize: '5m',
+        maxFiles: '2d',
+      }),
+    ];
+
+export const logger = winston.createLogger({
+  level: LOG_LEVEL,
+  format: winston.format.combine(
+    winston.format.errors({ stack: true }),
+    winston.format.splat(),
+    winston.format.simple(),
+  ),
+  transports,
+});
+
+logger.info(`Initialized logger with log level: ${LOG_LEVEL}`);