cherry-studio/src/main/utils/knowledge.ts

export const DEFAULT_DOCUMENT_COUNT = 6
export const DEFAULT_RELEVANT_SCORE = 0
export type UrlSource = 'normal' | 'github' | 'youtube'

const youtubeRegex = /^(https?:\/\/)?(www\.)?(youtube\.com|youtu\.be|youtube\.be|yt\.be)/i

export function getUrlSource(url: string): UrlSource {
  if (youtubeRegex.test(url)) {
    return 'youtube'
  } else {
    return 'normal'
  }
}