add generateText endpoint to openrouter wrapper

Author: charleslien

common/src/constants/analytics-events.ts

@@ -96,8 +96,8 @@ export enum AnalyticsEvent {
   CHAT_COMPLETIONS_AUTH_ERROR = 'api.chat_completions_auth_error',
   CHAT_COMPLETIONS_VALIDATION_ERROR = 'api.chat_completions_validation_error',
   CHAT_COMPLETIONS_INSUFFICIENT_CREDITS = 'api.chat_completions_insufficient_credits',
+  CHAT_COMPLETIONS_GENERATION_STARTED = 'api.chat_completions_generation_started',
   CHAT_COMPLETIONS_STREAM_STARTED = 'api.chat_completions_stream_started',
-  CHAT_COMPLETIONS_STREAM_ERROR = 'api.chat_completions_stream_error',
   CHAT_COMPLETIONS_ERROR = 'api.chat_completions_error',
 
   // Common

common/src/types/contracts/bigquery.ts

@@ -0,0 +1,23 @@
+import type { Logger } from './logger'
+
+export type MessageRow = {
+  id: string
+  user_id: string
+  finished_at: Date
+  created_at: Date
+  request: unknown
+  reasoning_text: string
+  response: string
+  output_tokens?: number | null
+  reasoning_tokens?: number | null
+  cost?: number | null
+  upstream_inference_cost?: number | null
+  input_tokens?: number | null
+  cache_read_input_tokens?: number | null
+}
+
+export type InsertMessageBigqueryFn = (params: {
+  row: MessageRow
+  dataset?: string
+  logger: Logger
+}) => Promise<boolean>

common/src/types/contracts/database.ts

@@ -35,7 +35,7 @@ export type GetAgentRunFromIdOutput<T extends AgentRunColumn> = Promise<
   | {
       [K in T]: AgentRun[K]
     }
-  | undefined
+  | null
 >
 export type GetAgentRunFromIdFn = <T extends AgentRunColumn>(
   params: GetAgentRunFromIdInput<T>,

packages/bigquery/src/client.ts

@@ -3,13 +3,8 @@ import { BigQuery } from '@google-cloud/bigquery'
 
 import { MESSAGE_SCHEMA, RELABELS_SCHEMA, TRACES_SCHEMA } from './schema'
 
-import type {
-  BaseTrace,
-  GetRelevantFilesTrace,
-  MessageRow,
-  Relabel,
-  Trace,
-} from './schema'
+import type { BaseTrace, GetRelevantFilesTrace, Relabel, Trace } from './schema'
+import type { MessageRow } from '@codebuff/common/types/contracts/bigquery'
 import type { Logger } from '@codebuff/common/types/contracts/logger'
 
 const DATASET =
@@ -99,7 +94,7 @@ export async function setupBigQuery({
   }
 }
 
-export async function insertMessage({
+export async function insertMessageBigquery({
   row,
   dataset = DATASET,
   logger,

packages/bigquery/src/schema.ts

@@ -126,22 +126,6 @@ export const RELABELS_SCHEMA: TableSchema = {
   ],
 }
 
-export type MessageRow = {
-  id: string
-  user_id: string
-  finished_at: Date
-  created_at: Date
-  request: unknown
-  reasoning_text: string
-  response: string
-  output_tokens?: number | null
-  reasoning_tokens?: number | null
-  cost?: number | null
-  upstream_inference_cost?: number | null
-  input_tokens?: number | null
-  cache_read_input_tokens?: number | null
-}
-
 export const MESSAGE_SCHEMA: TableSchema = {
   fields: [
     { name: 'id', type: 'STRING', mode: 'REQUIRED' },

scripts/fat-sdk-openrouter-example.ts

@@ -1,14 +1,15 @@
 import { createOpenAICompatible } from '@ai-sdk/openai-compatible'
 import { websiteUrl } from '@codebuff/npm-app/config'
-import { streamText } from 'ai'
+import { generateText } from 'ai'
 
 const codebuffBackendProvider = createOpenAICompatible({
   name: 'codebuff',
   apiKey: '12345',
   baseURL: websiteUrl + '/api/v1',
 })
 
-const response = streamText({
+// const response = await streamText({
+const response = await generateText({
   model: codebuffBackendProvider('anthropic/claude-sonnet-4.5'),
   messages: [
     {
@@ -44,6 +45,8 @@ const response = streamText({
     },
   },
 })
-for await (const chunk of response.fullStream) {
-  console.log({ chunk })
-}
+
+console.dir({ response }, { depth: null })
+// for await (const chunk of response.fullStream) {
+//   console.dir({ chunk }, { depth: null })
+// }