mirror of
https://github.com/deepseek-ai/deepseek-harness
synced 2026-08-15 21:04:50 +00:00
62 lines
2.4 KiB
TypeScript
62 lines
2.4 KiB
TypeScript
/**
|
|
* `@deepseek-ai/dsh-web-search-perplexity`: registers a Perplexity-backed
|
|
* `WebSearchProvider` with `ctx.web`. A function/namespace plugin (NOT a
|
|
* default-export service): it registers INTO the seam's provider registry, like
|
|
* `@deepseek-ai/dsh-llm-deepseek` registers an adapter into `ctx.llm`.
|
|
*
|
|
* @module @deepseek-ai/dsh-web-search-perplexity
|
|
*/
|
|
|
|
import type { Context } from 'cordis'
|
|
import z from 'schemastery'
|
|
import type {} from '@deepseek-ai/dsh-web'
|
|
import { PerplexitySearchProvider, PERPLEXITY_DEFAULT_BASE_URL, PERPLEXITY_DEFAULT_MAX_TOKENS, PERPLEXITY_DEFAULT_MODEL } from './provider.ts'
|
|
|
|
export {
|
|
PERPLEXITY_DEFAULT_BASE_URL,
|
|
PERPLEXITY_DEFAULT_MAX_TOKENS,
|
|
PERPLEXITY_DEFAULT_MODEL,
|
|
PERPLEXITY_PROVIDER_ID,
|
|
PerplexitySearchProvider,
|
|
} from './provider.ts'
|
|
export type { PerplexityRecency, PerplexitySearchProviderOptions } from './provider.ts'
|
|
|
|
/** Cordis plugin name used by loader diagnostics. */
|
|
export const name = 'web-search-perplexity'
|
|
|
|
/** The web seam this provider registers into. */
|
|
export const inject = ['web']
|
|
|
|
/** Plugin config (all optional — `apply` fills env-var and constant defaults). */
|
|
export interface Config {
|
|
/** Perplexity API key. Falls back to `$PERPLEXITY_API_KEY`. Empty → unavailable. */
|
|
apiKey?: string
|
|
/** Endpoint base; `/chat/completions` is appended. Defaults to the public API. */
|
|
baseURL?: string
|
|
/** Search model name. Defaults to `sonar`. */
|
|
model?: string
|
|
/** Upper bound on generated answer tokens. Defaults to 1024. */
|
|
maxTokens?: number
|
|
/** Recency window sent as `search_recency_filter`. Omitted = no filter. */
|
|
searchRecency?: 'day' | 'week' | 'month' | 'year'
|
|
}
|
|
|
|
export const Config: z<Config> = z.object({
|
|
apiKey: z.string(),
|
|
baseURL: z.string(),
|
|
model: z.string(),
|
|
maxTokens: z.number().step(1).min(1),
|
|
searchRecency: z.union(['day', 'week', 'month', 'year'] as const),
|
|
})
|
|
|
|
/** Register the Perplexity search provider with `ctx.web`. */
|
|
export function apply(ctx: Context, config: Config): void {
|
|
ctx.web.registerSearchProvider(new PerplexitySearchProvider({
|
|
apiKey: config.apiKey ?? process.env.PERPLEXITY_API_KEY ?? '',
|
|
baseURL: config.baseURL ?? PERPLEXITY_DEFAULT_BASE_URL,
|
|
model: config.model ?? PERPLEXITY_DEFAULT_MODEL,
|
|
maxTokens: config.maxTokens ?? PERPLEXITY_DEFAULT_MAX_TOKENS,
|
|
...config.searchRecency !== undefined ? { searchRecency: config.searchRecency } : {},
|
|
}))
|
|
}
|