obsolete messages db layer
This commit is contained in:
+10
-5
@@ -1,7 +1,7 @@
|
|||||||
import { router, publicProcedure, createCallerFactory } from "./server";
|
import { router, publicProcedure, createCallerFactory } from "./server";
|
||||||
import { MODEL_NAME } from "../provider.js";
|
import { MODEL_NAME } from "../provider.js";
|
||||||
import { generateObject, generateText, jsonSchema } from "ai";
|
import { generateObject, generateText, jsonSchema } from "ai";
|
||||||
import type { DraftMessage } from "../../types.js";
|
import type { CommittedMessage, DraftMessage } from "../../types.js";
|
||||||
|
|
||||||
const runningSummarySystemPrompt = ({
|
const runningSummarySystemPrompt = ({
|
||||||
previousRunningSummary,
|
previousRunningSummary,
|
||||||
@@ -46,13 +46,18 @@ Generate a new running summary of the conversation.`;
|
|||||||
export const messages = router({
|
export const messages = router({
|
||||||
fetchByConversationId: publicProcedure
|
fetchByConversationId: publicProcedure
|
||||||
.input((x) => x as { conversationId: string })
|
.input((x) => x as { conversationId: string })
|
||||||
.query(async ({ input: { conversationId }, ctx: { db } }) => {
|
.query(async ({ input: { conversationId }, ctx: { dbClient } }) => {
|
||||||
return await db.messages.findByConversationId(conversationId);
|
const rows = (await dbClient
|
||||||
|
.selectFrom("messages")
|
||||||
|
.selectAll()
|
||||||
|
.where("conversationId", "=", conversationId)
|
||||||
|
.execute()) as Array<CommittedMessage>;
|
||||||
|
return rows;
|
||||||
}),
|
}),
|
||||||
deleteOne: publicProcedure
|
deleteOne: publicProcedure
|
||||||
.input((x) => x as { id: string })
|
.input((x) => x as { id: string })
|
||||||
.mutation(async ({ input: { id }, ctx: { db } }) => {
|
.mutation(async ({ input: { id }, ctx: { dbClient } }) => {
|
||||||
await db.messages.delete(id);
|
await dbClient.deleteFrom("messages").where("id", "=", id).execute();
|
||||||
return { success: true };
|
return { success: true };
|
||||||
}),
|
}),
|
||||||
generateRunningSummary: publicProcedure
|
generateRunningSummary: publicProcedure
|
||||||
|
|||||||
Reference in New Issue
Block a user