Merge pull request #23 from ivo-toby/decrease_token_usage

Decrease token usage

Author: ivo-toby

src/handlers/asset-handlers.ts

@@ -17,7 +17,37 @@ const getCurrentAsset = async (params: BaseAssetParams) => {
   return contentfulClient.asset.get(params)
 }
 
+import { summarizeData } from "../utils/summarizer.js"
+
 export const assetHandlers = {
+  listAssets: async (args: {
+    spaceId: string
+    environmentId: string
+    limit: number
+    skip: number
+  }) => {
+    const spaceId = process.env.SPACE_ID || args.spaceId
+    const environmentId = process.env.ENVIRONMENT_ID || args.environmentId
+
+    const params = {
+      spaceId,
+      environmentId,
+      query: {
+        limit: Math.min(args.limit || 3, 3),
+        skip: args.skip || 0,
+      },
+    }
+
+    const contentfulClient = await getContentfulClient()
+    const assets = await contentfulClient.asset.getMany(params)
+
+    const summarized = summarizeData(assets, {
+      maxItems: 3,
+      remainingMessage: "To see more assets, please ask me to retrieve the next page.",
+    })
+
+    return formatResponse(summarized)
+  },
   uploadAsset: async (args: {
     spaceId: string
     environmentId: string

src/handlers/content-type-handlers.ts

@@ -18,7 +18,6 @@ export const contentTypeHandlers = {
       content: [{ type: "text", text: JSON.stringify(contentTypes, null, 2) }],
     }
   },
-
   getContentType: async (args: {
     spaceId: string
     environmentId: string

src/handlers/entry-handlers.ts

@@ -1,5 +1,6 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 import { getContentfulClient } from "../config/client.js"
+import { summarizeData } from "../utils/summarizer.js"
 import { CreateEntryProps, EntryProps, QueryOptions } from "contentful-management"
 
 export const entryHandlers = {
@@ -15,11 +16,20 @@ export const entryHandlers = {
     const contentfulClient = await getContentfulClient()
     const entries = await contentfulClient.entry.getMany({
       ...params,
-      query: args.query,
+      query: {
+        ...args.query,
+        limit: Math.min(args.query.limit || 3, 3),
+        skip: args.query.skip || 0
+      },
+    })
+
+    const summarized = summarizeData(entries, {
+      maxItems: 3,
+      remainingMessage: "To see more entries, please ask me to retrieve the next page."
     })
 
     return {
-      content: [{ type: "text", text: JSON.stringify(entries, null, 2) }],
+      content: [{ type: "text", text: JSON.stringify(summarized, null, 2) }],
     }
   },
   createEntry: async (args: {

src/index.ts

@@ -91,6 +91,7 @@ function getHandler(name: string) {
     delete_asset: assetHandlers.deleteAsset,
     publish_asset: assetHandlers.publishAsset,
     unpublish_asset: assetHandlers.unpublishAsset,
+    list_assets: assetHandlers.listAssets,
 
     // Space & Environment operations
     list_spaces: spaceHandlers.listSpaces,

src/types/tools.ts

@@ -32,7 +32,8 @@ export const getEntryTools = () => {
   return {
     SEARCH_ENTRIES: {
       name: "search_entries",
-      description: "Search for entries using query parameters",
+      description:
+        "Search for entries using query parameters. Returns a maximum of 3 items per request. Use skip parameter to paginate through results.",
       inputSchema: getSpaceEnvProperties({
         type: "object",
         properties: {
@@ -42,19 +43,30 @@ export const getEntryTools = () => {
             properties: {
               content_type: { type: "string" },
               select: { type: "string" },
-              limit: { type: "number" },
-              skip: { type: "number" },
+              limit: {
+                type: "number",
+                default: 3,
+                maximum: 3,
+                description: "Maximum number of items to return (max: 3)",
+              },
+              skip: {
+                type: "number",
+                default: 0,
+                description: "Number of items to skip for pagination",
+              },
               order: { type: "string" },
               query: { type: "string" },
             },
+            required: ["limit", "skip"],
           },
         },
         required: ["query"],
       }),
     },
     CREATE_ENTRY: {
       name: "create_entry",
-      description: "Create a new entry in Contentful",
+      description:
+        "Create a new entry in Contentful, before executing this function, you need to know the contentTypeId and the fields of that contentType, you can get the fields definition by using the GET_CONTENT_TYPE tool. ",
       inputSchema: getSpaceEnvProperties({
         type: "object",
         properties: {
@@ -130,8 +142,29 @@ export const getEntryTools = () => {
 // Tool definitions for Asset operations
 export const getAssetTools = () => {
   return {
+    LIST_ASSETS: {
+      name: "list_assets",
+      description: "List assets in a space. Returns a maximum of 3 items per request. Use skip parameter to paginate through results.",
+      inputSchema: getSpaceEnvProperties({
+        type: "object",
+        properties: {
+          limit: {
+            type: "number",
+            default: 3,
+            maximum: 3,
+            description: "Maximum number of items to return (max: 3)"
+          },
+          skip: {
+            type: "number",
+            default: 0,
+            description: "Number of items to skip for pagination"
+          }
+        },
+        required: ["limit", "skip"]
+      }),
+    },
     UPLOAD_ASSET: {
-      name: "upload_asset",
+      name: "upload_asset", 
       description: "Upload a new asset",
       inputSchema: getSpaceEnvProperties({
         type: "object",
@@ -226,11 +259,23 @@ export const getContentTypeTools = () => {
     LIST_CONTENT_TYPES: {
       name: "list_content_types",
       description:
-        "List content types in a space. Requires either spaceId parameter to identify the target space.",
+        "List content types in a space. Returns a maximum of 10 items per request. Use skip parameter to paginate through results.",
       inputSchema: getSpaceEnvProperties({
         type: "object",
-        properties: {},
-        required: [],
+        properties: {
+          limit: {
+            type: "number",
+            default: 10,
+            maximum: 20,
+            description: "Maximum number of items to return (max: 3)",
+          },
+          skip: {
+            type: "number",
+            default: 0,
+            description: "Number of items to skip for pagination",
+          },
+        },
+        required: ["limit", "skip"],
       }),
     },
     GET_CONTENT_TYPE: {

src/utils/summarizer.ts

@@ -0,0 +1,51 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+
+export interface SummarizeOptions {
+  maxItems?: number
+  indent?: number
+  showTotal?: boolean
+  remainingMessage?: string
+}
+
+export const summarizeData = (data: any, options: SummarizeOptions = {}): any => {
+  const { maxItems = 3, remainingMessage = "To see more items, please ask me to retrieve the next page." } =
+    options
+
+  // Handle Contentful-style responses with items and total
+  if (data && typeof data === "object" && "items" in data && "total" in data) {
+    const items = data.items
+    const total = data.total
+
+    if (items.length <= maxItems) {
+      return data
+    }
+
+    return {
+      items: items.slice(0, maxItems),
+      total: total,
+      showing: maxItems,
+      remaining: total - maxItems,
+      message: remainingMessage,
+      skip: maxItems // Add skip value for next page
+    }
+  }
+
+  // Handle plain arrays
+  if (Array.isArray(data)) {
+    if (data.length <= maxItems) {
+      return data
+    }
+
+    return {
+      items: data.slice(0, maxItems),
+      total: data.length,
+      showing: maxItems,
+      remaining: data.length - maxItems,
+      message: remainingMessage,
+      skip: maxItems // Add skip value for next page
+    }
+  }
+
+  // Return non-array data as-is
+  return data
+}