add "FINISH" action to /api/v1/agent-runs

Author: charleslien

backend/src/agent-run.ts

@@ -85,7 +85,7 @@ export async function finishAgentRun(
  * Adds a completed step to the agent_step table
  */
 export async function addAgentStep(
-  params: Parameters<AddAgentStepFn>[0],
+  params: ParamsOf<AddAgentStepFn>,
 ): ReturnType<AddAgentStepFn> {
   const {
     userId,

common/src/constants/analytics-events.ts

@@ -86,8 +86,10 @@ export enum AnalyticsEvent {
   // Web - API
   AGENT_RUN_API_REQUEST = 'api.agent_run_request',
   AGENT_RUN_CREATED = 'api.agent_run_created',
+  AGENT_RUN_COMPLETED = 'api.agent_run_completed',
   AGENT_RUN_VALIDATION_ERROR = 'api.agent_run_validation_error',
   AGENT_RUN_CREATION_ERROR = 'api.agent_run_creation_error',
+  AGENT_RUN_COMPLETION_ERROR = 'api.agent_run_completion_error',
   ME_API_REQUEST = 'api.me_request',
   ME_VALIDATION_ERROR = 'api.me_validation_error',
   CHAT_COMPLETIONS_REQUEST = 'api.chat_completions_request',

common/src/db/index.ts

@@ -4,7 +4,9 @@ import postgres from 'postgres'
 
 import * as schema from './schema'
 
+import type { CodebuffPgDatabase } from './types'
+
 const client = postgres(env.DATABASE_URL)
 
-export const db = drizzle(client, { schema })
+export const db: CodebuffPgDatabase = drizzle(client, { schema })
 export default db

common/src/db/types.ts

@@ -0,0 +1,8 @@
+import type * as schema from './schema'
+import type { PgDatabase } from 'drizzle-orm/pg-core'
+import type { PostgresJsQueryResultHKT } from 'drizzle-orm/postgres-js'
+
+export type CodebuffPgDatabase = PgDatabase<
+  PostgresJsQueryResultHKT,
+  typeof schema
+>