Merge branch 'master' into fix/docs

docs/mint.json

@@ -70,10 +70,6 @@
     {
       "name": "All Tools",
       "url": "https://composio.dev/tools"
-    },
-    {
-      "name": "Chat with Repo",
-      "url": "https://dub.composio.dev/composio-chat-with-repo"
     }
   ],
   "navigation": [

js/package.json

@@ -1,6 +1,6 @@
 {
   "name": "composio-core",
-  "version": "0.4.1-beta",
+  "version": "0.4.3-beta",
   "description": "",
   "main": "dist/index.js",
   "scripts": {

js/src/cli/execute.ts

@@ -0,0 +1,44 @@
+import { Command } from "commander";
+import client from "../sdk/client/client";
+import chalk from "chalk";
+import { getOpenAPIClient } from "../sdk/utils/config";
+
+export default class ExecuteCommand {
+  private program: Command;
+  constructor(program: Command) {
+    this.program = program;
+    this.program
+      .command("execute <action>")
+      .description("Execute a Composio action")
+      .option("-p, --params <params>", "Action parameters as a JSON string")
+      .action(this.handleAction.bind(this));
+  }
+  private async handleAction(
+    action: string,
+    options: {
+      params?: string;
+    }
+  ): Promise<void> {
+    getOpenAPIClient();
+    const { params } = options;
+    try {
+      const res = await client.actionsV2.executeActionV2({
+        body: params ? JSON.parse(params) : {},
+        path: {
+          actionId: action,
+        },
+      });
+      console.log(
+        chalk.green(
+          "Action executed successfully",
+          JSON.stringify(res?.data?.data, null, 2)
+        )
+      );
+    } catch (error) {
+      console.log(
+        chalk.red(`Error executing action: ${(error as Error).message}`)
+      );
+      return;
+    }
+  }
+}

js/src/cli/index.ts

@@ -13,6 +13,7 @@ import logout from "./logout";
 import triggers from "./triggers";
 import whoami from "./whoami";
 import actions from "./actions";
+import execute from "./execute";
 
 // SDK Imports
 import { TELEMETRY_EVENTS } from "../sdk/utils/telemetry/events";
@@ -30,6 +31,7 @@ new integrations(program);
 new triggers(program);
 new add(program);
 new actions(program);
+new execute(program);
 
 function formatLine(content: string): string {
   return `${content}`;

js/src/cli/triggers.ts

@@ -1,6 +1,7 @@
 /* eslint-disable no-console */
 import chalk from "chalk";
 import { Command } from "commander";
+import client from "../sdk/client/client";
 
 import { getOpenAPIClient } from "../sdk/utils/config";
 import { Composio } from "../sdk";
@@ -38,6 +39,8 @@ export default class ConnectionsCommand {
     new TriggerAdd(command);
     new TriggerDisable(command);
     new ActiveTriggers(command);
+    new TriggerEnable(command);
+    new TriggerCallback(command);
   }
 
   private async handleAction(options: {
@@ -176,6 +179,29 @@ export class TriggerDisable {
   }
 }
 
+export class TriggerEnable {
+  private program: Command;
+  constructor(program: Command) {
+    this.program = program;
+
+    this.program
+      .command("enable")
+      .description("Enable an existing trigger")
+      .argument("<triggerid>", "The trigger id")
+      .action(this.handleAction.bind(this));
+  }
+
+  async handleAction(triggerId: string): Promise<void> {
+    const composioClient = new Composio();
+    try {
+      await composioClient.triggers.enable({ triggerId });
+      console.log(chalk.green(`Trigger ${triggerId} enabled`));
+    } catch (error) {
+      console.log(chalk.red(`Error enabling trigger ${triggerId}: ${error}`));
+    }
+  }
+}
+
 export class ActiveTriggers {
   private program: Command;
   constructor(program: Command, register: boolean = true) {
@@ -203,3 +229,57 @@ export class ActiveTriggers {
     }
   }
 }
+
+export class TriggerCallback {
+  private program: Command;
+  constructor(program: Command) {
+    this.program = program;
+
+    const callbackCommand = this.program
+      .command("callback")
+      .description("Manage trigger callback URLs");
+
+    callbackCommand
+      .command("set")
+      .description("Set a callback URL for a trigger")
+      .argument("<callbackURL>", "Callback URL that needs to be set")
+      .action(this.handleSetAction.bind(this));
+
+    callbackCommand
+      .command("get")
+      .description("Get the current callback URL for a trigger")
+      .action(this.handleGetAction.bind(this));
+  }
+
+  async handleSetAction(callbackURL: string): Promise<void> {
+    getOpenAPIClient();
+    try {
+      await client.triggers.setCallbackUrl({
+        body: {
+          callbackURL: callbackURL,
+        },
+      });
+      console.log(chalk.green(`Callback URL set to ${callbackURL}`));
+    } catch (error) {
+      console.log(
+        chalk.red(
+          `Error setting callback URL to ${callbackURL}: ${(error as Error).message}`
+        )
+      );
+    }
+  }
+
+  async handleGetAction(): Promise<void> {
+    getOpenAPIClient();
+    try {
+      const res = await client.triggers.getWebhookUrl();
+      console.log(
+        chalk.green(`Current callback URL is ${res?.data?.callbackURL}`)
+      );
+    } catch (error) {
+      console.log(
+        chalk.red(`Error getting callback URL: ${(error as Error).message}`)
+      );
+    }
+  }
+}

js/src/constants.js

@@ -8,7 +8,7 @@ const ACTIONS = {
   // actions list end here
 };
 
-const COMPOSIO_VERSION = `0.4.1-beta`;
+const COMPOSIO_VERSION = `0.4.3-beta`;
 
 module.exports = {
   APPS,

js/src/frameworks/cloudflare.ts

@@ -7,7 +7,8 @@ import {
 import { COMPOSIO_BASE_URL } from "../sdk/client/core/OpenAPI";
 import { WorkspaceConfig } from "../env/config";
 import { Workspace } from "../env";
-import { ActionsListResponseDTO } from "../sdk/client";
+import { TELEMETRY_LOGGER } from "../sdk/utils/telemetry";
+import { TELEMETRY_EVENTS } from "../sdk/utils/telemetry/events";
 
 // Type definitions
 type Optional<T> = T | null;
@@ -21,6 +22,7 @@ export class CloudflareToolSet extends BaseComposioToolSet {
   // Class constants
   static FRAMEWORK_NAME = "cloudflare";
   static DEFAULT_ENTITY_ID = "default";
+  fileName: string = "js/src/frameworks/cloudflare.ts";
 
   /**
    * Initialize a new CloudflareToolSet instance
@@ -58,6 +60,11 @@ export class CloudflareToolSet extends BaseComposioToolSet {
     usecaseLimit?: Optional<number>;
     filterByAvailableApps?: Optional<boolean>;
   }): Promise<Sequence<AiTextGenerationToolInput>> {
+    TELEMETRY_LOGGER.manualTelemetry(TELEMETRY_EVENTS.SDK_METHOD_INVOKED, {
+      method: "getTools",
+      file: this.fileName,
+      params: filters,
+    });
     const actions = await this.getToolsSchema(filters);
     return (
       actions.map((action) => {
@@ -99,6 +106,11 @@ export class CloudflareToolSet extends BaseComposioToolSet {
     },
     entityId: Optional<string> = null
   ): Promise<string> {
+    TELEMETRY_LOGGER.manualTelemetry(TELEMETRY_EVENTS.SDK_METHOD_INVOKED, {
+      method: "executeToolCall",
+      file: this.fileName,
+      params: { tool, entityId },
+    });
     return JSON.stringify(
       await this.executeAction({
         action: tool.name,
@@ -122,6 +134,11 @@ export class CloudflareToolSet extends BaseComposioToolSet {
     result: AiTextGenerationOutput,
     entityId: Optional<string> = null
   ): Promise<Sequence<string>> {
+    TELEMETRY_LOGGER.manualTelemetry(TELEMETRY_EVENTS.SDK_METHOD_INVOKED, {
+      method: "handleToolCall",
+      file: this.fileName,
+      params: { result, entityId },
+    });
     const outputs = [];
     if ("tool_calls" in result && Array.isArray(result.tool_calls)) {
       for (const tool_call of result.tool_calls) {

js/src/frameworks/langchain.ts

@@ -5,6 +5,8 @@ import { COMPOSIO_BASE_URL } from "../sdk/client/core/OpenAPI";
 import type { Optional, Dict, Sequence } from "../sdk/types";
 import { WorkspaceConfig } from "../env/config";
 import { Workspace } from "../env";
+import { TELEMETRY_EVENTS } from "../sdk/utils/telemetry/events";
+import { TELEMETRY_LOGGER } from "../sdk/utils/telemetry";
 
 export class LangchainToolSet extends BaseComposioToolSet {
   /**
@@ -13,6 +15,7 @@ export class LangchainToolSet extends BaseComposioToolSet {
    */
   static FRAMEWORK_NAME = "langchain";
   static DEFAULT_ENTITY_ID = "default";
+  fileName: string = "js/src/frameworks/langchain.ts";
 
   constructor(
     config: {
@@ -72,6 +75,12 @@ export class LangchainToolSet extends BaseComposioToolSet {
     },
     entityId: Optional<string> = null
   ): Promise<Sequence<DynamicStructuredTool>> {
+    TELEMETRY_LOGGER.manualTelemetry(TELEMETRY_EVENTS.SDK_METHOD_INVOKED, {
+      method: "getTools",
+      file: this.fileName,
+      params: { filters, entityId },
+    });
+
     const tools = await this.getToolsSchema(filters, entityId);
     return tools.map((tool) => this._wrapTool(tool, entityId || this.entityId));
   }

js/src/frameworks/openai.ts

@@ -7,6 +7,8 @@ import { Workspace } from "../env";
 import logger from "../utils/logger";
 import { ActionsListResponseDTO } from "../sdk/client";
 import { Stream } from "openai/streaming";
+import { TELEMETRY_LOGGER } from "../sdk/utils/telemetry";
+import { TELEMETRY_EVENTS } from "../sdk/utils/telemetry/events";
 
 type Optional<T> = T | null;
 type Sequence<T> = Array<T>;
@@ -15,6 +17,8 @@ export class OpenAIToolSet extends BaseComposioToolSet {
   static FRAMEWORK_NAME = "openai";
   static DEFAULT_ENTITY_ID = "default";
 
+  fileName: string = "js/src/frameworks/openai.ts";
+
   /**
    * Composio toolset for OpenAI framework.
    *
@@ -51,6 +55,12 @@ export class OpenAIToolSet extends BaseComposioToolSet {
     },
     entityId?: Optional<string>
   ): Promise<Sequence<OpenAI.ChatCompletionTool>> {
+    TELEMETRY_LOGGER.manualTelemetry(TELEMETRY_EVENTS.SDK_METHOD_INVOKED, {
+      method: "getTools",
+      file: this.fileName,
+      params: filters,
+    });
+
     const mainActions = await this.getToolsSchema(filters, entityId);
     return (
       mainActions.map(
@@ -74,6 +84,11 @@ export class OpenAIToolSet extends BaseComposioToolSet {
     tool: OpenAI.ChatCompletionMessageToolCall,
     entityId: Optional<string> = null
   ): Promise<string> {
+    TELEMETRY_LOGGER.manualTelemetry(TELEMETRY_EVENTS.SDK_METHOD_INVOKED, {
+      method: "executeToolCall",
+      file: this.fileName,
+      params: { tool, entityId },
+    });
     return JSON.stringify(
       await this.executeAction({
         action: tool.function.name,
@@ -87,6 +102,11 @@ export class OpenAIToolSet extends BaseComposioToolSet {
     chatCompletion: OpenAI.ChatCompletion,
     entityId: Optional<string> = null
   ): Promise<Sequence<string>> {
+    TELEMETRY_LOGGER.manualTelemetry(TELEMETRY_EVENTS.SDK_METHOD_INVOKED, {
+      method: "handleToolCall",
+      file: this.fileName,
+      params: { chatCompletion, entityId },
+    });
     const outputs = [];
     for (const message of chatCompletion.choices) {
       if (message.message.tool_calls) {
@@ -104,6 +124,11 @@ export class OpenAIToolSet extends BaseComposioToolSet {
   ): Promise<
     Array<OpenAI.Beta.Threads.Runs.RunSubmitToolOutputsParams.ToolOutput>
   > {
+    TELEMETRY_LOGGER.manualTelemetry(TELEMETRY_EVENTS.SDK_METHOD_INVOKED, {
+      method: "handleAssistantMessage",
+      file: this.fileName,
+      params: { run, entityId },
+    });
     const tool_calls =
       run.required_action?.submit_tool_outputs?.tool_calls || [];
     const tool_outputs: Array<OpenAI.Beta.Threads.Runs.RunSubmitToolOutputsParams.ToolOutput> =
@@ -134,6 +159,12 @@ export class OpenAIToolSet extends BaseComposioToolSet {
     thread: OpenAI.Beta.Threads.Thread,
     entityId: string | null = null
   ): AsyncGenerator<any, void, unknown> {
+    TELEMETRY_LOGGER.manualTelemetry(TELEMETRY_EVENTS.SDK_METHOD_INVOKED, {
+      method: "waitAndHandleAssistantStreamToolCalls",
+      file: this.fileName,
+      params: { client, runStream, thread, entityId },
+    });
+
     let runId = null;
 
     // Start processing the runStream events
@@ -213,6 +244,12 @@ export class OpenAIToolSet extends BaseComposioToolSet {
     thread: OpenAI.Beta.Threads.Thread,
     entityId: Optional<string> = null
   ): Promise<OpenAI.Beta.Threads.Run> {
+    TELEMETRY_LOGGER.manualTelemetry(TELEMETRY_EVENTS.SDK_METHOD_INVOKED, {
+      method: "waitAndHandleAssistantToolCalls",
+      file: this.fileName,
+      params: { client, run, thread, entityId },
+    });
+
     while (["queued", "in_progress", "requires_action"].includes(run.status)) {
       logger.debug(`Current run status: ${run.status}`);
       const tool_outputs = await this.handleAssistantMessage(