flare-chat-core 0.2.1 → 0.2.2

package/src/chat-core/pipelines/transport/send-sourcing-search.js

@@ -0,0 +1,86 @@
+import { parseSourcingCandidates } from '../../../contracts/sse-events.js';
+import { resolvePayloadExtra, resolveStreamRefs } from '../build-stream-request.js';
+import { resolvePersistExchangeResult } from '../stream-persist-utils.js';
+import { buildSourcingResponseText } from './sourcing-response-templates.js';
+import { requestSourcingSearchApi } from './sourcing-search-api.js';
+
+const SOURCING_REFRESH_COMMANDS = new Set(['换一批', '继续', '继续寻源']);
+
+function normalizeText(value) {
+  return String(value || '').trim();
+}
+
+export function resolveSourcingSearchQuery(content, payloadExtra = {}) {
+  const normalizedContent = normalizeText(content);
+  const payloadSourcingQuery = normalizeText(payloadExtra?.sourcing_query);
+  if (payloadSourcingQuery && SOURCING_REFRESH_COMMANDS.has(normalizedContent)) {
+    return payloadSourcingQuery;
+  }
+  return normalizedContent || payloadSourcingQuery;
+}
+
+export async function requestSourcingSearch({
+  apiBaseUrl,
+  content,
+  options,
+  scope,
+  handlers,
+  persistExchange,
+  resolvedSessionId,
+}) {
+  const payloadExtra = resolvePayloadExtra(options);
+  const resolvedQuery = resolveSourcingSearchQuery(content, payloadExtra);
+  const streamRefs = resolveStreamRefs(options);
+  const sourceIds = Array.isArray(streamRefs.knowledge_refs)
+    ? streamRefs.knowledge_refs.map((item) => String(item || '').trim()).filter(Boolean)
+    : [];
+  const appendMode = payloadExtra.append_mode === true;
+  const { payload, results, sourceBreakdown, sourcingPayload } = await requestSourcingSearchApi({
+    apiBaseUrl,
+    projectId: String(scope?.projectId || '').trim() || 'default',
+    userId: String(scope?.userId || '').trim() || 'default',
+    sessionId: String(resolvedSessionId || '').trim() || 'default',
+    query: resolvedQuery,
+    topK: Number.isFinite(Number(payloadExtra.top_k)) ? Number(payloadExtra.top_k) : 8,
+    appendMode,
+    lockExistingOrder: payloadExtra.lock_existing_order !== false,
+    baseResultIds: Array.isArray(payloadExtra.base_result_ids) ? payloadExtra.base_result_ids : [],
+    sourceScope: String(payloadExtra.source_scope || 'hybrid').trim() || 'hybrid',
+    contextRefs: Array.isArray(streamRefs.context_refs) ? streamRefs.context_refs : [],
+    knowledgeRefs: sourceIds,
+    sourceIds,
+  });
+  handlers.onSourcingCandidates?.(parseSourcingCandidates(sourcingPayload));
+
+  const sourceSummary = `来源分布：本地资料 ${Number(sourceBreakdown.local) || 0} / 外部连接 ${Number(sourceBreakdown.mcp) || 0} / 网页来源 ${Number(sourceBreakdown.web) || 0}`;
+  const citableResults = results
+    .filter((item) => String(item?.canonical_url || item?.url || '').trim())
+    .slice(0, 2);
+  const topRefs = citableResults
+    .map((item, index) => `- [来源${index + 1}] ${String(item?.source_title || item?.title || '未命名来源')}（${String(item?.canonical_url || item?.url || '')}）`)
+    .join('\n');
+
+  const finalContent = buildSourcingResponseText({
+    count: results.length,
+    sourceBreakdownText: sourceSummary,
+    topRefsText: topRefs,
+    nextActionText: '继续补充你的目标与约束，我会基于当前上下文递进处理。',
+  });
+
+  if (appendMode) {
+    handlers.onComplete?.(finalContent, { sessionId: String(resolvedSessionId || '').trim() });
+    return;
+  }
+
+  let finalSessionId = String(resolvedSessionId || '').trim();
+  if (typeof persistExchange === 'function') {
+    const persistedResult = persistExchange(finalSessionId, content, finalContent, {
+      functionType: 'intelligent_sourcing',
+      projectId: String(scope?.projectId || '').trim() || null,
+      userId: String(scope?.userId || '').trim() || null,
+    });
+    const persistedSession = resolvePersistExchangeResult(persistedResult, finalSessionId);
+    finalSessionId = persistedSession.sessionId;
+  }
+  handlers.onComplete?.(finalContent, { sessionId: finalSessionId });
+}

package/src/chat-core/pipelines/transport/send-sourcing-search.test.mjs

@@ -0,0 +1,14 @@
+import test from 'node:test';
+import assert from 'node:assert/strict';
+import { resolveSourcingSearchQuery } from './send-sourcing-search.js';
+
+test('resolveSourcingSearchQuery uses payload sourcing_query for refresh commands', () => {
+  const query = resolveSourcingSearchQuery('换一批', { sourcing_query: '露营地 上海 亲子' });
+  assert.equal(query, '露营地 上海 亲子');
+});
+
+test('resolveSourcingSearchQuery keeps direct semantic input', () => {
+  const query = resolveSourcingSearchQuery('上海露营地推荐', { sourcing_query: '旧查询' });
+  assert.equal(query, '上海露营地推荐');
+});
+

package/src/chat-core/pipelines/transport/sourcing-response-templates.js

@@ -0,0 +1,55 @@
+const SOURCING_RESPONSE_TEMPLATES = {
+  sufficient_hits: [
+    '已完成寻源：共 {count} 条候选。',
+    '{source_breakdown}',
+    '{top_refs}',
+    '{next_action}',
+  ],
+  limited_hits: [
+    '已完成初步寻源：当前仅命中 {count} 条候选。',
+    '{source_breakdown}',
+    '{top_refs}',
+    '建议：补充更具体的地区/行业/预算关键词后，我可以继续扩展检索。',
+    '{next_action}',
+  ],
+  no_hits: [
+    '本轮未命中有效候选。',
+    '{source_breakdown}',
+    '可能原因：关键词过宽或外部通道未返回有效内容。',
+    '建议：补充地区/行业词，或检查外部网络与网关配置后重试。',
+    '{next_action}',
+  ],
+};
+
+function renderTemplate(templateLines, variables = {}) {
+  const lines = Array.isArray(templateLines) ? templateLines : [];
+  return lines
+    .map((line) => String(line || '').replace(/\{([^}]+)\}/g, (_, key) => String(variables[key] || '')))
+    .map((line) => line.trim())
+    .filter(Boolean)
+    .join('\n');
+}
+
+export function buildSourcingResponseText({
+  count = 0,
+  sourceBreakdownText = '',
+  topRefsText = '',
+  nextActionText = '你可以直接回复序号或补充信息，我来继续整理。',
+} = {}) {
+  const normalizedCount = Number.isFinite(Number(count)) ? Number(count) : 0;
+  const variables = {
+    count: normalizedCount,
+    source_breakdown: sourceBreakdownText,
+    top_refs: topRefsText || '暂无可展示引用来源。',
+    next_action: nextActionText,
+  };
+  if (normalizedCount <= 0) {
+    return renderTemplate(SOURCING_RESPONSE_TEMPLATES.no_hits, variables);
+  }
+  if (normalizedCount <= 2) {
+    return renderTemplate(SOURCING_RESPONSE_TEMPLATES.limited_hits, variables);
+  }
+  return renderTemplate(SOURCING_RESPONSE_TEMPLATES.sufficient_hits, variables);
+}
+
+export { SOURCING_RESPONSE_TEMPLATES };

package/src/chat-core/pipelines/transport/sourcing-search-api.js

@@ -0,0 +1,155 @@
+import { resolveEntities } from '../entity-extraction.js';
+
+function joinUrl(baseUrl = '', path = '') {
+  if (!baseUrl) {
+    return path || '';
+  }
+  if (!path) {
+    return baseUrl;
+  }
+  return `${baseUrl.replace(/\/+$/, '')}/${path.replace(/^\/+/, '')}`;
+}
+
+function normalizeSourceType(value) {
+  const rawType = String(value || '').trim().toLowerCase();
+  if (rawType.includes('web')) {
+    return 'web';
+  }
+  if (rawType.includes('mcp')) {
+    return 'mcp';
+  }
+  return 'local';
+}
+
+function normalizeSourceBreakdown(payload = {}) {
+  const sourceBreakdown = (
+    payload?.source_breakdown
+    && typeof payload.source_breakdown === 'object'
+    && !Array.isArray(payload.source_breakdown)
+  ) ? payload.source_breakdown : {};
+  return {
+    local: Number(sourceBreakdown.local) || 0,
+    mcp: Number(sourceBreakdown.mcp) || 0,
+    web: Number(sourceBreakdown.web) || 0,
+  };
+}
+
+function mapResultToCandidate(item = {}, runId = '', resolvedQuery = '') {
+  return {
+    id: String(item?.result_id || item?.record_id || item?.title || ''),
+    result_id: String(item?.result_id || item?.record_id || item?.title || ''),
+    title: String(item?.source_title || item?.title || ''),
+    supplier_name: String(item?.source_title || item?.title || ''),
+    evidence_kind: 'sourcing',
+    match_score: Number(item?.score || 0),
+    match_reasons: [String(item?.snippet || '')],
+    evidence_refs: [
+      {
+        url: String(item?.canonical_url || item?.url || ''),
+        snippet: String(item?.snippet || ''),
+        title: String(item?.source_title || item?.title || ''),
+      },
+    ],
+    source_type: normalizeSourceType(item?.source_type || item?.source),
+    source_title: String(item?.source_title || item?.title || ''),
+    source_domain: String(item?.source_domain || ''),
+    source_meta: (
+      item?.source_meta
+      && typeof item.source_meta === 'object'
+      && !Array.isArray(item.source_meta)
+    ) ? item.source_meta : {},
+    canonical_url: String(item?.canonical_url || item?.url || ''),
+    confidence: item?.confidence ?? '',
+    confidence_detail: (
+      item?.confidence_detail
+      && typeof item.confidence_detail === 'object'
+      && !Array.isArray(item.confidence_detail)
+    ) ? item.confidence_detail : null,
+    summary: String(item?.snippet || ''),
+    run_id: runId,
+    query: resolvedQuery,
+  };
+}
+
+export async function requestSourcingSearchApi({
+  apiBaseUrl,
+  projectId = 'default',
+  userId = 'default',
+  sessionId = 'default',
+  query,
+  topK = 8,
+  sourceScope = 'hybrid',
+  contextRefs = [],
+  knowledgeRefs = [],
+  sourceIds = [],
+  entities = [],
+  appendMode = false,
+  lockExistingOrder = true,
+  baseResultIds = [],
+}) {
+  const resolvedEntities = resolveEntities({ text: query, entities });
+  const response = await fetch(joinUrl(apiBaseUrl, '/kernel/sourcing/search'), {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({
+      project_id: String(projectId || '').trim() || 'default',
+      user_id: String(userId || '').trim() || 'default',
+      session_id: String(sessionId || '').trim() || 'default',
+      query: String(query || '').trim(),
+      top_k: Number.isFinite(Number(topK)) ? Number(topK) : 8,
+      append_mode: appendMode === true,
+      lock_existing_order: lockExistingOrder !== false,
+      base_result_ids: Array.isArray(baseResultIds)
+        ? baseResultIds.map((item) => String(item || '').trim()).filter(Boolean)
+        : [],
+      source_scope: String(sourceScope || 'hybrid').trim() || 'hybrid',
+      sourcing_enabled: true,
+      context_refs: Array.isArray(contextRefs) ? contextRefs : [],
+      knowledge_refs: Array.isArray(knowledgeRefs) ? knowledgeRefs : [],
+      source_ids: Array.isArray(sourceIds) ? sourceIds : [],
+      entities: resolvedEntities,
+    }),
+  });
+
+  if (!response.ok) {
+    throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+  }
+
+  const payload = await response.json();
+  const runId = String(payload?.retrieval_run_id || '').trim();
+  const resolvedQuery = String(payload?.query || query).trim();
+  const results = Array.isArray(payload?.results) ? payload.results : [];
+  const newResults = Array.isArray(payload?.new_results) ? payload.new_results : [];
+  const allResults = Array.isArray(payload?.all_results) ? payload.all_results : [];
+  const sourceBreakdown = normalizeSourceBreakdown(payload);
+  const sourcingPayload = {
+    run_id: runId,
+    query: resolvedQuery,
+    candidate_count: results.length,
+    requested_top_k: Number(payload?.requested_top_k) || Number(topK) || 8,
+    returned_count: Number(payload?.returned_count) || results.length,
+    has_more: payload?.has_more === true,
+    next_top_k: Number(payload?.next_top_k) || Number(topK) || 8,
+    append_mode: appendMode === true,
+    lock_existing_order: lockExistingOrder !== false,
+    candidates: results.map((item) => mapResultToCandidate(item, runId, resolvedQuery)),
+    new_results: newResults.map((item) => mapResultToCandidate(item, runId, resolvedQuery)),
+    all_results: allResults.map((item) => mapResultToCandidate(item, runId, resolvedQuery)),
+    shortlist: Array.isArray(payload?.shortlist) ? payload.shortlist : [],
+    source_meta: {
+      source_breakdown: sourceBreakdown,
+    },
+    summary: '',
+  };
+
+  return {
+    payload,
+    runId,
+    resolvedQuery,
+    results,
+    sourceBreakdown,
+    sourcingPayload,
+  };
+}
+
+export { normalizeSourceType, joinUrl };

package/src/chat-core/runtime/runtimeMode.js

@@ -0,0 +1,69 @@
+/**
+ * Core runtime mode + debug gating.
+ *
+ * Env contract:
+ * - VITE_FLARE_CHAT_RUNTIME_MODE: optional override, "development" | "production"
+ * - VITE_FLARE_CHAT_DEBUG: optional debug switch, "1" | "true" enables debug logs
+ *
+ * Safety rule:
+ * - Debug logs are allowed only when runtime mode is non-production.
+ */
+
+function toBooleanFlag(value) {
+  const normalized = String(value ?? '').trim().toLowerCase();
+  return normalized === '1' || normalized === 'true' || normalized === 'yes' || normalized === 'on';
+}
+
+function normalizeMode(value) {
+  const normalized = String(value ?? '').trim().toLowerCase();
+  if (normalized === 'production' || normalized === 'prod') {
+    return 'production';
+  }
+  if (normalized === 'development' || normalized === 'dev') {
+    return 'development';
+  }
+  return '';
+}
+
+export function resolveRuntimeMode(envInput) {
+  const env = (
+    envInput
+    && typeof envInput === 'object'
+    && !Array.isArray(envInput)
+  ) ? envInput : (import.meta?.env || {});
+
+  const overrideMode = normalizeMode(env.VITE_FLARE_CHAT_RUNTIME_MODE);
+  const modeFromVite = normalizeMode(env.MODE);
+  const fallbackMode = env.PROD === true ? 'production' : 'development';
+  const mode = overrideMode || modeFromVite || fallbackMode;
+  const isProduction = mode === 'production';
+  const isDevelopment = !isProduction;
+  const debugRequested = toBooleanFlag(env.VITE_FLARE_CHAT_DEBUG);
+  const debugEnabled = isDevelopment && debugRequested;
+
+  return {
+    mode,
+    isProduction,
+    isDevelopment,
+    debugEnabled,
+  };
+}
+
+const runtimeMode = resolveRuntimeMode();
+
+export function getCoreRuntimeMode() {
+  return runtimeMode;
+}
+
+export function isCoreDebugEnabled() {
+  return runtimeMode.debugEnabled;
+}
+
+export function debugCoreLog(...args) {
+  if (!runtimeMode.debugEnabled) {
+    return;
+  }
+  console.debug('[flare-chat-core:debug]', ...args);
+}
+
+export default getCoreRuntimeMode;

package/src/chat-core/session/chatSessionActionTypes.js

@@ -0,0 +1,24 @@
+export const SESSION_LOADED = 'SESSION_LOADED';
+export const SESSION_RESET = 'SESSION_RESET';
+export const SESSION_ERROR = 'SESSION_ERROR';
+export const TITLE_UPDATED = 'TITLE_UPDATED';
+export const MESSAGES_REFRESHED = 'MESSAGES_REFRESHED';
+export const MESSAGE_APPENDED = 'MESSAGE_APPENDED';
+export const STREAMING_RESET = 'STREAMING_RESET';
+export const STREAMING_CHUNK = 'STREAMING_CHUNK';
+export const STREAMING_REPLACE = 'STREAMING_REPLACE';
+export const AGENT_STATUS_SET = 'AGENT_STATUS_SET';
+export const THINKING_TRACE_SET = 'THINKING_TRACE_SET';
+export const EXECUTION_TRACE_SET = 'EXECUTION_TRACE_SET';
+export const STREAM_KNOWLEDGE_SEARCH_SET = 'STREAM_KNOWLEDGE_SEARCH_SET';
+export const STREAM_SOURCING_CANDIDATES_SET = 'STREAM_SOURCING_CANDIDATES_SET';
+export const STREAM_KNOWLEDGE_CITATION_SET = 'STREAM_KNOWLEDGE_CITATION_SET';
+export const STREAMING_DONE = 'STREAMING_DONE';
+export const DOC_GENERATED = 'DOC_GENERATED';
+export const LEGEND_HINT_SHOWN = 'LEGEND_HINT_SHOWN';
+export const EXECUTION_CARD_UPSERTED = 'EXECUTION_CARD_UPSERTED';
+export const UI_CARDS_SET = 'UI_CARDS_SET';
+export const LAST_USER_MESSAGE_SET = 'LAST_USER_MESSAGE_SET';
+export const UI_CARD_UPDATED = 'UI_CARD_UPDATED';
+export const INSTANCE_PROFILE_SET = 'INSTANCE_PROFILE_SET';
+export const SESSION_PROMOTED = 'SESSION_PROMOTED';