Update input/output schemas and payloads

Add more descriptive and accurate input and output schemas and payloads for the stagehand sample app.

Author: dprevoznik

index.ts

@@ -103,7 +103,7 @@ const INVOKE_SAMPLES: Record<
     [TEMPLATE_SAMPLE_APP]:
       'kernel invoke ts-basic get-page-title --payload \'{"url": "https://www.google.com"}\'',
     [TEMPLATE_STAGEHAND]:
-      'kernel invoke ts-stagehand headcount-task --payload \'{"query": "Kernel"}\'',
+      'kernel invoke ts-stagehand teamsize-task --payload \'{"company": "Kernel"}\'',
     [TEMPLATE_ADVANCED_SAMPLE]: "kernel invoke ts-advanced test-captcha-solver",
     [TEMPLATE_COMPUTER_USE]:
       'kernel invoke ts-cu cu-task --payload \'{"query": "Return the first url of a search result for NYC restaurant reviews Pete Wells"}\'',

templates/typescript/stagehand/README.md

@@ -4,13 +4,13 @@ A Stagehand-powered browser automation app that extracts team size information f
 
 ## What it does
 
-The `headcount-task` searches for a startup on Y Combinator's company directory and extracts the team size (number of employees).
+The `teamsize-task` searches for a startup on Y Combinator's company directory and extracts the team size (number of employees).
 
 ## Input
 
 ```json
 {
-  "query": "kernel"  // Startup name to search (optional, defaults to "kernel")
+  "company": "kernel"  // Startup name to search (optional, defaults to "kernel")
 }
 ```
 
@@ -41,10 +41,10 @@ kernel deploy index.ts --env-file .env
 
 Default query (searches for "kernel"):
 ```bash
-kernel invoke ts-stagehand headcount-task
+kernel invoke ts-stagehand teamsize-task
 ```
 
 Custom query:
 ```bash
-kernel invoke ts-stagehand headcount-task --payload '{"query": "Mixpanel"}'
+kernel invoke ts-stagehand teamsize-task --payload '{"company": "Mixpanel"}'
 ```

templates/typescript/stagehand/index.ts

@@ -6,11 +6,11 @@ const kernel = new Kernel();
 
 const app = kernel.app('ts-stagehand');
 
-interface SearchQueryInput {
-  query: string;
+interface CompanyInput {
+  company: string;
 }
 
-interface SearchQueryOutput {
+interface TeamSizeOutput {
   teamSize: string;
 }
 
@@ -23,9 +23,9 @@ if (!OPENAI_API_KEY) {
   throw new Error('OPENAI_API_KEY is not set');
 }
 
-app.action<SearchQueryInput, SearchQueryOutput>(
-  'headcount-task',
-  async (ctx: KernelContext, payload?: SearchQueryInput): Promise<SearchQueryOutput> => {
+app.action<CompanyInput, TeamSizeOutput>(
+  'teamsize-task',
+  async (ctx: KernelContext, payload?: CompanyInput): Promise<TeamSizeOutput> => {
     // A function that returns the team size of a Y Combinator startup
 
     // Args:
@@ -35,7 +35,7 @@ app.action<SearchQueryInput, SearchQueryOutput>(
     // Returns:
     //     output: The team size (number of employees) of the startup
 
-    const query = payload?.query || 'kernel';
+    const company = payload?.company || 'kernel';
 
     const kernelBrowser = await kernel.browsers.create({
       invocation_id: ctx.invocation_id,
@@ -62,7 +62,7 @@ app.action<SearchQueryInput, SearchQueryOutput>(
     const page = stagehand.context.pages()[0];
     await page.goto("https://www.ycombinator.com/companies");
 
-    await stagehand.act(`Type in ${query} into the search box`);
+    await stagehand.act(`Type in ${company} into the search box`);
     await stagehand.act("Click on the first search result");
 
     // Schema definition