Ryan/Reactive-Resume
1
0
Fork 0
mirror of https://github.com/AmruthPillai/Reactive-Resume.git synced 2026-07-16 07:58:10 +10:00
Code Issues Packages Projects Releases Wiki Activity

fix #2587 - allow for customizable endpoints for all AI providers (#2589)

Browse Source
This commit is contained in:
Amruth Pillai
2026-01-25 18:29:36 +01:00
committed by GitHub
parent 3f55c24e36
commit 982220430e
2 changed files with 59 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/integrations/orpc/router/ai.ts
+10 -9
View File
@@ -12,7 +12,7 @@ import pdfParserUserPrompt from "@/integrations/ai/prompts/pdf-parser-user.md?ra
import { defaultResumeData, resumeDataSchema } from "@/schema/resume/data";
import { protectedProcedure } from "../context";
const aiProviderSchema = z.enum(["vercel-ai-gateway", "openai", "anthropic", "gemini", "ollama"]);
const aiProviderSchema = z.enum(["ollama", "openai", "gemini", "anthropic", "vercel-ai-gateway"]);
type AIProvider = z.infer<typeof aiProviderSchema>;
@@ -20,18 +20,19 @@ type GetModelInput = {
provider: AIProvider;
model: string;
apiKey: string;
baseURL?: string;
baseURL: string;
};
function getModel(input: GetModelInput) {
const { provider, model, apiKey, baseURL } = input;
const { provider, model, apiKey } = input;
const baseURL = input.baseURL || undefined;
return match(provider)
.with("vercel-ai-gateway", () => createGateway({ apiKey }).languageModel(model))
.with("openai", () => createOpenAI({ apiKey }).languageModel(model))
.with("anthropic", () => createAnthropic({ apiKey }).languageModel(model))
.with("gemini", () => createGoogleGenerativeAI({ apiKey }).languageModel(model))
.with("openai", () => createOpenAI({ apiKey, baseURL }).languageModel(model))
.with("ollama", () => createOllama({ apiKey, baseURL }).languageModel(model))
.with("anthropic", () => createAnthropic({ apiKey, baseURL }).languageModel(model))
.with("vercel-ai-gateway", () => createGateway({ apiKey, baseURL }).languageModel(model))
.with("gemini", () => createGoogleGenerativeAI({ apiKey, baseURL }).languageModel(model))
.exhaustive();
}
@@ -39,7 +40,7 @@ const aiCredentialsSchema = z.object({
provider: aiProviderSchema,
model: z.string(),
apiKey: z.string(),
baseURL: z.string().optional(),
baseURL: z.string(),
});
const fileInputSchema = z.object({
@@ -54,7 +55,7 @@ export const aiRouter = {
provider: aiProviderSchema,
model: z.string(),
apiKey: z.string(),
baseURL: z.string().optional(),
baseURL: z.string(),
}),
)
.handler(async function* ({ input }) {