@loopstack/github-oauth-example 0.1.1 → 0.2.0

package/src/workflows/github-agent.workflow.ts

@@ -1,21 +1,22 @@
-import { Injectable } from '@nestjs/common';
 import { z } from 'zod';
 import {
   ClaudeGenerateText,
+  ClaudeGenerateTextResult,
   ClaudeMessageDocument,
   DelegateToolCalls,
+  DelegateToolCallsResult,
   UpdateToolResult,
 } from '@loopstack/claude-module';
 import {
-  InjectDocument,
+  BaseWorkflow,
+  Guard,
+  Initial,
   InjectTool,
   InjectWorkflow,
-  Runtime,
-  State,
+  ToolResult,
+  Transition,
   Workflow,
-  WorkflowInterface,
 } from '@loopstack/common';
-import { CreateDocument, LinkDocument, Task } from '@loopstack/core';
 import {
   GitHubCreateIssueCommentTool,
   GitHubCreateIssueTool,
@@ -46,21 +47,15 @@ import {
 import { OAuthWorkflow } from '@loopstack/oauth-module';
 import { AuthenticateGitHubTask } from '../tools/authenticate-github-task.tool';
 
-@Injectable()
 @Workflow({
-  configFile: __dirname + '/github-agent.workflow.yaml',
+  uiConfig: __dirname + '/github-agent.ui.yaml',
 })
-export class GitHubAgentWorkflow implements WorkflowInterface {
-  @InjectTool() createDocument: CreateDocument;
+export class GitHubAgentWorkflow extends BaseWorkflow {
   @InjectTool() claudeGenerateText: ClaudeGenerateText;
   @InjectTool() delegateToolCalls: DelegateToolCalls;
   @InjectTool() updateToolResult: UpdateToolResult;
-  @InjectTool() task: Task;
   @InjectTool() authenticateGitHub: AuthenticateGitHubTask;
 
-  @InjectDocument() claudeMessageDocument: ClaudeMessageDocument;
-  @InjectDocument() linkDocument: LinkDocument;
-
   // GitHub Repos tools
   @InjectTool() gitHubListRepos: GitHubListReposTool;
   @InjectTool() gitHubGetRepo: GitHubGetRepoTool;
@@ -102,17 +97,104 @@ export class GitHubAgentWorkflow implements WorkflowInterface {
 
   @InjectWorkflow() oAuth: OAuthWorkflow;
 
-  @State({
-    schema: z.object({
-      llmResult: z.any().optional(),
-      delegateResult: z.any().optional(),
-    }),
-  })
-  state: {
-    llmResult?: any;
-    delegateResult?: any;
-  };
-
-  @Runtime()
-  runtime: any;
+  llmResult?: ClaudeGenerateTextResult;
+  delegateResult?: DelegateToolCallsResult;
+
+  @Initial({ to: 'waiting_for_user' })
+  async setup() {
+    await this.repository.save(
+      ClaudeMessageDocument,
+      {
+        role: 'user',
+        content: this.render(__dirname + '/templates/systemMessage.md'),
+      },
+      { meta: { hidden: true } },
+    );
+  }
+
+  @Transition({ from: 'waiting_for_user', to: 'ready', wait: true, schema: z.string() })
+  async userMessage(payload: string) {
+    await this.repository.save(ClaudeMessageDocument, {
+      role: 'user',
+      content: payload,
+    });
+  }
+
+  @Transition({ from: 'ready', to: 'prompt_executed' })
+  async llmTurn() {
+    const result: ToolResult<ClaudeGenerateTextResult> = await this.claudeGenerateText.call({
+      system: `You are a helpful GitHub assistant with access to repository, issue, PR, code, actions,
+and search tools. When a tool returns an unauthorized error, use authenticateGitHub
+to let the user sign in, then retry. Be concise and format results using markdown.`,
+      claude: { model: 'claude-sonnet-4-6' },
+      messagesSearchTag: 'message',
+      tools: [
+        'gitHubListRepos',
+        'gitHubGetRepo',
+        'gitHubCreateRepo',
+        'gitHubListBranches',
+        'gitHubListIssues',
+        'gitHubGetIssue',
+        'gitHubCreateIssue',
+        'gitHubCreateIssueComment',
+        'gitHubListPullRequests',
+        'gitHubGetPullRequest',
+        'gitHubCreatePullRequest',
+        'gitHubMergePullRequest',
+        'gitHubListPrReviews',
+        'gitHubGetFileContent',
+        'gitHubCreateOrUpdateFile',
+        'gitHubListDirectory',
+        'gitHubGetCommit',
+        'gitHubListWorkflowRuns',
+        'gitHubTriggerWorkflow',
+        'gitHubGetWorkflowRun',
+        'gitHubSearchCode',
+        'gitHubSearchRepos',
+        'gitHubSearchIssues',
+        'gitHubGetAuthenticatedUser',
+        'gitHubListUserOrgs',
+        'authenticateGitHub',
+      ],
+    });
+    this.llmResult = result.data;
+  }
+
+  @Transition({ from: 'prompt_executed', to: 'awaiting_tools', priority: 10 })
+  @Guard('hasToolCalls')
+  async executeToolCalls() {
+    const result: ToolResult<DelegateToolCallsResult> = await this.delegateToolCalls.call({
+      message: this.llmResult!,
+      document: ClaudeMessageDocument,
+      callback: { transition: 'toolResultReceived' },
+    });
+    this.delegateResult = result.data;
+  }
+
+  hasToolCalls(): boolean {
+    return this.llmResult?.stop_reason === 'tool_use';
+  }
+
+  @Transition({ from: 'awaiting_tools', to: 'awaiting_tools', wait: true, schema: z.record(z.string(), z.unknown()) })
+  async toolResultReceived(payload: Record<string, unknown>) {
+    const result: ToolResult<DelegateToolCallsResult> = await this.updateToolResult.call({
+      delegateResult: this.delegateResult!,
+      completedTool: payload,
+      document: ClaudeMessageDocument,
+    });
+    this.delegateResult = result.data;
+  }
+
+  @Transition({ from: 'awaiting_tools', to: 'ready' })
+  @Guard('allToolsComplete')
+  allToolsCompleteTransition() {}
+
+  allToolsComplete(): boolean {
+    return this.delegateResult?.allCompleted ?? false;
+  }
+
+  @Transition({ from: 'prompt_executed', to: 'waiting_for_user' })
+  async respond() {
+    await this.repository.save(ClaudeMessageDocument, this.llmResult!, { id: this.llmResult!.id });
+  }
 }

package/src/workflows/github-repos-overview.ui.yaml

@@ -0,0 +1,17 @@
+title: 'GitHub Repository Overview'
+
+description: |
+  Comprehensive GitHub example that exercises every GitHub tool.
+  Fetches user info, repository details, issues, pull requests, branches,
+  directory contents, workflow runs, and search results for a given repository.
+  If not authenticated, launches the OAuth sub-workflow and retries.
+
+ui:
+  form:
+    properties:
+      owner:
+        title: 'Repository Owner'
+        placeholder: 'octocat'
+      repo:
+        title: 'Repository Name'
+        placeholder: 'Hello-World'