feat: Attach timestamp tool

inferablehq · Dec 16, 2024 · 75a9ac7 · 75a9ac7
1 parent 58e4fff
commit 75a9ac7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/control-plane/src/modules/workflows/agent/run.ts b/control-plane/src/modules/workflows/agent/run.ts
@@ -397,8 +397,11 @@ export const findRelevantTools = async (state: WorkflowAgentState) => {
     const accessKnowledgeArtifactsTool =
       await buildAccessKnowledgeArtifacts(workflow);
 
+    const currentDateTimeTool = buildCurrentDateTimeTool();
+
     // When tools are not specified, attach all internalTools
     tools.push(accessKnowledgeArtifactsTool);
+    tools.push(currentDateTimeTool);
 
     events.write({
       type: "functionRegistrySearchCompleted",

diff --git a/control-plane/src/modules/workflows/agent/tools/date-time.ts b/control-plane/src/modules/workflows/agent/tools/date-time.ts
@@ -6,9 +6,12 @@ export const CURRENT_DATE_TIME_TOOL_NAME = "currentDateTime";
 export const buildCurrentDateTimeTool = (): AgentTool =>
   new AgentTool({
     name: CURRENT_DATE_TIME_TOOL_NAME,
-    description: "Retrieves the current date and time in ISO 8601 format.",
+    description: "Retrieves the current date and time in ISO 8601 format and unix timestamp.",
     schema: z.object({}),
     func: async () => {
-      return new Date().toISOString();
+      return JSON.stringify({
+        iso8601: new Date().toISOString(),
+        unix: new Date().getTime()
+      })
     },
   });