fix: create proper tool arguments schema

Author: blurrah

dev/graphql.ts

@@ -274,7 +274,15 @@ const yoga = createYoga({ schema });
 // Start server with proper request handler
 const server = Bun.serve({
   port: 4000,
-  fetch: (request) => yoga.fetch(request),
+  fetch: (request) => {
+    // Add dev logger for incoming requests
+    console.log(
+      `[${new Date().toISOString()}] Incoming request: ${request.method} ${
+        request.url
+      }`
+    );
+    return yoga.fetch(request);
+  },
 });
 
 console.info(

src/index.ts

@@ -33,8 +33,6 @@ const server = new Server(
 server.setRequestHandler(ListToolsRequestSchema, async (request) => {
   await handler.loadTools();
 
-  console.error(handler.tools);
-
   const tools = Array.from(handler.tools.values()).map((tool) => ({
     name: tool.name,
     description: tool.description,
@@ -56,8 +54,12 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
     };
   }
 
+  console.error("tool call", request.params.arguments);
+
   const parsedArguments = tool.parameters.parse(request.params.arguments);
 
+  console.error("parsed arguments", parsedArguments);
+
   const result = await handler.execute(
     parsedArguments.query,
     parsedArguments.variables

src/lib/graphql.ts

@@ -102,17 +102,6 @@ export function getOperations(
         } satisfies Operation),
         parameters: field.args,
       });
-
-      if (fieldName === "commentsByPost") {
-        console.error(
-          createOperationDescription(schema, {
-            name: fieldName,
-            type: "query",
-            parameters: field.args,
-            description: field.description,
-          } satisfies Operation)
-        );
-      }
     }
   }
 
@@ -190,7 +179,10 @@ function buildZodSchemaFromVariables(
     schemaObj[definition.name] = argumentToZodSchema(definition);
   }
 
-  return z.object(schemaObj);
+  return z.object({
+    variables: z.object(schemaObj),
+    query: z.string(),
+  });
 }
 
 function argumentToZodSchema(argument: GraphQLArgument): z.ZodTypeAny {
@@ -292,11 +284,15 @@ export async function createGraphQLHandler(config: Config) {
     tools,
     loadTools,
     async execute(query: string, variables: unknown) {
+      const body = JSON.stringify({ query, variables });
+      console.error("body", body);
       const result = await fetch(config.endpoint, {
         method: "POST",
-        body: JSON.stringify({ query, variables }),
+        body,
       });
 
+      console.error("result", await result.json());
+
       return {
         status: "success",
         data: await result.json(),