update openrouter

This commit is contained in:
2026-02-01 16:55:14 +01:00
parent 6dbcadcaee
commit e2f69e68cd
8 changed files with 199 additions and 75 deletions

View File

@@ -3,8 +3,8 @@
*/
import { createLogger } from "../../core/logger";
import { ReplicateProvider } from "./replicate";
import type { AiProvider, AiResponse } from "./types";
import { OpenRouterProvider } from "./openrouter";
import type { AiProvider, AiResponse, MessageStyle } from "./types";
const logger = createLogger("AI:Service");
@@ -12,7 +12,7 @@ export class AiService {
private provider: AiProvider;
constructor(provider?: AiProvider) {
this.provider = provider ?? new ReplicateProvider();
this.provider = provider ?? new OpenRouterProvider();
}
async health(): Promise<boolean> {
@@ -26,6 +26,17 @@ export class AiService {
logger.debug("Generating response", { promptLength: prompt.length });
return this.provider.ask({ prompt, systemPrompt });
}
/**
* Classify a message to determine the appropriate response style
*/
async classifyMessage(message: string): Promise<MessageStyle> {
if (this.provider.classifyMessage) {
return this.provider.classifyMessage(message);
}
// Default to snarky if provider doesn't support classification
return "snarky";
}
}
// Singleton instance
@@ -38,4 +49,4 @@ export function getAiService(): AiService {
return aiService;
}
export type { AiProvider, AiResponse } from "./types";
export type { AiProvider, AiResponse, MessageStyle } from "./types";