Index/aero
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add USE_FIREHOSE env var for faster responses

Browse source
This commit is contained in:
Index 2025-10-26 15:30:22 -05:00
parent df5933cbc9
commit f237f33946
3 changed files with 16 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
.env.example
View file

@ -14,4 +14,7 @@ HANDLE=""
BSKY_PASSWORD="" BSKY_PASSWORD=""
# https://aistudio.google.com/apikey # https://aistudio.google.com/apikey
GEMINI_API_KEY="" GEMINI_API_KEY=""
DAILY_QUERY_LIMIT=15
USE_FIREHOSE=false

7
src/core.ts
View file

@ -1,10 +1,15 @@
import { GoogleGenAI } from "@google/genai"; import { GoogleGenAI } from "@google/genai";
import { Bot } from "@skyware/bot"; import { Bot, EventStrategy } from "@skyware/bot";
import { env } from "./env"; import { env } from "./env";
export const bot = new Bot({ export const bot = new Bot({
service: env.SERVICE, service: env.SERVICE,
emitChatEvents: true, emitChatEvents: true,
eventEmitterOptions: {
strategy: env.USE_FIREHOSE
? EventStrategy.Firehose
: EventStrategy.Polling,
},
}); });
export const ai = new GoogleGenAI({ export const ai = new GoogleGenAI({

7
src/env.ts
View file

@ -17,9 +17,14 @@ const envSchema = z.object({
GEMINI_API_KEY: z.string(), GEMINI_API_KEY: z.string(),
DAILY_QUERY_LIMIT: z.preprocess( DAILY_QUERY_LIMIT: z.preprocess(
(val) => (typeof val === "string" && val.trim() !== "") ? Number(val) : undefined, (val) =>
(typeof val === "string" && val.trim() !== "") ? Number(val) : undefined,
z.number().int().positive().default(15), z.number().int().positive().default(15),
), ),
USE_FIREHOSE: z.preprocess(
(val) => val === "true",
z.boolean().default(false),
),
}); });
export type Env = z.infer<typeof envSchema>; export type Env = z.infer<typeof envSchema>;