sdk v0.1

Author: charleslien

common/src/actions.ts

@@ -8,7 +8,7 @@ import {
   toolCallSchema,
   toolResultSchema,
 } from './types/session-state'
-import { FileVersionSchema, ProjectFileContextSchema } from './util/file'
+import { ProjectFileContextSchema } from './util/file'
 
 export const FileChangeSchema = z.object({
   type: z.enum(['patch', 'file']),
@@ -96,22 +96,6 @@ export const InitResponseSchema = z
   )
 export type InitResponse = z.infer<typeof InitResponseSchema>
 
-export const ResponseCompleteSchema = z
-  .object({
-    type: z.literal('response-complete'),
-    userInputId: z.string(),
-    response: z.string(),
-    changes: CHANGES,
-    changesAlreadyApplied: CHANGES,
-    addedFileVersions: z.array(FileVersionSchema),
-    resetFileVersions: z.boolean(),
-  })
-  .merge(
-    UsageReponseSchema.omit({
-      type: true,
-    }).partial(),
-  )
-
 export const MessageCostResponseSchema = z.object({
   type: z.literal('message-cost-response'),
   promptId: z.string(),
@@ -142,7 +126,6 @@ export const SERVER_ACTION_SCHEMA = z.discriminatedUnion('type', [
     chunk: z.string(),
     prompt: z.string().optional(),
   }),
-  ResponseCompleteSchema,
   PromptResponseSchema,
   z.object({
     type: z.literal('read-files'),
@@ -157,26 +140,6 @@ export const SERVER_ACTION_SCHEMA = z.discriminatedUnion('type', [
     args: z.record(z.any()),
     timeout: z.number().optional(),
   }),
-  z.object({
-    type: z.literal('tool-call'),
-    userInputId: z.string(),
-    response: z.string(),
-    data: toolCallSchema,
-    changes: CHANGES,
-    changesAlreadyApplied: CHANGES,
-    addedFileVersions: z.array(FileVersionSchema),
-    resetFileVersions: z.boolean(),
-  }),
-  z.object({
-    type: z.literal('terminal-command-result'),
-    userInputId: z.string(),
-    result: z.string(),
-  }),
-  z.object({
-    type: z.literal('npm-version-status'),
-    isUpToDate: z.boolean(),
-    latestVersion: z.string(),
-  }),
   InitResponseSchema,
   UsageReponseSchema,
   MessageCostResponseSchema,
@@ -186,10 +149,6 @@ export const SERVER_ACTION_SCHEMA = z.discriminatedUnion('type', [
     error: z.string().optional(),
     remainingBalance: z.number().optional(),
   }),
-  z.object({
-    type: z.literal('commit-message-response'),
-    commitMessage: z.string(),
-  }),
   z.object({
     // The server is imminently going to shutdown, and the client should reconnect
     type: z.literal('request-reconnect'),

common/src/websockets/websocket-client.ts

@@ -60,10 +60,14 @@ export class APIRealtimeClient {
   connectTimeout?: any
   heartbeat?: any
   hadError = false
-  onError: () => void
+  onError: (event: WebSocket.ErrorEvent) => void
   onReconnect: () => void
 
-  constructor(url: string, onError: () => void, onReconnect: () => void) {
+  constructor(
+    url: string,
+    onError: (event: WebSocket.ErrorEvent) => void,
+    onReconnect: () => void,
+  ) {
     this.url = url
     this.txid = 0
     this.txns = new Map()
@@ -94,7 +98,7 @@ export class APIRealtimeClient {
     }
     this.ws.onerror = (ev) => {
       if (!this.hadError) {
-        this.onError()
+        this.onError(ev)
         this.hadError = true
       }
       // this can fire without an onclose if this is the first time we ever try

npm-app/src/client.ts

@@ -832,17 +832,6 @@ export class Client {
       }
     })
 
-    this.webSocket.subscribe('npm-version-status', (action) => {
-      const { isUpToDate } = action
-      if (!isUpToDate) {
-        console.warn(
-          yellow(
-            `\nThere's a new version of Codebuff! Please update to ensure proper functionality.\nUpdate now by running: npm install -g codebuff`,
-          ),
-        )
-      }
-    })
-
     this.webSocket.subscribe('message-cost-response', (action) => {
       const parsedAction = MessageCostResponseSchema.safeParse(action)
       if (!parsedAction.success) return
@@ -934,25 +923,6 @@ export class Client {
     }
   }
 
-  async generateCommitMessage(stagedChanges: string): Promise<string> {
-    return new Promise(async (resolve, reject) => {
-      const unsubscribe = this.webSocket.subscribe(
-        'commit-message-response',
-        (action) => {
-          unsubscribe()
-          resolve(action.commitMessage)
-        },
-      )
-
-      this.webSocket.sendAction({
-        type: 'generate-commit-message',
-        fingerprintId: await this.fingerprintId,
-        authToken: this.user?.authToken,
-        stagedChanges,
-      })
-    })
-  }
-
   async sendUserInput(prompt: string): Promise<{
     responsePromise: Promise<
       ServerAction & { type: 'prompt-response' | 'manager-prompt-response' } & {
@@ -1576,7 +1546,7 @@ Go to https://www.codebuff.com/config for more information.`) +
       this.setUsage(parsedAction.data)
     })
 
-    const initAction: ClientAction = {
+    const initAction: Extract<ClientAction, { type: 'init' }> = {
       type: 'init',
       fingerprintId: await this.fingerprintId,
       authToken: this.user?.authToken,

sdk/package.json

@@ -2,7 +2,7 @@
   "name": "@codebuff/sdk",
   "private": false,
   "access": "public",
-  "version": "0.0.3",
+  "version": "0.1.0",
   "description": "Official SDK for Codebuff — AI coding agent & framework",
   "license": "MIT",
   "type": "module",

sdk/src/client.ts

@@ -1,20 +1,14 @@
 import { execFileSync } from 'child_process'
 
-import { API_KEY_ENV_VAR } from '../../common/src/constants'
 import { CODEBUFF_BINARY } from './constants'
 import { processStream } from './process-stream'
+import { API_KEY_ENV_VAR } from '../../common/src/constants'
 
-import type {
-  CodebuffClientOptions,
-  ChatContext,
-  ContinueChatOptions,
-  NewChatOptions,
-} from './types'
-
+/** @deprecated Migrate to WebSocketHandler */
 export class CodebuffClient {
   public cwd: string
 
-  constructor({ cwd }: CodebuffClientOptions) {
+  constructor({ cwd }: { cwd: string }) {
     // TODO: download binary automatically
     if (execFileSync('which', [CODEBUFF_BINARY]).toString().trim() === '') {
       throw new Error(
@@ -35,37 +29,14 @@ export class CodebuffClient {
     prompt,
     params,
     handleEvent,
-  }: NewChatOptions): Promise<ChatContext> {
-    const args = [prompt, '-p', '--agent', agent]
-    if (prompt) {
-      args.push(prompt)
-    }
-    if (params) {
-      args.push('--params', JSON.stringify(params))
-    }
-    if (this.cwd) {
-      args.push('--cwd', this.cwd)
-    }
-
-    await processStream({
-      codebuffArgs: args,
-      handleEvent,
-    })
-
-    return {
-      agentId: agent,
-    }
-  }
-
-  // WIP
-  private async continueChat({
-    agent,
-    prompt,
-    params,
-    context,
-    handleEvent,
-  }: ContinueChatOptions): Promise<ChatContext> {
-    agent = agent ?? context.agentId
+  }: {
+    agent: string
+    prompt: string
+    params?: Record<string, any>
+    handleEvent: (event: any) => void
+  }): Promise<{
+    agentId: string
+  }> {
     const args = [prompt, '-p', '--agent', agent]
     if (prompt) {
       args.push(prompt)

sdk/src/constants.ts

@@ -1 +1,15 @@
 export const CODEBUFF_BINARY = 'codebuff'
+
+export const IS_DEV = process.env.NEXT_PUBLIC_CB_ENVIRONMENT === 'dev'
+export const IS_TEST = process.env.NEXT_PUBLIC_CB_ENVIRONMENT === 'test'
+export const IS_PROD = !IS_DEV && !IS_TEST
+
+export const WEBSOCKET_URL = IS_PROD
+  ? 'wss://manicode-backend.onrender.com/ws'
+  : 'ws://localhost:4242/ws'
+export const WEBSITE_URL = IS_PROD
+  ? 'https://codebuff.com'
+  : 'http://localhost:3000'
+export const BACKEND_URL = IS_PROD
+  ? 'https://manicode-backend.onrender.com'
+  : 'http://localhost:4242'