add mb-metadata to local toolcalls

Author: Markeljan

src/components/chat/ChatContent.tsx

@@ -79,7 +79,15 @@ export const ChatContent = ({
       if (!localAgent) return undefined;
 
       try {
-        return await executeLocalToolCall(localAgent, toolCall);
+        return await executeLocalToolCall({
+          localAgent,
+          toolCall,
+          metadata: {
+            accountId,
+            evmAddress,
+            chainId,
+          },
+        });
       } catch (error) {
         const errorMessage =
           error instanceof Error ? error.message : "Unknown error";

src/lib/local-agent.ts

@@ -1,8 +1,9 @@
 import { BitteToolResult } from "../types";
 
-interface ToolCall {
-  toolName: string;
-  args?: any;
+interface ToolCall<NAME extends string, ARGS> {
+  toolCallId: string;
+  toolName: NAME;
+  args: ARGS;
 }
 
 interface LocalAgent {
@@ -12,10 +13,17 @@ interface LocalAgent {
   };
 }
 
-export const executeLocalToolCall = async (
-  localAgent: LocalAgent,
-  toolCall: ToolCall
-): Promise<BitteToolResult | undefined> => {
+interface LocalToolCallProps {
+  localAgent: LocalAgent;
+  toolCall: ToolCall<string, unknown>;
+  metadata?: Record<string, unknown>;
+}
+
+export const executeLocalToolCall = async ({
+  localAgent,
+  toolCall,
+  metadata,
+}: LocalToolCallProps): Promise<BitteToolResult | undefined> => {
   const baseUrl = localAgent.spec.servers?.[0]?.url;
   if (!baseUrl) return undefined;
 
@@ -32,7 +40,12 @@ export const executeLocalToolCall = async (
   try {
     const args = toolCall.args ? JSON.parse(JSON.stringify(toolCall.args)) : {};
     const { url, remainingArgs } = buildUrlWithParams(baseUrl, toolPath, args);
-    const { options } = buildRequestOptions(httpMethod, remainingArgs);
+    
+    const { options } = buildRequestOptions({
+      httpMethod,
+      remainingArgs,
+      metadata,
+    });
 
     const finalUrl =
       httpMethod === "GET" ? handleQueryParams(url, remainingArgs) : url;
@@ -103,12 +116,18 @@ export const buildUrlWithParams = (
   return { url, remainingArgs };
 };
 
-export const buildRequestOptions = (
-  httpMethod: string,
-  remainingArgs: any
-): { options: RequestInit } => {
+export const buildRequestOptions = ({
+  httpMethod,
+  remainingArgs,
+  metadata,
+}: {
+  httpMethod: string;
+  remainingArgs: any;
+  metadata?: Record<string, unknown>;
+}): { options: RequestInit } => {
   const headers: HeadersInit = {
     "Content-Type": "application/json",
+    ...(metadata ? { "mb-metadata": JSON.stringify(metadata) } : {}),
   };
 
   const fetchOptions: RequestInit = {

src/lib/sign-message.ts

@@ -37,7 +37,6 @@ export const signMessage = async (
       console.error("failed in signMessage", errorMessage);
       throw new Error(errorMessage);
     });
-  console.log("signedMessage", signedMessage);
 
   // Bitte Wallet redirects so signMessage will not have a result
   if (signedMessage) {