@lota-sdk/core 0.1.15 → 0.1.17

package/src/workers/utils/workstream-message-query.ts

@@ -1,34 +1,20 @@
-import { toTimestamp } from '@lota-sdk/shared'
+import { requireTimestamp } from '@lota-sdk/shared'
 import { BoundQuery } from 'surrealdb'
 import { z } from 'zod'
 
-import { ensureRecordId } from '../../db/record-id'
+import { ensureRecordId, recordIdToString } from '../../db/record-id'
 import type { RecordIdRef } from '../../db/record-id'
 import { databaseService } from '../../db/service'
 import { TABLES } from '../../db/tables'
+import { WorkstreamMessageRowSchema } from '../../db/workstream-message-row'
+import type { WorkstreamMessageRow } from '../../db/workstream-message-row'
 import { normalizeTextBody } from '../../document/parsing'
+import type { LotaRuntimeBackgroundCursor } from '../../runtime/runtime-extensions'
 
-const RecordTimestampSchema = z.union([z.date(), z.string(), z.number()])
-const MessageRoleSchema = z.enum(['system', 'user', 'assistant'])
-const MessagePartSchema = z.record(z.string(), z.unknown())
-const MessageMetadataSchema = z.record(z.string(), z.unknown()).nullish()
-
-const WorkstreamMessageRowSchema = z.object({
-  id: z.string(),
-  workstreamId: z.string(),
-  role: MessageRoleSchema,
-  parts: z.array(MessagePartSchema).optional(),
-  metadata: MessageMetadataSchema,
-  createdAt: RecordTimestampSchema,
-})
-
-export interface DigestCursor {
-  createdAt: Date
-  id: string
-}
+export type DigestCursor = LotaRuntimeBackgroundCursor
 
 export interface DigestMessage {
-  source: 'workstream'
+  source: 'workstream' | 'social'
   sourceId: string
   role: 'system' | 'user' | 'assistant'
   parts: Array<Record<string, unknown>>
@@ -36,14 +22,17 @@ export interface DigestMessage {
   cursor: DigestCursor
 }
 
-function mapWorkstreamRow(row: z.infer<typeof WorkstreamMessageRowSchema>): DigestMessage {
+function mapWorkstreamRow(row: WorkstreamMessageRow): DigestMessage {
   return {
     source: 'workstream',
-    sourceId: row.workstreamId,
+    sourceId: recordIdToString(row.workstreamId, TABLES.WORKSTREAM),
     role: row.role,
-    parts: row.parts ?? [],
+    parts: row.parts as Array<Record<string, unknown>>,
     metadata: row.metadata ?? undefined,
-    cursor: { createdAt: new Date(toTimestamp(row.createdAt) ?? Date.now()), id: row.id },
+    cursor: {
+      createdAt: new Date(requireTimestamp(row.createdAt)),
+      id: recordIdToString(row.id, TABLES.WORKSTREAM_MESSAGE),
+    },
   }
 }
 

package/src/workers/worker-utils.ts

@@ -4,6 +4,7 @@ import { fileURLToPath } from 'node:url'
 import type { Job, Worker } from 'bullmq'
 
 import { chatLogger } from '../config/logger'
+import { truncateText } from '../utils/string'
 
 export const DEFAULT_JOB_RETENTION = { removeOnComplete: 200, removeOnFail: 200 }
 export const LOW_JOB_RETENTION = { removeOnComplete: 50, removeOnFail: 50 }
@@ -32,8 +33,7 @@ interface TracedWorkerJobLike {
 }
 
 function truncateTraceString(value: string, maxChars = MAX_TRACE_STRING_CHARS): string {
-  if (value.length <= maxChars) return value
-  return `${value.slice(0, maxChars - 3)}...`
+  return truncateText(value, maxChars)
 }
 
 function normalizeTraceValue(value: unknown, depth = 0): unknown {

package/src/runtime/workstream-routing-policy.ts

@@ -1,267 +0,0 @@
-import { extractAgentMentions, getLeadAgentId } from '../config/agent-defaults'
-
-export type MessageRoute =
-  | { type: 'direct'; agents: [string] }
-  | { type: 'mentions'; agents: string[] }
-  | { type: 'group-default'; agents: [string] }
-
-export type ReasoningProfileName = 'fast' | 'standard' | 'deep'
-
-export interface ReasoningProfile {
-  name: ReasoningProfileName
-  maxSteps: number
-  toolCallBudget: number
-  maxInputTokensHint: number
-}
-
-export type HighImpactResponseClass =
-  | 'architecture-recommendation'
-  | 'gtm-strategy'
-  | 'pricing-positioning'
-  | 'product-capability-claim'
-  | 'community-targeting'
-  | 'mutating-operation'
-
-export type PolicyClass = 'external-mutation' | 'security-privacy' | 'legal-compliance' | 'financial-decision'
-
-export function uniqueMentionOrder(message: string): string[] {
-  const ordered: string[] = []
-  const seen = new Set<string>()
-  for (const mention of extractAgentMentions(message)) {
-    const agent = mention.agent
-    if (seen.has(agent)) continue
-    seen.add(agent)
-    ordered.push(agent)
-  }
-  return ordered
-}
-
-const HIGH_IMPACT_CLASS_PATTERNS: Array<{ className: HighImpactResponseClass; patterns: RegExp[] }> = [
-  {
-    className: 'architecture-recommendation',
-    patterns: [
-      /\barchitecture\b/i,
-      /\btech(?:nical)?\s+stack\b/i,
-      /\binfra(?:structure)?\b/i,
-      /\brefactor\b/i,
-      /\bsystem\s+design\b/i,
-    ],
-  },
-  {
-    className: 'gtm-strategy',
-    patterns: [
-      /\bgo[-\s]?to[-\s]?market\b/i,
-      /\bgtm\b/i,
-      /\bdistribution\s+strategy\b/i,
-      /\blaunch\s+strategy\b/i,
-      /\bcontent\s+strategy\b/i,
-      /\bdemand\s+gen(?:eration)?\b/i,
-    ],
-  },
-  {
-    className: 'pricing-positioning',
-    patterns: [/\bpricing\b/i, /\bpositioning\b/i, /\bpackaging\b/i, /\bfreemium\b/i, /\bbilling\b/i],
-  },
-  {
-    className: 'product-capability-claim',
-    patterns: [
-      /\bdoes\s+it\s+(already\s+)?(support|have|do)\b/i,
-      /\bwhat\s+can\s+(it|we)\s+do\b/i,
-      /\bis\s+this\s+implemented\b/i,
-      /\bcurrent(?:ly)?\s+supports?\b/i,
-      /\bcapabilities?\b/i,
-    ],
-  },
-  {
-    className: 'community-targeting',
-    patterns: [
-      /\bwhere\s+to\s+post\b/i,
-      /\bcommunity\b/i,
-      /\btarget\s+audience\b/i,
-      /\bchannels?\b/i,
-      /\bdistribution\s+channels?\b/i,
-    ],
-  },
-  {
-    className: 'mutating-operation',
-    patterns: [
-      /\bdelete\s+(issue|comment|task|record|workspace|project)\b/i,
-      /\bremove\s+(issue|comment|task|record|workspace|project)\b/i,
-      /\bupdate\s+(issue|comment|task|record|workspace|project)\b/i,
-      /\bcreate\s+(issue|comment|task|record|workspace|project)\b/i,
-      /\bchange\s+(issue|comment|task|record|workspace|project)\b/i,
-      /\barchive\b/i,
-      /\bskip\s+onboarding\b/i,
-      /\bproceed\s+in\s+onboarding\b/i,
-    ],
-  },
-]
-
-const POLICY_CLASS_PATTERNS: Array<{ className: PolicyClass; patterns: RegExp[] }> = [
-  {
-    className: 'external-mutation',
-    patterns: [
-      /\bdelete\s+(issue|comment|task|record|workspace|project)\b/i,
-      /\bremove\s+(issue|comment|task|record|workspace|project)\b/i,
-      /\bupdate\s+(issue|comment|task|record|workspace|project)\b/i,
-      /\bcreate\s+(issue|comment|task|record|workspace|project)\b/i,
-      /\bchange\s+(issue|comment|task|record|workspace|project)\b/i,
-      /\barchive\b/i,
-      /\bskip\s+onboarding\b/i,
-      /\bproceed\s+in\s+onboarding\b/i,
-    ],
-  },
-  {
-    className: 'security-privacy',
-    patterns: [
-      /\bsecurity\b/i,
-      /\bprivacy\b/i,
-      /\bpii\b/i,
-      /\bauth(?:entication|orization)?\b/i,
-      /\bencrypt(?:ion|ed)?\b/i,
-      /\bpermission(?:s)?\b/i,
-      /\baccess\s+control\b/i,
-      /\bsecret(?:s)?\b/i,
-    ],
-  },
-  {
-    className: 'legal-compliance',
-    patterns: [
-      /\blegal\b/i,
-      /\bcompliance\b/i,
-      /\bregulation(?:s)?\b/i,
-      /\bgdpr\b/i,
-      /\bhipaa\b/i,
-      /\bsoc\s*2\b/i,
-      /\blicense\b/i,
-      /\bterms?\b/i,
-    ],
-  },
-  {
-    className: 'financial-decision',
-    patterns: [
-      /\bpricing\b/i,
-      /\bbilling\b/i,
-      /\brevenue\b/i,
-      /\bbudget\b/i,
-      /\bburn\s+rate\b/i,
-      /\brunway\b/i,
-      /\bfundrais(?:e|ing)\b/i,
-      /\bcash\s+flow\b/i,
-    ],
-  },
-]
-
-const REASONING_PROFILES: Record<ReasoningProfileName, ReasoningProfile> = {
-  fast: { name: 'fast', maxSteps: 5, toolCallBudget: 3, maxInputTokensHint: 12_000 },
-  standard: { name: 'standard', maxSteps: 10, toolCallBudget: 6, maxInputTokensHint: 24_000 },
-  deep: { name: 'deep', maxSteps: 15, toolCallBudget: 10, maxInputTokensHint: 36_000 },
-}
-
-const STRATEGIC_COMPLEXITY_PATTERNS: RegExp[] = [
-  /\bstrategy\b/i,
-  /\btrade[-\s]?off\b/i,
-  /\bcompare\b/i,
-  /\broadmap\b/i,
-  /\barchitecture\b/i,
-  /\bconstraints?\b/i,
-  /\bprioriti(?:y|es)\b/i,
-]
-
-const CROSS_SOURCE_COMPLEXITY_PATTERNS: RegExp[] = [
-  /\bcitation\b/i,
-  /\bevidence\b/i,
-  /\bsource\b/i,
-  /\bground(?:ed|ing)\b/i,
-  /\brepo\b/i,
-  /\bmemory\b/i,
-]
-
-const STANDARD_FLOOR_INTENT_PATTERNS: RegExp[] = [
-  /\bplan(?:ning)?\b/i,
-  /\bthink(?:ing)?\b/i,
-  /\bcareful(?:ly)?\b/i,
-  /\bimportant\b/i,
-  /\bcritical(?:ly)?\b/i,
-  /\bthorough(?:ly)?\b/i,
-]
-
-export function classifyHighImpactResponse(params: { message: string }): { classes: HighImpactResponseClass[] } {
-  const message = params.message.trim()
-  if (!message) return { classes: [] }
-
-  const classes: HighImpactResponseClass[] = []
-  for (const entry of HIGH_IMPACT_CLASS_PATTERNS) {
-    if (entry.patterns.some((pattern) => pattern.test(message))) {
-      classes.push(entry.className)
-    }
-  }
-
-  return { classes }
-}
-
-export function classifyPolicyClasses(params: { message: string }): { classes: PolicyClass[] } {
-  const message = params.message.trim()
-  if (!message) return { classes: [] }
-
-  const classes: PolicyClass[] = []
-  for (const entry of POLICY_CLASS_PATTERNS) {
-    if (entry.patterns.some((pattern) => pattern.test(message))) {
-      classes.push(entry.className)
-    }
-  }
-
-  return { classes }
-}
-
-export function resolveReasoningProfile(params: {
-  message: string
-  forceDeep?: boolean
-  explicitProfile?: ReasoningProfileName
-}): ReasoningProfile {
-  if (params.explicitProfile) {
-    return REASONING_PROFILES[params.explicitProfile]
-  }
-
-  const text = params.message.trim()
-  if (!text) {
-    return REASONING_PROFILES.standard
-  }
-
-  let score = 0
-  if (text.length >= 700) score += 3
-  else if (text.length >= 350) score += 2
-  else if (text.length >= 180) score += 1
-
-  const numberedListCount = (text.match(/\n\s*\d+\./g) ?? []).length
-  if (numberedListCount >= 3) score += 2
-  else if (numberedListCount >= 1) score += 1
-
-  if (STRATEGIC_COMPLEXITY_PATTERNS.some((pattern) => pattern.test(text))) score += 2
-  if (CROSS_SOURCE_COMPLEXITY_PATTERNS.some((pattern) => pattern.test(text))) score += 2
-  if (params.forceDeep) score += 2
-  if (STANDARD_FLOOR_INTENT_PATTERNS.some((pattern) => pattern.test(text))) {
-    score += 3
-  }
-
-  if (score >= 7) return REASONING_PROFILES.deep
-  if (score >= 3) return REASONING_PROFILES.standard
-  return REASONING_PROFILES.fast
-}
-
-export function resolveMessageRoute(params: {
-  workstreamMode: 'direct' | 'group'
-  workstreamAgentId?: string
-  message: string
-}): MessageRoute {
-  if (params.workstreamMode === 'direct') {
-    return { type: 'direct', agents: [params.workstreamAgentId ?? getLeadAgentId()] }
-  }
-
-  const mentions = uniqueMentionOrder(params.message)
-  if (mentions.length > 0) {
-    return { type: 'mentions', agents: mentions }
-  }
-
-  return { type: 'group-default', agents: [getLeadAgentId()] }
-}

package/src/tools/log-hello-world.tool.ts

@@ -1,17 +0,0 @@
-import { tool } from 'ai'
-import { z } from 'zod'
-
-export function createLogHelloWorldTool() {
-  return tool({
-    description: 'Logs "Hello World" to the server console. Requires user approval before running.',
-    inputSchema: z.object({
-      message: z.string().optional().describe('Optional custom message to log alongside Hello World'),
-    }),
-    needsApproval: true,
-    execute: async ({ message }: { message?: string }) => {
-      const output = message ? `Hello World: ${message}` : 'Hello World'
-      console.log('[logHelloWorld]', output)
-      return { logged: output, timestamp: new Date().toISOString() }
-    },
-  })
-}