From 40347293b7f42019c81a15aa2e8a1dbbb040636c Mon Sep 17 00:00:00 2001 From: Apoorv Taneja Date: Thu, 27 Feb 2025 13:21:39 +0530 Subject: [PATCH 1/2] added allow tracing in toolset --- js/src/frameworks/cloudflare.ts | 2 ++ js/src/frameworks/langchain.ts | 2 ++ js/src/frameworks/langgraph.ts | 2 ++ js/src/frameworks/openai.ts | 2 ++ js/src/frameworks/vercel.ts | 4 +++- js/src/sdk/base.toolset.ts | 4 ++++ 6 files changed, 15 insertions(+), 1 deletion(-) diff --git a/js/src/frameworks/cloudflare.ts b/js/src/frameworks/cloudflare.ts index 1a907f6ac22..17ded7b8f78 100644 --- a/js/src/frameworks/cloudflare.ts +++ b/js/src/frameworks/cloudflare.ts @@ -32,6 +32,7 @@ export class CloudflareToolSet extends BaseComposioToolSet { baseUrl?: Optional; entityId?: string; connectedAccountIds?: Record; + allowTracing?: boolean; } = {} ) { super({ @@ -40,6 +41,7 @@ export class CloudflareToolSet extends BaseComposioToolSet { runtime: null, entityId: config.entityId || CloudflareToolSet.DEFAULT_ENTITY_ID, connectedAccountIds: config.connectedAccountIds, + allowTracing: config.allowTracing || false, }); } diff --git a/js/src/frameworks/langchain.ts b/js/src/frameworks/langchain.ts index ade2c145383..b71618934be 100644 --- a/js/src/frameworks/langchain.ts +++ b/js/src/frameworks/langchain.ts @@ -24,6 +24,7 @@ export class LangchainToolSet extends BaseComposioToolSet { entityId?: string; runtime?: string; connectedAccountIds?: Record; + allowTracing?: boolean; } = {} ) { super({ @@ -32,6 +33,7 @@ export class LangchainToolSet extends BaseComposioToolSet { runtime: config?.runtime || LangchainToolSet.FRAMEWORK_NAME, entityId: config.entityId || LangchainToolSet.DEFAULT_ENTITY_ID, connectedAccountIds: config.connectedAccountIds, + allowTracing: config.allowTracing || false, }); } diff --git a/js/src/frameworks/langgraph.ts b/js/src/frameworks/langgraph.ts index 5611b607ffd..4c4c77b7a18 100644 --- a/js/src/frameworks/langgraph.ts +++ b/js/src/frameworks/langgraph.ts @@ -17,6 +17,7 @@ export class LangGraphToolSet extends BaseComposioToolSet { baseUrl?: Optional; entityId?: string; connectedAccountIds?: Record; + allowTracing?: boolean; } = {} ) { super({ @@ -25,6 +26,7 @@ export class LangGraphToolSet extends BaseComposioToolSet { entityId: config.entityId || LangGraphToolSet.DEFAULT_ENTITY_ID, runtime: LangGraphToolSet.FRAMEWORK_NAME, connectedAccountIds: config.connectedAccountIds, + allowTracing: config.allowTracing || false, }); } } diff --git a/js/src/frameworks/openai.ts b/js/src/frameworks/openai.ts index b852f70b95f..ccac7a80894 100644 --- a/js/src/frameworks/openai.ts +++ b/js/src/frameworks/openai.ts @@ -30,6 +30,7 @@ export class OpenAIToolSet extends BaseComposioToolSet { baseUrl?: Optional; entityId?: string; connectedAccountIds?: Record; + allowTracing?: boolean; } = {} ) { super({ @@ -38,6 +39,7 @@ export class OpenAIToolSet extends BaseComposioToolSet { runtime: OpenAIToolSet.FRAMEWORK_NAME, entityId: config.entityId || OpenAIToolSet.DEFAULT_ENTITY_ID, connectedAccountIds: config.connectedAccountIds, + allowTracing: config.allowTracing || false, }); } diff --git a/js/src/frameworks/vercel.ts b/js/src/frameworks/vercel.ts index d040d05efd2..3c26108f871 100644 --- a/js/src/frameworks/vercel.ts +++ b/js/src/frameworks/vercel.ts @@ -28,6 +28,7 @@ export class VercelAIToolSet extends BaseComposioToolSet { baseUrl?: Optional; entityId?: string; connectedAccountIds?: Record; + allowTracing?: boolean; } = {} ) { super({ @@ -36,6 +37,7 @@ export class VercelAIToolSet extends BaseComposioToolSet { runtime: "vercel-ai", entityId: config.entityId || "default", connectedAccountIds: config.connectedAccountIds, + allowTracing: config.allowTracing || false, }); } @@ -139,4 +141,4 @@ export class VercelAIToolSet extends BaseComposioToolSet { }) ); } -} +} \ No newline at end of file diff --git a/js/src/sdk/base.toolset.ts b/js/src/sdk/base.toolset.ts index 6b5fede4610..72e1aa211b0 100644 --- a/js/src/sdk/base.toolset.ts +++ b/js/src/sdk/base.toolset.ts @@ -79,6 +79,7 @@ export class ComposioToolSet { * @param {string|null} config.runtime - Runtime environment * @param {string} config.entityId - Entity ID for operations * @param {Record} config.connectedAccountIds - Map of app names to their connected account IDs + * @param {boolean} config.allowTracing - Whether to allow tracing for the SDK */ constructor({ apiKey, @@ -86,12 +87,14 @@ export class ComposioToolSet { runtime, entityId, connectedAccountIds, + allowTracing, }: { apiKey?: string | null; baseUrl?: string | null; runtime?: string | null; entityId?: string; connectedAccountIds?: Record; + allowTracing?: boolean; } = {}) { const clientApiKey: string | undefined = apiKey || @@ -102,6 +105,7 @@ export class ComposioToolSet { apiKey: this.apiKey, baseUrl: baseUrl || undefined, runtime: runtime as string, + allowTracing: allowTracing || false, }); this.runtime = runtime || null; From 6b851d92cf026316a98e50bb15ccee4f60d373c8 Mon Sep 17 00:00:00 2001 From: Apoorv Taneja Date: Thu, 27 Feb 2025 13:23:19 +0530 Subject: [PATCH 2/2] fixed linting --- js/src/frameworks/vercel.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/js/src/frameworks/vercel.ts b/js/src/frameworks/vercel.ts index 3c26108f871..48cb5ba12d2 100644 --- a/js/src/frameworks/vercel.ts +++ b/js/src/frameworks/vercel.ts @@ -141,4 +141,4 @@ export class VercelAIToolSet extends BaseComposioToolSet { }) ); } -} \ No newline at end of file +}