kunthawat/moreminimore-vibe
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Extract AI rules (#188)

Browse Source
This commit is contained in:
Will Chen
2025-05-17 23:04:48 -07:00
committed by GitHub
parent 38535bea46
commit 59c18c3859
4 changed files with 83 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/ipc/handlers/chat_stream_handlers.ts
View File

@@ -3,7 +3,10 @@ import { CoreMessage, TextPart, ImagePart } from "ai";
import { db } from "../../db";
import { chats, messages } from "../../db/schema";
import { and, eq, isNull } from "drizzle-orm";
import { SYSTEM_PROMPT } from "../../prompts/system_prompt";
import {
constructSystemPrompt,
readAiRules,
} from "../../prompts/system_prompt";
import {
SUPABASE_AVAILABLE_SYSTEM_PROMPT,
SUPABASE_NOT_AVAILABLE_SYSTEM_PROMPT,
@@ -283,7 +286,9 @@ export function registerChatStreamHandlers() {
);
}
let systemPrompt = SYSTEM_PROMPT;
let systemPrompt = constructSystemPrompt({
aiRules: await readAiRules(getDyadAppPath(updatedChat.app.path)),
});
if (
updatedChat.app?.supabaseProjectId &&
settings.supabase?.accessToken?.value

9
src/ipc/handlers/token_count_handlers.ts
View File

@@ -1,7 +1,10 @@
import { db } from "../../db";
import { chats } from "../../db/schema";
import { eq } from "drizzle-orm";
import { SYSTEM_PROMPT } from "../../prompts/system_prompt";
import {
constructSystemPrompt,
readAiRules,
} from "../../prompts/system_prompt";
import {
SUPABASE_AVAILABLE_SYSTEM_PROMPT,
SUPABASE_NOT_AVAILABLE_SYSTEM_PROMPT,
@@ -48,7 +51,9 @@ export function registerTokenCountHandlers() {
const inputTokens = estimateTokens(req.input);
// Count system prompt tokens
let systemPrompt = SYSTEM_PROMPT;
let systemPrompt = constructSystemPrompt({
aiRules: await readAiRules(getDyadAppPath(chat.app.path)),
});
let supabaseContext = "";
if (chat.app?.supabaseProjectId) {