Skip to content

Stop rebuilding the tool_calls primary key on every boot #159

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
salluexez wants to merge 1 commit into from
+13 −2
Closed

Stop rebuilding the tool_calls primary key on every boot #159

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 13 additions & 2 deletions src/runs/postgres-run-store.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -73,8 +73,19 @@ export function createPostgresRunStore(connectionString: string, opts?: { maxCla
PRIMARY KEY(run_id, attempt, call_index)
)`,
`ALTER TABLE tool_calls ADD COLUMN IF NOT EXISTS attempt INT NOT NULL DEFAULT 1`,
`ALTER TABLE tool_calls DROP CONSTRAINT IF EXISTS tool_calls_pkey`,
`ALTER TABLE tool_calls ADD PRIMARY KEY (run_id, attempt, call_index)`,
`DO $$
BEGIN
IF EXISTS (
SELECT 1 FROM information_schema.constraint_column_usage
WHERE table_name = 'tool_calls' AND constraint_name = 'tool_calls_pkey' AND column_name = 'run_id'
Comment on lines +76 to +80
) AND NOT EXISTS (
SELECT 1 FROM information_schema.constraint_column_usage
WHERE table_name = 'tool_calls' AND constraint_name = 'tool_calls_pkey' AND column_name = 'attempt'
) THEN
ALTER TABLE tool_calls DROP CONSTRAINT tool_calls_pkey;
ALTER TABLE tool_calls ADD PRIMARY KEY (run_id, attempt, call_index);
END IF;
END $$`,
]);

async function getRun(id: string): Promise<Run | null> {
Expand Down