mirror of
https://gitea.toothfairyai.com/ToothFairyAI/tf_code.git
synced 2026-03-31 22:32:28 +00:00
feat(id): brand SessionID through Drizzle and Zod schemas (#16953)
This commit is contained in:
@@ -1,12 +1,13 @@
|
||||
import { test, expect, describe } from "bun:test"
|
||||
import { extractResponseText, formatPromptTooLargeError } from "../../src/cli/cmd/github"
|
||||
import type { MessageV2 } from "../../src/session/message-v2"
|
||||
import { SessionID } from "../../src/session/schema"
|
||||
|
||||
// Helper to create minimal valid parts
|
||||
function createTextPart(text: string): MessageV2.Part {
|
||||
return {
|
||||
id: "1",
|
||||
sessionID: "s",
|
||||
sessionID: SessionID.make("s"),
|
||||
messageID: "m",
|
||||
type: "text" as const,
|
||||
text,
|
||||
@@ -16,7 +17,7 @@ function createTextPart(text: string): MessageV2.Part {
|
||||
function createReasoningPart(text: string): MessageV2.Part {
|
||||
return {
|
||||
id: "1",
|
||||
sessionID: "s",
|
||||
sessionID: SessionID.make("s"),
|
||||
messageID: "m",
|
||||
type: "reasoning" as const,
|
||||
text,
|
||||
@@ -28,7 +29,7 @@ function createToolPart(tool: string, title: string, status: "completed" | "runn
|
||||
if (status === "completed") {
|
||||
return {
|
||||
id: "1",
|
||||
sessionID: "s",
|
||||
sessionID: SessionID.make("s"),
|
||||
messageID: "m",
|
||||
type: "tool" as const,
|
||||
callID: "c1",
|
||||
@@ -45,7 +46,7 @@ function createToolPart(tool: string, title: string, status: "completed" | "runn
|
||||
}
|
||||
return {
|
||||
id: "1",
|
||||
sessionID: "s",
|
||||
sessionID: SessionID.make("s"),
|
||||
messageID: "m",
|
||||
type: "tool" as const,
|
||||
callID: "c1",
|
||||
@@ -61,7 +62,7 @@ function createToolPart(tool: string, title: string, status: "completed" | "runn
|
||||
function createStepStartPart(): MessageV2.Part {
|
||||
return {
|
||||
id: "1",
|
||||
sessionID: "s",
|
||||
sessionID: SessionID.make("s"),
|
||||
messageID: "m",
|
||||
type: "step-start" as const,
|
||||
}
|
||||
@@ -70,7 +71,7 @@ function createStepStartPart(): MessageV2.Part {
|
||||
function createStepFinishPart(): MessageV2.Part {
|
||||
return {
|
||||
id: "1",
|
||||
sessionID: "s",
|
||||
sessionID: SessionID.make("s"),
|
||||
messageID: "m",
|
||||
type: "step-finish" as const,
|
||||
reason: "done",
|
||||
|
||||
Reference in New Issue
Block a user