@sourcepress/ai 0.1.0

package/src/types.ts

@@ -0,0 +1,225 @@
+import type { Entity } from "@sourcepress/core";
+
+export interface AIProviderConfig {
+	provider: "anthropic" | "openai" | "local";
+	model: string;
+	apiKey?: string;
+	baseUrl?: string;
+}
+
+export interface BudgetConfig {
+	daily_limit_usd: number;
+	warn_at_usd?: number;
+}
+
+export interface BudgetStatus {
+	spent_today_usd: number;
+	limit_usd: number;
+	warn_at_usd: number;
+	remaining_usd: number;
+	is_over_limit: boolean;
+	is_warned: boolean;
+	reset_at: string;
+}
+
+export interface TokenUsage {
+	input_tokens: number;
+	output_tokens: number;
+	estimated_cost_usd: number;
+	function_name: string;
+	timestamp: string;
+}
+
+// 1. Classify
+export interface ClassifyInput {
+	text: string;
+	available_types?: string[];
+}
+
+export interface ClassifyResult {
+	quality: "structured" | "draft" | "thoughts";
+	quality_score: number;
+	type: string;
+	reasoning: string;
+	usage: TokenUsage;
+}
+
+// 2. Extract
+export interface ExtractInput {
+	text: string;
+	file_path: string;
+	existing_entities?: Entity[];
+}
+
+export interface ExtractedEntityResult {
+	type: string;
+	name: string;
+	aliases: string[];
+	confidence: number;
+}
+
+export interface ExtractedRelationResult {
+	from_entity: string;
+	to_entity: string;
+	relation_type: string;
+	confidence: number;
+	evidence: string;
+}
+
+export interface ExtractResult {
+	entities: ExtractedEntityResult[];
+	relations: ExtractedRelationResult[];
+	usage: TokenUsage;
+}
+
+// 3. Score
+export interface ScoreInput {
+	content: string;
+	intent: string;
+	collection_name?: string;
+}
+
+export interface ScoreResult {
+	score: number;
+	issues: string[];
+	strengths: string[];
+	usage: TokenUsage;
+}
+
+// 4. Judge
+export interface JudgeInput {
+	draft: string;
+	gold_standard: string;
+	judge_prompt: string;
+	intent?: string;
+}
+
+export interface JudgeResult {
+	score: number;
+	reasoning: string;
+	usage: TokenUsage;
+}
+
+// 5. Detect staleness
+export interface StalenessInput {
+	content: string;
+	content_generated_at: string;
+	source_texts: Array<{ path: string; text: string; updated_at: string }>;
+}
+
+export interface StalenessResult {
+	stale: boolean;
+	reason: string;
+	stale_sources: string[];
+	confidence: number;
+	usage: TokenUsage;
+}
+
+// 6. Intent impact
+export interface IntentImpactInput {
+	changed_intent: string;
+	previous_intent: string;
+	content_pages: Array<{ path: string; text: string }>;
+}
+
+export interface IntentImpactResult {
+	affected_pages: Array<{ path: string; impact: "high" | "medium" | "low"; reason: string }>;
+	summary: string;
+	usage: TokenUsage;
+}
+
+// 7. Find gaps (AI-assisted)
+export interface FindGapsInput {
+	entities: Entity[];
+	relations: Array<{ from: string; to: string; type: string }>;
+	existing_content: Array<{ path: string; title: string; summary: string }>;
+	business_context?: string;
+}
+
+export interface FindGapsResult {
+	gaps: Array<{
+		entity_name: string;
+		entity_type: string;
+		suggested_content_type: string;
+		priority: "high" | "medium" | "low";
+		reason: string;
+	}>;
+	usage: TokenUsage;
+}
+
+export interface IntelligenceProvider {
+	classify(input: ClassifyInput): Promise<ClassifyResult>;
+	extract(input: ExtractInput): Promise<ExtractResult>;
+	score(input: ScoreInput): Promise<ScoreResult>;
+	judge(input: JudgeInput): Promise<JudgeResult>;
+	detectStaleness(input: StalenessInput): Promise<StalenessResult>;
+	analyzeIntentImpact(input: IntentImpactInput): Promise<IntentImpactResult>;
+	findGaps(input: FindGapsInput): Promise<FindGapsResult>;
+	generate(input: GenerateInput): Promise<GenerateResult>;
+	improvePrompt(input: ImprovePromptInput): Promise<ImprovePromptResult>;
+}
+
+// 8. Generate content
+export interface GenerateInput {
+	content_type: string;
+	generation_prompt: string;
+	knowledge_context: string;
+	intent?: string;
+	collection_schema?: Record<string, unknown>;
+}
+
+export interface GenerateResult {
+	content: string;
+	frontmatter: Record<string, unknown>;
+	usage: TokenUsage;
+}
+
+// 9. Improve prompt
+export interface ImprovePromptInput {
+	current_prompt: string;
+	judge_reasoning: string;
+	judge_score: number;
+	content_type: string;
+}
+
+export interface ImprovePromptResult {
+	improved_prompt: string;
+	changes_summary: string;
+	usage: TokenUsage;
+}
+
+// Eval types
+export interface EvalResult {
+	id: string;
+	content_type: string;
+	prompt_version: string;
+	score: number;
+	status: "keep" | "discard" | "improved";
+	reasoning: string;
+	iteration: number;
+	generated_content?: string;
+	timestamp: string;
+	total_usage: TokenUsage;
+}
+
+export interface EvalRunConfig {
+	content_type: string;
+	knowledge_context: string;
+	gold_standard: string;
+	judge_prompt: string;
+	generation_prompt: string;
+	intent?: string;
+	threshold: number;
+	max_iterations: number;
+	collection_schema?: Record<string, unknown>;
+}
+
+export interface EvalRunResult {
+	iterations: EvalResult[];
+	final_score: number;
+	final_status: "keep" | "discard";
+	final_content?: string;
+	final_prompt: string;
+	prompt_improved: boolean;
+	total_usage: TokenUsage;
+}

package/tsconfig.json

@@ -0,0 +1,5 @@
+{
+	"extends": "../../tsconfig.json",
+	"compilerOptions": { "outDir": "dist", "rootDir": "src" },
+	"include": ["src"]
+}

package/vitest.config.ts

@@ -0,0 +1,2 @@
+import { defineConfig } from "vitest/config";
+export default defineConfig({ test: { include: ["src/__tests__/**/*.test.ts"] } });