Add search and RAG workspace flows

lib/query/options.ts

@@ -5,6 +5,7 @@ import {
   getCompanyFinancialStatements,
   getLatestPortfolioInsight,
   getPortfolioSummary,
+  searchKnowledge,
   getTask,
   getTaskTimeline,
   listFilings,
@@ -16,7 +17,8 @@ import {
 import { queryKeys } from '@/lib/query/keys';
 import type {
   FinancialCadence,
-  FinancialSurfaceKind
+  FinancialSurfaceKind,
+  SearchSource
 } from '@/lib/types';
 
 export function companyAnalysisQueryOptions(ticker: string) {
@@ -86,6 +88,31 @@ export function filingsQueryOptions(input: { ticker?: string; limit?: number } =
   });
 }
 
+export function searchQueryOptions(input: {
+  query: string;
+  ticker?: string | null;
+  sources?: SearchSource[];
+  limit?: number;
+}) {
+  const normalizedQuery = input.query.trim();
+  const normalizedTicker = input.ticker?.trim().toUpperCase() ?? null;
+  const sources = input.sources && input.sources.length > 0
+    ? [...new Set(input.sources)]
+    : ['documents', 'filings', 'research'] as SearchSource[];
+  const limit = input.limit ?? 10;
+
+  return queryOptions({
+    queryKey: queryKeys.search(normalizedQuery, normalizedTicker, sources, limit),
+    queryFn: () => searchKnowledge({
+      query: normalizedQuery,
+      ticker: normalizedTicker ?? undefined,
+      sources,
+      limit
+    }),
+    staleTime: 30_000
+  });
+}
+
 export function aiReportQueryOptions(accessionNumber: string) {
   const normalizedAccession = accessionNumber.trim();