@lota-sdk/core 0.1.20 → 0.1.21

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lota-sdk/core",
-  "version": "0.1.20",
+  "version": "0.1.21",
   "type": "module",
   "main": "./src/index.ts",
   "types": "./src/index.ts",
@@ -32,11 +32,11 @@
     "@chat-adapter/slack": "^4.23.0",
     "@chat-adapter/state-ioredis": "^4.23.0",
     "@logtape/logtape": "^2.0.5",
-    "@lota-sdk/shared": "0.1.20",
-    "@mendable/firecrawl-js": "^4.17.0",
+    "@lota-sdk/shared": "0.1.21",
+    "@mendable/firecrawl-js": "^4.18.0",
     "@surrealdb/node": "^3.0.3",
-    "ai": "^6.0.137",
-    "bullmq": "^5.71.0",
+    "ai": "^6.0.141",
+    "bullmq": "^5.71.1",
     "chat": "^4.23.0",
     "cron-parser": "^5.5.0",
     "hono": "^4.12.9",

package/src/ai-gateway/ai-gateway.ts

@@ -349,7 +349,7 @@ function createAiGatewayFetch(extraParams?: AiGatewayExtraParams): typeof fetch
         ? injectAiGatewayExtraParamsRequestBody(bodyWithPromptCacheRetention, extraParams)
         : bodyWithPromptCacheRetention
 
-    const headers = new Headers(init?.headers as HeadersInit | undefined)
+    const headers = new Headers(init?.headers)
     if (extraParams !== undefined || (isAiGatewayOpenAIModelRequest(body) && hasAiGatewayPromptCacheRetention(body))) {
       // Bifrost only forwards provider-specific extra params when passthrough is enabled.
       headers.set(AI_GATEWAY_EXTRA_PARAMS_HEADER, 'true')
@@ -370,9 +370,7 @@ function createAiGatewayProvider(extraParams?: AiGatewayExtraParams) {
   return createOpenAI({
     baseURL,
     apiKey,
-    headers: {
-      [AI_GATEWAY_VIRTUAL_KEY_HEADER]: apiKey,
-    },
+    headers: { [AI_GATEWAY_VIRTUAL_KEY_HEADER]: apiKey },
     fetch: createAiGatewayFetch(extraParams),
   })
 }

package/src/queues/document-processor.queue.ts

@@ -1,4 +1,5 @@
 import { Queue, Worker } from 'bullmq'
+import type { QueueOptions } from 'bullmq'
 import type IORedis from 'ioredis'
 
 import type { chatLogger } from '../config/logger'
@@ -70,22 +71,24 @@ export function createDocumentProcessorQueueRuntime<TJob extends DocumentProcess
   enqueue: (job: TJob) => Promise<unknown>
   startWorker: (options?: { registerSignals?: boolean }) => WorkerHandle
 } {
+  type QueueShape = Queue<TJob, unknown, string, TJob, unknown, string>
+
   const queueName = params.queueName ?? DEFAULT_DOCUMENT_PROCESSOR_QUEUE
   const workerName = params.workerName ?? DEFAULT_WORKER_NAME
   const concurrency = params.concurrency ?? 10
   const lockDuration = params.lockDuration ?? 300_000
-  const jobName = 'process-document' as Parameters<Queue<TJob, unknown, string>['add']>[0]
-  const toQueueData = (job: TJob): Parameters<Queue<TJob, unknown, string>['add']>[1] =>
-    job as Parameters<Queue<TJob, unknown, string>['add']>[1]
-  let queue: Queue<TJob, unknown, string> | null = null
+  const jobName = 'process-document' as Parameters<QueueShape['add']>[0]
+  const toQueueData = (job: TJob): Parameters<QueueShape['add']>[1] => job
+  let queue: QueueShape | null = null
+  const getConnection = (): IORedis => params.getConnectionForBullMQ()
 
-  const getQueue = (): Queue<TJob, unknown, string> => {
+  const getQueue = (): QueueShape => {
     if (queue) {
       return queue
     }
 
-    queue = new Queue<TJob, unknown, string>(queueName, {
-      connection: params.getConnectionForBullMQ(),
+    queue = new Queue<TJob, unknown, string, TJob, unknown, string>(queueName, {
+      connection: getConnection() as QueueOptions['connection'],
       defaultJobOptions: { ...DEFAULT_JOB_RETENTION, attempts: 3, backoff: { type: 'exponential', delay: 1000 } },
     })
 
@@ -108,7 +111,7 @@ export function createDocumentProcessorQueueRuntime<TJob extends DocumentProcess
     startWorker: (options = {}) => {
       const { registerSignals = import.meta.main } = options
       const worker = new Worker(queueName, params.getWorkerPath(), {
-        connection: params.getConnectionForBullMQ(),
+        connection: getConnection() as QueueOptions['connection'],
         concurrency,
         lockDuration,
       })

package/src/queues/queue-factory.ts

@@ -1,5 +1,5 @@
 import { Queue, Worker } from 'bullmq'
-import type { Job, JobsOptions, WorkerOptions } from 'bullmq'
+import type { Job, JobsOptions, QueueOptions, WorkerOptions } from 'bullmq'
 import type IORedis from 'ioredis'
 
 import { serverLogger } from '../config/logger'
@@ -39,18 +39,20 @@ interface QueueFactoryConfigFile extends QueueFactoryConfigBase {
 export type QueueFactoryConfig<TJob> = QueueFactoryConfigInline<TJob> | QueueFactoryConfigFile
 
 export interface QueueFactory<TJob> {
-  getQueue: () => Queue<TJob, unknown, string>
+  getQueue: () => Queue<TJob, unknown, string, TJob, unknown, string>
   enqueue: (job: TJob, options?: JobsOptions) => Promise<void>
   startWorker: (options?: { registerSignals?: boolean }) => WorkerHandle
 }
 
 export function createQueueFactory<TJob>(config: QueueFactoryConfig<TJob>): QueueFactory<TJob> {
-  let _queue: Queue<TJob, unknown, string> | null = null
+  type QueueShape = Queue<TJob, unknown, string, TJob, unknown, string>
+
+  let _queue: QueueShape | null = null
   let _queueConnection: IORedis | null = null
 
-  const getConnection = () => config.connectionProvider?.() ?? getRedisConnectionForBullMQ()
+  const getConnection = (): IORedis => config.connectionProvider?.() ?? getRedisConnectionForBullMQ()
 
-  const getQueue = (): Queue<TJob, unknown, string> => {
+  const getQueue = (): QueueShape => {
     const connection = getConnection()
     const shouldRecreateQueue =
       _queue === null ||
@@ -66,8 +68,8 @@ export function createQueueFactory<TJob>(config: QueueFactoryConfig<TJob>): Queu
         })
       }
 
-      _queue = new Queue<TJob, unknown, string>(config.name, {
-        connection,
+      _queue = new Queue<TJob, unknown, string, TJob, unknown, string>(config.name, {
+        connection: connection as QueueOptions['connection'],
         defaultJobOptions: { ...DEFAULT_JOB_RETENTION, ...config.defaultJobOptions },
       })
       _queueConnection = connection
@@ -78,9 +80,8 @@ export function createQueueFactory<TJob>(config: QueueFactoryConfig<TJob>): Queu
     return _queue
   }
 
-  type QueueAdd = Queue<TJob, unknown, string>['add']
-  const jobName = config.jobName as Parameters<QueueAdd>[0]
-  const toData = (job: TJob) => job as Parameters<QueueAdd>[1]
+  const jobName = config.jobName
+  const toData = (job: TJob) => job
 
   const enqueue = async (job: TJob, options?: JobsOptions): Promise<void> => {
     const queuedJob = await getQueue().add(jobName, toData(job), options)
@@ -99,7 +100,7 @@ export function createQueueFactory<TJob>(config: QueueFactoryConfig<TJob>): Queu
     const { registerSignals = import.meta.main } = options
 
     const workerOptions: WorkerOptions = {
-      connection: getConnection(),
+      connection: getConnection() as QueueOptions['connection'],
       concurrency: config.concurrency,
       ...(config.lockDuration !== undefined ? { lockDuration: config.lockDuration } : {}),
       ...(config.stalledInterval !== undefined ? { stalledInterval: config.stalledInterval } : {}),