```diff
--- test_files/510-original.txt	2025-03-07 19:07:05
+++ test_files/510-modified.txt	2025-03-07 19:07:05
@@ -1,8 +1,6 @@
-import { LangchainToolSet as BaseComposioToolSet } from "./langchain";
 import { COMPOSIO_BASE_URL } from "../sdk/client/core/OpenAPI";
-import type { Optional } from "../sdk/types";
-import { WorkspaceConfig } from "../env/config";
-import { Workspace } from "../env";
+import type { Optional } from "../types/util";
+import { LangchainToolSet as BaseComposioToolSet } from "./langchain";
 
 export class LangGraphToolSet extends BaseComposioToolSet {
   /**
@@ -18,15 +16,17 @@
       apiKey?: Optional<string>;
       baseUrl?: Optional<string>;
       entityId?: string;
-      workspaceConfig?: WorkspaceConfig;
+      connectedAccountIds?: Record<string, string>;
+      allowTracing?: boolean;
     } = {}
   ) {
     super({
       apiKey: config.apiKey || null,
       baseUrl: config.baseUrl || COMPOSIO_BASE_URL,
       entityId: config.entityId || LangGraphToolSet.DEFAULT_ENTITY_ID,
-      workspaceConfig: config.workspaceConfig || Workspace.Host(),
       runtime: LangGraphToolSet.FRAMEWORK_NAME,
+      connectedAccountIds: config.connectedAccountIds,
+      allowTracing: config.allowTracing || false,
     });
   }
 }
```
