Function injectChat

packages/ai-angular/src/inject-chat.ts:37–255

(
  options: InjectChatOptions<
    TTools,
    TSchema,
    TContext
  > = {} as InjectChatOptions<TTools, TSchema, TContext>,
)

let nextId = 0

export function injectChat<
  TTools extends ReadonlyArray<AnyClientTool> = any,
  TSchema extends SchemaInput | undefined = undefined,
  TContext = InferredClientContext<TTools>,
>(
  options: InjectChatOptions<
    TTools,
    TSchema,
    TContext
  > = {} as InjectChatOptions<TTools, TSchema, TContext>,
): InjectChatResult<TTools, TSchema> {
  assertInInjectionContext(injectChat)

  type Partial = DeepPartial<InferSchemaType<NonNullable<TSchema>>>
  type Final = InferSchemaType<NonNullable<TSchema>>

  const destroyRef = inject(DestroyRef)
  const injector = inject(Injector)
  const clientId = options.id || `injectChat-${nextId++}`

  const messages = signal<Array<UIMessage<TTools>>>(
    options.initialMessages || [],
  )
  const isLoading = signal(false)
  const error = signal<Error | undefined>(undefined)
  const status = signal<ChatClientState>('ready')
  const isSubscribed = signal(false)
  const connectionStatus = signal<ConnectionStatus>('disconnected')
  const sessionGenerating = signal(false)

  // Reactive option sources. Plain values become constant computeds.
  const bodySource =
    options.body !== undefined ? toReactive(options.body) : undefined
  const forwardedPropsSource =
    options.forwardedProps !== undefined
      ? toReactive(options.forwardedProps)
      : undefined
  const contextSource =
    options.context !== undefined ? toReactive(options.context) : undefined
  const liveSource =
    options.live !== undefined ? toReactive(options.live) : undefined

  const transport = options.connection
    ? { connection: options.connection }
    : { fetcher: options.fetcher }

  const client = new ChatClient<TTools, TContext>({
    devtoolsBridgeFactory: createChatDevtoolsBridge,
    ...transport,
    id: clientId,
    ...(options.initialMessages !== undefined && {
      initialMessages: options.initialMessages,
    }),
    ...(options.persistence !== undefined && {
      persistence: options.persistence,
    }),
    ...(bodySource !== undefined && { body: bodySource() }),
    ...(options.threadId !== undefined && { threadId: options.threadId }),