From a86630d4222512b66320c540a7f0c4943ff29cf2 Mon Sep 17 00:00:00 2001
From: tylerslaton <mtslaton1@gmail.com>
Date: Tue, 15 Oct 2024 12:06:05 -0400
Subject: [PATCH] fix: don't error when tool call has no metadata

This resolves an issue defined in #183 where tool
calls with no metadata would cause an error when
loading thread history.

Signed-off-by: tylerslaton <mtslaton1@gmail.com>
---
 ui/admin/app/components/chat/Message.tsx | 2 +-
 ui/admin/app/lib/model/chatEvents.ts     | 2 +-
 ui/admin/app/lib/model/messages.ts       | 2 +-
 ui/admin/app/lib/model/toolReferences.ts | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/ui/admin/app/components/chat/Message.tsx b/ui/admin/app/components/chat/Message.tsx
index ee0ad549..845dd984 100644
--- a/ui/admin/app/components/chat/Message.tsx
+++ b/ui/admin/app/components/chat/Message.tsx
@@ -67,7 +67,7 @@ export function Message({ message }: MessageProps) {
                         )}
                     >
                         <div className="max-w-full overflow-hidden p-4 flex gap-2 items-center pl-[20px]">
-                            {toolCall?.metadata.icon && (
+                            {toolCall?.metadata?.icon && (
                                 <ToolIcon
                                     icon={toolCall.metadata.icon}
                                     category={toolCall.metadata.category}
diff --git a/ui/admin/app/lib/model/chatEvents.ts b/ui/admin/app/lib/model/chatEvents.ts
index 1c036f35..025a4c3c 100644
--- a/ui/admin/app/lib/model/chatEvents.ts
+++ b/ui/admin/app/lib/model/chatEvents.ts
@@ -7,7 +7,7 @@ export type ToolCall = {
     name: string;
     description: string;
     input: string;
-    metadata: {
+    metadata?: {
         category?: string;
         icon?: string;
     };
diff --git a/ui/admin/app/lib/model/messages.ts b/ui/admin/app/lib/model/messages.ts
index 5da519b9..e6324be9 100644
--- a/ui/admin/app/lib/model/messages.ts
+++ b/ui/admin/app/lib/model/messages.ts
@@ -35,7 +35,7 @@ export const runsToMessages = (runs: Run[]) => {
 export const toolCallMessage = (toolCall: ToolCall) => {
     return {
         sender: "agent",
-        text: `Tool call: ${[toolCall.metadata.category, toolCall.name].filter((x) => !!x).join(" - ")}`,
+        text: `Tool call: ${[toolCall.metadata?.category, toolCall.name].filter((x) => !!x).join(" - ")}`,
         tools: [toolCall],
     } as Message;
 };
diff --git a/ui/admin/app/lib/model/toolReferences.ts b/ui/admin/app/lib/model/toolReferences.ts
index 6b242451..79adcb7c 100644
--- a/ui/admin/app/lib/model/toolReferences.ts
+++ b/ui/admin/app/lib/model/toolReferences.ts
@@ -5,7 +5,7 @@ export type ToolReferenceBase = {
     name: string;
     toolType: ToolReferenceType;
     reference: string;
-    metadata: Record<string, string>;
+    metadata?: Record<string, string>;
 };
 
 export type ToolReferenceType = "tool" | "stepTemplate";