Merge pull request #581 from dcSpark/feature/versions

Adding versions
dcSpark · Dec 27, 2024 · 4081e10 · 4081e10
2 parents 2555f49 + face908
commit 4081e10
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 1 deletion.
diff --git a/apps/shinkai-desktop/src/components/playground-tool/schemas.ts b/apps/shinkai-desktop/src/components/playground-tool/schemas.ts
@@ -13,6 +13,8 @@ export const ToolMetadataSchema = z.object({
 
   author: z.string(),
 
+  version: z.string().default('1.0.0'),
+
   keywords: z.array(z.string()).default([]),
 
   configurations: z

diff --git a/apps/shinkai-desktop/src/pages/tools.tsx b/apps/shinkai-desktop/src/pages/tools.tsx
@@ -25,7 +25,7 @@ import {
   TooltipProvider,
   TooltipTrigger,
 } from '@shinkai_network/shinkai-ui';
-import { formatText } from '@shinkai_network/shinkai-ui/helpers';
+import { formatText, getVersionFromTool } from '@shinkai_network/shinkai-ui/helpers';
 import { cn } from '@shinkai_network/shinkai-ui/utils';
 import {
   BoltIcon,
@@ -150,6 +150,9 @@ export const Tools = () => {
                   <span className="text-sm font-medium capitalize text-white">
                     {formatText(tool.name)}{' '}
                   </span>
+                  <Badge className="text-gray-80 bg-gray-200 text-xs font-normal">
+                    {getVersionFromTool(tool)}
+                  </Badge>
                   {tool.author !== '@@official.shinkai' && (
                     <Badge className="text-gray-80 bg-gray-200 text-xs font-normal">
                       {tool.author}
@@ -213,6 +216,9 @@ export const Tools = () => {
                   <span className="text-sm font-medium text-white">
                     {formatText(tool.name)}{' '}
                   </span>
+                  <Badge className="text-gray-80 bg-gray-200 text-xs font-normal">
+                    {getVersionFromTool(tool)}
+                  </Badge>
                   {tool.author !== '@@official.shinkai' && (
                     <Badge className="text-gray-80 bg-gray-200 text-xs font-normal">
                       {tool.author}

diff --git a/libs/shinkai-ui/src/helpers/format-text.ts b/libs/shinkai-ui/src/helpers/format-text.ts
@@ -1,3 +1,5 @@
+import { ShinkaiToolHeader } from "@shinkai_network/shinkai-message-ts/api/tools/types";
+
 export const formatText = (text: string) => {
   const words = text.split('_');
 
@@ -21,3 +23,14 @@ export const formatCamelCaseText = (text: string) => {
 
   return result.charAt(0).toUpperCase() + result.slice(1);
 };
+
+export const getVersionFromTool = (toolRouterKey: ShinkaiToolHeader) => {
+  if (toolRouterKey.version) {
+    return toolRouterKey.version;
+  }
+  const parts = toolRouterKey.name.split(':::');
+  if (parts.length === 4) {
+    return parts[3];
+  }
+  return 'latest';
+};