Replace Azure AI dependencies with Google Generative AI integration

This commit is contained in:
암냥 2026-04-04 00:06:04 +09:00
commit 66308334ff
No known key found for this signature in database
3 changed files with 33 additions and 64 deletions

View file

@ -1,5 +1,4 @@
import ModelClient, { isUnexpected } from "@azure-rest/ai-inference";
import { AzureKeyCredential } from "@azure/core-auth";
import { GoogleGenerativeAI } from "@google/generative-ai";
const KEY = process.env.NEIS_API_KEY;
@ -46,18 +45,14 @@ export async function getMealInfo(MLSV_YMD: string): Promise<{ meal: string; dat
}
export async function NameToEmoji(name: string): Promise<string> {
const token = process.env.GITHUB_TOKEN;
if (!token) {
throw new Error("GITHUB_TOKEN environment variable is not set.");
export async function NameToEmoji(name: string): Promise<string | undefined> {
const apiKey = process.env.GOOGLE_API_KEY;
if (!apiKey) {
throw new Error("GOOGLE_API_KEY environment variable is not set.");
}
const endpoint = "https://models.github.ai/inference";
const model = "openai/gpt-5-mini";
const client = ModelClient(
endpoint,
new AzureKeyCredential(token),
);
const genAI = new GoogleGenerativeAI(apiKey);
const model = genAI.getGenerativeModel({ model: "gemini-3.1-flash-lite-preview" });
const systemPrompt = `⚠️ 중요한 지침: 당신은 오직 이모지로만 응답하는 AI입니다. 다음 규칙을 예외 없이 철저히 준수해야 합니다. ⚠️
@ -105,24 +100,27 @@ A: 🥰,☮️,🗽
`;
const response = await client.path("/chat/completions").post({
body: {
messages: [
{ role: "system", content: systemPrompt },
{ role: "user", content: name }
],
const response = await model.generateContent({
contents: [
{ role: "user", parts: [{ text: systemPrompt + "\n\nUser input: " + name }] }
],
generationConfig: {
temperature: 1.0,
top_p: 1.0,
model: model
}
topP: 1.0,
candidateCount: 1,
},
});
if (isUnexpected(response)) {
throw response.body.error;
}
const choices = response.body?.choices;
if (!choices || !choices[0]?.message?.content) {
const result = response.response;
if (!result.candidates || !result.candidates[0]) {
throw new Error("No valid response from the model.");
}
return choices[0].message.content as string;
const textContent = result.candidates[0].content.parts[0];
if (!textContent) {
throw new Error("Invalid response format from the model.");
}
return textContent.text;
}