aslan/aslobot-matrix
SHA256
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Rework AI and add more stats

Browse source
This commit is contained in:
Aslan 2025-12-28 17:35:36 +01:00
parent 29832dfce3
commit c8de53bfc7
14 changed files with 197 additions and 169 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/services/ai/ai.ts
View file

@ -1,22 +1,24 @@
import { GoogleGenAI } from "@google/genai";
import { config } from "../../config.js";
import type { AIResponseImage, AIResponseText } from "./types.js";
const googleAI = new GoogleGenAI({
apiKey: config.app.ai.api.key,
});
const getTextGemini = async (input: string): Promise<string> => {
const getTextGemini = async (input: string): Promise<AIResponseText> => {
const response = await googleAI.models.generateContent({
model: "gemini-3-flash-preview",
contents: input,
});
return response.text ?? "AI Error";
return {
text: response.text ?? "AI Error",
tokens: response.usageMetadata?.totalTokenCount ?? 0,
};
};
const getImageGemini = async (
input: string,
): Promise<Buffer<ArrayBuffer> | undefined> => {
const getImageGemini = async (input: string): Promise<AIResponseImage> => {
const response = await googleAI.models.generateContent({
model: "gemini-2.5-flash-image",
contents: input,
@ -38,7 +40,10 @@ const getImageGemini = async (
}
});
return buffer;
return {
image: buffer,
tokens: response.usageMetadata?.totalTokenCount ?? 0,
};
};
export { getTextGemini, getImageGemini };