feat: add LlamaIndex TS framework support

ComposioHQ · Jan 10, 2025 · b7a8481 · b7a8481
1 parent f33976d
commit b7a8481
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 40 deletions.
diff --git a/js/src/frameworks/llamaindex.spec.ts b/js/src/frameworks/llamaindex.spec.ts
@@ -8,7 +8,7 @@ describe("Apps class tests", () => {
   let llamaindexToolSet: LlamaIndexToolSet;
   beforeAll(() => {
     llamaindexToolSet = new LlamaIndexToolSet({
-        apiKey: getTestConfig().COMPOSIO_API_KEY,
+      apiKey: getTestConfig().COMPOSIO_API_KEY,
       baseUrl: getTestConfig().BACKEND_HERMES_URL,
     });
   });

diff --git a/js/src/frameworks/llamaindex.ts b/js/src/frameworks/llamaindex.ts
@@ -1,4 +1,4 @@
-import { FunctionTool, OpenAI, OpenAIAgent, type JSONValue } from "llamaindex";
+import { FunctionTool, type JSONValue } from "llamaindex";
 import { z } from "zod";
 import { ComposioToolSet as BaseComposioToolSet } from "../sdk/base.toolset";
 import { COMPOSIO_BASE_URL } from "../sdk/client/core/OpenAPI";
@@ -15,7 +15,6 @@ export class LlamaIndexToolSet extends BaseComposioToolSet {
   static FRAMEWORK_NAME = "llamaindex";
   static DEFAULT_ENTITY_ID = "default";
   fileName: string = "js/src/frameworks/llamaindex.ts";
-  private llm: OpenAI;
 
   constructor(
     config: {
@@ -31,9 +30,6 @@ export class LlamaIndexToolSet extends BaseComposioToolSet {
       runtime: config?.runtime || null,
       entityId: config.entityId || LlamaIndexToolSet.DEFAULT_ENTITY_ID,
     });
-    this.llm = new OpenAI({
-      apiKey: config.apiKey || process.env.OPENAI_API_KEY,
-    });
   }
 
   private _wrapTool(
@@ -78,7 +74,6 @@ export class LlamaIndexToolSet extends BaseComposioToolSet {
     const tools = await this.getToolsSchema(filters, entityId);
     return tools.map((tool) => {
       const wrappedTool = this._wrapTool(tool, entityId || this.entityId);
-      // Add additional properties for better logging
       Object.assign(wrappedTool, {
         name: tool.name,
         description: tool.description,
@@ -87,37 +82,4 @@ export class LlamaIndexToolSet extends BaseComposioToolSet {
       return wrappedTool;
     });
   }
-
-  async executeToolCall(
-    tool: { name: string; parameters: Record<string, unknown> },
-    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,
-        params: tool.parameters,
-        entityId: entityId || this.entityId,
-      })
-    );
-  }
-
-  async createAgent(
-    systemPrompt: string,
-    tools?: FunctionTool<Record<string, unknown>, Promise<string>>[],
-    entityId: Optional<string> = null
-  ) {
-    const agentTools = tools || (await this.getTools({}, entityId));
-
-    return new OpenAIAgent({
-      llm: this.llm,
-      tools: agentTools,
-      systemPrompt,
-    });
-  }
 }