@distri/core 0.1.8 → 0.2.1

package/dist/index.js

@@ -17,23 +17,25 @@ var __copyProps = (to, from, except, desc) => {
   return to;
 };
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var __publicField = (obj, key, value) => {
-  __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
-  return value;
-};
+var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
 
 // src/index.ts
-var src_exports = {};
-__export(src_exports, {
-  A2AClient: () => A2AClient,
-  A2AProtocolError: () => A2AProtocolError,
-  ApiError: () => ApiError,
-  ConnectionError: () => ConnectionError,
+var index_exports = {};
+__export(index_exports, {
+  Agent: () => Agent,
   DistriClient: () => DistriClient,
-  DistriError: () => DistriError,
+  convertA2AMessageToDistri: () => convertA2AMessageToDistri,
+  convertA2APartToDistri: () => convertA2APartToDistri,
+  convertDistriMessageToA2A: () => convertDistriMessageToA2A,
+  convertDistriPartToA2A: () => convertDistriPartToA2A,
+  extractTextFromDistriMessage: () => extractTextFromDistriMessage,
+  extractToolCallsFromDistriMessage: () => extractToolCallsFromDistriMessage,
+  extractToolResultsFromDistriMessage: () => extractToolResultsFromDistriMessage,
+  isDistriEvent: () => isDistriEvent,
+  isDistriMessage: () => isDistriMessage,
   uuidv4: () => uuidv4
 });
-module.exports = __toCommonJS(src_exports);
+module.exports = __toCommonJS(index_exports);
 
 // ../../node_modules/.pnpm/@a2a-js+sdk@https+++codeload.github.com+v3g42+a2a-js+tar.gz+51444c9/node_modules/@a2a-js/sdk/dist/chunk-CUGIRVQB.js
 var A2AClient = class {
@@ -152,8 +154,7 @@ var A2AClient = class {
           throw new Error(`HTTP error for ${method}! Status: ${httpResponse.status} ${httpResponse.statusText}. Response: ${errorBodyText}`);
         }
       } catch (e) {
-        if (e.message.startsWith("RPC error for") || e.message.startsWith("HTTP error for"))
-          throw e;
+        if (e.message.startsWith("RPC error for") || e.message.startsWith("HTTP error for")) throw e;
         throw new Error(`HTTP error for ${method}! Status: ${httpResponse.status} ${httpResponse.statusText}. Response: ${errorBodyText}`);
       }
     }
@@ -215,8 +216,7 @@ var A2AClient = class {
           throw new Error(`HTTP error establishing stream for message/stream: ${response.status} ${response.statusText}. RPC Error: ${errorJson.error.message} (Code: ${errorJson.error.code})`);
         }
       } catch (e) {
-        if (e.message.startsWith("HTTP error establishing stream"))
-          throw e;
+        if (e.message.startsWith("HTTP error establishing stream")) throw e;
         throw new Error(`HTTP error establishing stream for message/stream: ${response.status} ${response.statusText}. Response: ${errorBody || "(empty)"}`);
       }
       throw new Error(`HTTP error establishing stream for message/stream: ${response.status} ${response.statusText}`);
@@ -307,8 +307,7 @@ var A2AClient = class {
           throw new Error(`HTTP error establishing stream for tasks/resubscribe: ${response.status} ${response.statusText}. RPC Error: ${errorJson.error.message} (Code: ${errorJson.error.code})`);
         }
       } catch (e) {
-        if (e.message.startsWith("HTTP error establishing stream"))
-          throw e;
+        if (e.message.startsWith("HTTP error establishing stream")) throw e;
         throw new Error(`HTTP error establishing stream for tasks/resubscribe: ${response.status} ${response.statusText}. Response: ${errorBody || "(empty)"}`);
       }
       throw new Error(`HTTP error establishing stream for tasks/resubscribe: ${response.status} ${response.statusText}`);
@@ -430,12 +429,113 @@ var ApiError = class extends DistriError {
     this.name = "ApiError";
   }
 };
-var ConnectionError = class extends DistriError {
-  constructor(message, details) {
-    super(message, "CONNECTION_ERROR", details);
-    this.name = "ConnectionError";
+function isDistriMessage(event) {
+  return "id" in event && "role" in event && "parts" in event;
+}
+function isDistriEvent(event) {
+  return "type" in event && "metadata" in event;
+}
+
+// src/encoder.ts
+function convertA2AMessageToDistri(a2aMessage) {
+  const role = a2aMessage.role === "agent" ? "assistant" : "user";
+  return {
+    id: a2aMessage.messageId,
+    role,
+    parts: a2aMessage.parts.map(convertA2APartToDistri),
+    created_at: a2aMessage.createdAt
+  };
+}
+function convertA2APartToDistri(a2aPart) {
+  switch (a2aPart.kind) {
+    case "text":
+      return { type: "text", text: a2aPart.text };
+    case "file":
+      if ("uri" in a2aPart.file) {
+        return { type: "image_url", image: { mime_type: a2aPart.file.mimeType, url: a2aPart.file.uri } };
+      } else {
+        return { type: "image_bytes", image: { mime_type: a2aPart.file.mimeType, data: a2aPart.file.bytes } };
+      }
+    case "data":
+      switch (a2aPart.data.part_type) {
+        case "tool_call":
+          return { type: "tool_call", tool_call: a2aPart.data };
+        case "tool_result":
+          return { type: "tool_result", tool_result: a2aPart.data };
+        case "code_observation":
+          return { type: "code_observation", thought: a2aPart.data.thought, code: a2aPart.data.code };
+        case "plan":
+          return { type: "plan", plan: a2aPart.data.plan };
+        default:
+          return { type: "data", data: a2aPart.data };
+      }
+    default:
+      return { type: "text", text: JSON.stringify(a2aPart) };
   }
-};
+}
+function convertDistriMessageToA2A(distriMessage, context) {
+  let role;
+  switch (distriMessage.role) {
+    case "assistant":
+      role = "agent";
+      break;
+    case "user":
+      role = "user";
+      break;
+    case "system":
+    case "tool":
+      role = "user";
+      break;
+    default:
+      role = "user";
+  }
+  return {
+    messageId: distriMessage.id,
+    role,
+    parts: distriMessage.parts.map(convertDistriPartToA2A),
+    kind: "message",
+    contextId: context.thread_id,
+    taskId: context.run_id
+  };
+}
+function convertDistriPartToA2A(distriPart) {
+  switch (distriPart.type) {
+    case "text":
+      return { kind: "text", text: distriPart.text };
+    case "image_url":
+      return { kind: "file", file: { mimeType: distriPart.image.mime_type, uri: distriPart.image.url } };
+    case "image_bytes":
+      return { kind: "file", file: { mimeType: distriPart.image.mime_type, bytes: distriPart.image.data } };
+    case "tool_call":
+      return { kind: "data", data: { part_type: "tool_call", tool_call: distriPart.tool_call } };
+    case "tool_result":
+      let val = {
+        kind: "data",
+        data: {
+          tool_call_id: distriPart.tool_result.tool_call_id,
+          result: distriPart.tool_result.result,
+          part_type: "tool_result"
+        }
+      };
+      console.log("<> val", val);
+      return val;
+    case "code_observation":
+      return { kind: "data", data: { ...distriPart, part_type: "code_observation" } };
+    case "plan":
+      return { kind: "data", data: { ...distriPart, part_type: "plan" } };
+    case "data":
+      return { kind: "data", ...distriPart.data };
+  }
+}
+function extractTextFromDistriMessage(message) {
+  return message.parts.filter((part) => part.type === "text").map((part) => part.text).join("\n");
+}
+function extractToolCallsFromDistriMessage(message) {
+  return message.parts.filter((part) => part.type === "tool_call").map((part) => part.tool_call);
+}
+function extractToolResultsFromDistriMessage(message) {
+  return message.parts.filter((part) => part.type === "tool_result").map((part) => part.tool_result);
+}
 
 // src/distri-client.ts
 var DistriClient = class {
@@ -474,8 +574,7 @@ var DistriClient = class {
       });
       return agents;
     } catch (error) {
-      if (error instanceof ApiError)
-        throw error;
+      if (error instanceof ApiError) throw error;
       throw new DistriError("Failed to fetch agents", "FETCH_ERROR", error);
     }
   }
@@ -501,8 +600,7 @@ var DistriClient = class {
       }
       return agent;
     } catch (error) {
-      if (error instanceof ApiError)
-        throw error;
+      if (error instanceof ApiError) throw error;
       throw new DistriError(`Failed to fetch agent ${agentId}`, "FETCH_ERROR", error);
     }
   }
@@ -536,8 +634,7 @@ var DistriClient = class {
       }
       throw new DistriError("Invalid response format", "INVALID_RESPONSE");
     } catch (error) {
-      if (error instanceof A2AProtocolError || error instanceof DistriError)
-        throw error;
+      if (error instanceof A2AProtocolError || error instanceof DistriError) throw error;
       throw new DistriError(`Failed to send message to agent ${agentId}`, "SEND_MESSAGE_ERROR", error);
     }
   }
@@ -569,8 +666,7 @@ var DistriClient = class {
       }
       throw new DistriError("Invalid response format", "INVALID_RESPONSE");
     } catch (error) {
-      if (error instanceof A2AProtocolError || error instanceof DistriError)
-        throw error;
+      if (error instanceof A2AProtocolError || error instanceof DistriError) throw error;
       throw new DistriError(`Failed to get task ${taskId} from agent ${agentId}`, "GET_TASK_ERROR", error);
     }
   }
@@ -597,8 +693,7 @@ var DistriClient = class {
       }
       return await response.json();
     } catch (error) {
-      if (error instanceof ApiError)
-        throw error;
+      if (error instanceof ApiError) throw error;
       throw new DistriError("Failed to fetch threads", "FETCH_ERROR", error);
     }
   }
@@ -610,8 +705,7 @@ var DistriClient = class {
       }
       return await response.json();
     } catch (error) {
-      if (error instanceof ApiError)
-        throw error;
+      if (error instanceof ApiError) throw error;
       throw new DistriError(`Failed to fetch thread ${threadId}`, "FETCH_ERROR", error);
     }
   }
@@ -629,11 +723,28 @@ var DistriClient = class {
       }
       return await response.json();
     } catch (error) {
-      if (error instanceof ApiError)
-        throw error;
+      if (error instanceof ApiError) throw error;
       throw new DistriError(`Failed to fetch messages for thread ${threadId}`, "FETCH_ERROR", error);
     }
   }
+  /**
+   * Get messages from a thread as DistriMessage format
+   */
+  async getThreadMessagesAsDistri(threadId) {
+    const messages = await this.getThreadMessages(threadId);
+    return messages.map(convertA2AMessageToDistri);
+  }
+  /**
+   * Send a DistriMessage to a thread
+   */
+  async sendDistriMessage(threadId, message, context) {
+    const a2aMessage = convertDistriMessageToA2A(message, context);
+    const params = {
+      message: a2aMessage,
+      metadata: context.metadata
+    };
+    await this.sendMessage(threadId, params);
+  }
   /**
    * Get the base URL for making direct requests
    */
@@ -694,20 +805,32 @@ var DistriClient = class {
   /**
    * Helper method to create A2A messages
    */
-  static initMessage(input, role = "user", contextId, messageId, taskId) {
+  static initMessage(parts, role = "user", message) {
     return {
-      messageId: messageId || uuidv4(),
+      messageId: message.messageId || uuidv4(),
+      taskId: message.taskId || uuidv4(),
+      contextId: message.contextId,
       role,
-      parts: [{ kind: "text", text: input.trim() }],
-      contextId,
-      taskId: taskId || uuidv4(),
+      parts: Array.isArray(parts) ? parts : [{ kind: "text", text: parts.trim() }],
+      ...message,
       kind: "message"
     };
   }
+  /**
+   * Create a DistriMessage instance
+   */
+  static initDistriMessage(role, parts, id, created_at) {
+    return {
+      id: id || uuidv4(),
+      role,
+      parts,
+      created_at
+    };
+  }
   /**
    * Helper method to create message send parameters
    */
-  static initMessageParams(message, configuration) {
+  static initMessageParams(message, configuration, metadata) {
     return {
       message,
       configuration: {
@@ -715,7 +838,18 @@ var DistriClient = class {
         blocking: false,
         // Default to non-blocking for streaming
         ...configuration
-      }
+      },
+      metadata
+    };
+  }
+  /**
+   * Create MessageSendParams from a DistriMessage using InvokeContext
+   */
+  static initDistriMessageParams(message, context) {
+    const a2aMessage = convertDistriMessageToA2A(message, context);
+    return {
+      message: a2aMessage,
+      metadata: context.metadata
     };
   }
 };
@@ -731,14 +865,139 @@ function uuidv4() {
     (b, i) => ([4, 6, 8, 10].includes(i) ? "-" : "") + b.toString(16).padStart(2, "0")
   ).join("");
 }
+
+// src/agent.ts
+var Agent = class _Agent {
+  constructor(agentDefinition, client) {
+    this.tools = /* @__PURE__ */ new Map();
+    this.agentDefinition = agentDefinition;
+    this.client = client;
+  }
+  /**
+   * Add a tool to the agent (AG-UI style)
+   */
+  registerTool(tool) {
+    this.tools.set(tool.name, tool);
+  }
+  /**
+   * Add multiple tools at once
+   */
+  registerTools(tools) {
+    tools.forEach((tool) => this.registerTool(tool));
+  }
+  /**
+   * Remove a tool
+   */
+  unregisterTool(toolName) {
+    this.tools.delete(toolName);
+  }
+  /**
+   * Get all registered tools
+   */
+  getTools() {
+    return Array.from(this.tools.values());
+  }
+  /**
+   * Check if a tool is registered
+   */
+  hasTool(toolName) {
+    return this.tools.has(toolName);
+  }
+  /**
+   * Get agent information
+   */
+  get id() {
+    return this.agentDefinition.id;
+  }
+  get name() {
+    return this.agentDefinition.name;
+  }
+  get description() {
+    return this.agentDefinition.description;
+  }
+  get iconUrl() {
+    return this.agentDefinition.icon_url;
+  }
+  /**
+   * Fetch messages for a thread (public method for useChat)
+   */
+  async getThreadMessages(threadId) {
+    return this.client.getThreadMessages(threadId);
+  }
+  /**
+   * Direct (non-streaming) invoke
+   */
+  async invoke(params) {
+    const enhancedParams = this.enhanceParamsWithTools(params);
+    console.log("enhancedParams", enhancedParams);
+    return await this.client.sendMessage(this.agentDefinition.id, enhancedParams);
+  }
+  /**
+   * Streaming invoke
+   */
+  async invokeStream(params) {
+    const enhancedParams = this.enhanceParamsWithTools(params);
+    console.log("enhancedParams", enhancedParams);
+    const a2aStream = this.client.sendMessageStream(this.agentDefinition.id, enhancedParams);
+    return async function* () {
+      for await (const event of a2aStream) {
+        if (event.kind === "message") {
+          yield convertA2AMessageToDistri(event);
+        } else if (event.kind === "status-update") {
+          yield event;
+        } else if (event.kind === "artifact-update") {
+          yield event;
+        } else {
+          yield event;
+        }
+      }
+    }();
+  }
+  /**
+   * Enhance message params with tool definitions
+   */
+  enhanceParamsWithTools(params) {
+    const tools = this.getTools();
+    return {
+      ...params,
+      metadata: {
+        ...params.metadata,
+        tools: tools.map((tool) => ({
+          name: tool.name,
+          description: tool.description,
+          input_schema: tool.input_schema
+        }))
+      }
+    };
+  }
+  /**
+   * Create an agent instance from an agent ID
+   */
+  static async create(agentId, client) {
+    const agentDefinition = await client.getAgent(agentId);
+    return new _Agent(agentDefinition, client);
+  }
+  /**
+   * List all available agents
+   */
+  static async list(client) {
+    const agentDefinitions = await client.getAgents();
+    return agentDefinitions.map((def) => new _Agent(def, client));
+  }
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  A2AClient,
-  A2AProtocolError,
-  ApiError,
-  ConnectionError,
+  Agent,
   DistriClient,
-  DistriError,
+  convertA2AMessageToDistri,
+  convertA2APartToDistri,
+  convertDistriMessageToA2A,
+  convertDistriPartToA2A,
+  extractTextFromDistriMessage,
+  extractToolCallsFromDistriMessage,
+  extractToolResultsFromDistriMessage,
+  isDistriEvent,
+  isDistriMessage,
   uuidv4
 });
 //# sourceMappingURL=index.js.map