Function retrieveWorkspaceMemory

runtime/api-server/src/workspace-memory.ts:362–476

(params: {
  store: RuntimeStateStore;
  workspaceId: string;
  query: string;
  intent?: string | null;
  categories?: WorkspaceMemoryCategory[] | null;
  treeIds?: string[] | null;
  retrievalPolicy?: MemoryRetrievalPolicy | null;
  answerGoal?: string | null;
  selectedModel?: string | null;
  sessionId?: string | null;
  inputId?: string | null;
  executionProfile?: WorkspaceMemoryExecutionProfile | null;
})

}

export async function retrieveWorkspaceMemory(params: {
  store: RuntimeStateStore;
  workspaceId: string;
  query: string;
  intent?: string | null;
  categories?: WorkspaceMemoryCategory[] | null;
  treeIds?: string[] | null;
  retrievalPolicy?: MemoryRetrievalPolicy | null;
  answerGoal?: string | null;
  selectedModel?: string | null;
  sessionId?: string | null;
  inputId?: string | null;
  executionProfile?: WorkspaceMemoryExecutionProfile | null;
}): Promise<WorkspaceMemoryRetrieveResult> {
  const useEmbeddings = params.executionProfile?.useEmbeddings !== false;
  const useLlmRerank = params.executionProfile?.useLlmRerank !== false;
  const categories = planWorkspaceMemoryCategories({
    store: params.store,
    workspaceId: params.workspaceId,
    requestedCategories: params.categories ?? undefined,
    treeId: Array.isArray(params.treeIds) && params.treeIds[0] ? params.treeIds[0] : null,
  });
  const candidateLimit = Math.max(
    VECTOR_FIRST_PASS_LIMIT_FLOOR,
    Math.min((params.retrievalPolicy?.max_evidence ?? 8) * 3, VECTOR_FIRST_PASS_LIMIT_CEILING),
  );
  const vectorFirstPass = useEmbeddings
    ? await buildWorkspaceVectorFirstPassHits({
        store: params.store,
        workspaceId: params.workspaceId,
        query: params.query,
        categories,
        treeIds: params.treeIds ?? null,
        selectedModel: params.selectedModel ?? null,
        sessionId: params.sessionId ?? null,
        inputId: params.inputId ?? null,
        maxCandidates: candidateLimit,
      })
    : {
        modelId: null,
        interactionHits: [],
        integrationHits: [],
      };
  const lexicalSupportLimit = Math.max(
    LEXICAL_SUPPORT_LIMIT_FLOOR,
    Math.min(params.retrievalPolicy?.max_evidence ?? 8, LEXICAL_SUPPORT_LIMIT_CEILING),
  );
  const interactionSupportScopes = categories.includes("interaction")
    ? distinctTreeIds(vectorFirstPass.interactionHits, params.treeIds ?? null, "interaction:")
    : [];
  const integrationSupportScopes = categories.includes("integration")
    ? distinctTreeIds(vectorFirstPass.integrationHits, params.treeIds ?? null, "integration:")
    : [];
  const interactionSupportResults = categories.includes("interaction")
    ? await Promise.all(interactionSupportScopes.map(async (treeId) =>
      retrieveInteractionMemory({
        store: params.store,
        workspaceId: params.workspaceId,