MarketTakers/codetaker
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(agent): add comprehensive prompt audit system

Browse Source
This commit is contained in:
hdbg
2026-02-27 13:23:21 +01:00
parent 3738272f80
commit 3a210809cf
21 changed files with 1800 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
crates/codetaker-db/migrations/2026-02-27-094438-0000_initial/down.sql
View File

@@ -1,3 +1,11 @@
DROP TABLE IF EXISTS prompt_hook_on_stream_completion_response_finish_events;
DROP TABLE IF EXISTS prompt_hook_on_tool_call_delta_events;
DROP TABLE IF EXISTS prompt_hook_on_text_delta_events;
DROP TABLE IF EXISTS prompt_hook_on_tool_result_events;
DROP TABLE IF EXISTS prompt_hook_on_tool_call_events;
DROP TABLE IF EXISTS prompt_hook_on_completion_response_events;
DROP TABLE IF EXISTS prompt_hook_on_completion_call_events;
DROP TABLE IF EXISTS prompt_audit_sessions;
DROP TABLE IF EXISTS review_thread_messages;
DROP TABLE IF EXISTS review_threads;
DROP TABLE IF EXISTS project_memory_summaries;

115
crates/codetaker-db/migrations/2026-02-27-094438-0000_initial/up.sql
View File

@@ -30,6 +30,7 @@ CREATE TABLE project_memory_summaries (
CREATE TABLE review_threads (
id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
project_id INTEGER NOT NULL,
pull_request_id BIGINT NOT NULL,
file TEXT NOT NULL,
line INTEGER NOT NULL,
initial_comment TEXT NOT NULL,
@@ -46,11 +47,125 @@ CREATE TABLE review_thread_messages (
FOREIGN KEY(thread_id) REFERENCES review_threads(id) ON DELETE CASCADE
);
CREATE TABLE prompt_audit_sessions (
id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
project_id INTEGER NOT NULL,
pull_request_id BIGINT NOT NULL,
entrypoint TEXT NOT NULL,
started_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
ended_at TIMESTAMP,
status TEXT NOT NULL,
error_message TEXT,
FOREIGN KEY(project_id) REFERENCES projects(id) ON DELETE CASCADE
);
CREATE TABLE prompt_hook_on_completion_call_events (
id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
session_id INTEGER NOT NULL,
event_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
sequence_no INTEGER NOT NULL,
prompt_json TEXT NOT NULL,
history_json TEXT NOT NULL,
FOREIGN KEY(session_id) REFERENCES prompt_audit_sessions(id) ON DELETE CASCADE
);
CREATE TABLE prompt_hook_on_completion_response_events (
id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
session_id INTEGER NOT NULL,
event_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
sequence_no INTEGER NOT NULL,
prompt_json TEXT NOT NULL,
assistant_choice_json TEXT NOT NULL,
usage_input_tokens BIGINT NOT NULL,
usage_output_tokens BIGINT NOT NULL,
usage_total_tokens BIGINT NOT NULL,
FOREIGN KEY(session_id) REFERENCES prompt_audit_sessions(id) ON DELETE CASCADE
);
CREATE TABLE prompt_hook_on_tool_call_events (
id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
session_id INTEGER NOT NULL,
event_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
sequence_no INTEGER NOT NULL,
tool_name TEXT NOT NULL,
tool_call_id TEXT,
internal_call_id TEXT NOT NULL,
args_json TEXT NOT NULL,
action TEXT NOT NULL,
reason TEXT,
FOREIGN KEY(session_id) REFERENCES prompt_audit_sessions(id) ON DELETE CASCADE
);
CREATE TABLE prompt_hook_on_tool_result_events (
id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
session_id INTEGER NOT NULL,
event_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
sequence_no INTEGER NOT NULL,
tool_name TEXT NOT NULL,
tool_call_id TEXT,
internal_call_id TEXT NOT NULL,
args_json TEXT NOT NULL,
result_json TEXT NOT NULL,
FOREIGN KEY(session_id) REFERENCES prompt_audit_sessions(id) ON DELETE CASCADE
);
CREATE TABLE prompt_hook_on_text_delta_events (
id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
session_id INTEGER NOT NULL,
event_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
sequence_no INTEGER NOT NULL,
text_delta TEXT NOT NULL,
aggregated_text TEXT NOT NULL,
FOREIGN KEY(session_id) REFERENCES prompt_audit_sessions(id) ON DELETE CASCADE
);
CREATE TABLE prompt_hook_on_tool_call_delta_events (
id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
session_id INTEGER NOT NULL,
event_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
sequence_no INTEGER NOT NULL,
tool_call_id TEXT NOT NULL,
internal_call_id TEXT NOT NULL,
tool_name TEXT,
tool_call_delta TEXT NOT NULL,
FOREIGN KEY(session_id) REFERENCES prompt_audit_sessions(id) ON DELETE CASCADE
);
CREATE TABLE prompt_hook_on_stream_completion_response_finish_events (
id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
session_id INTEGER NOT NULL,
event_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
sequence_no INTEGER NOT NULL,
prompt_json TEXT NOT NULL,
response_summary_json TEXT NOT NULL,
FOREIGN KEY(session_id) REFERENCES prompt_audit_sessions(id) ON DELETE CASCADE
);
CREATE INDEX idx_project_memory_entries_project_id
ON project_memory_entries(project_id);
CREATE INDEX idx_project_memory_summaries_project_id
ON project_memory_summaries(project_id);
CREATE INDEX idx_review_threads_project_id
ON review_threads(project_id);
CREATE INDEX idx_review_threads_project_pr
ON review_threads(project_id, pull_request_id);
CREATE INDEX idx_review_thread_messages_thread_id
ON review_thread_messages(thread_id);
CREATE INDEX idx_prompt_audit_sessions_project_pr
ON prompt_audit_sessions(project_id, pull_request_id);
CREATE INDEX idx_prompt_audit_sessions_entrypoint_started
ON prompt_audit_sessions(entrypoint, started_at);
CREATE INDEX idx_prompt_hook_completion_call_session
ON prompt_hook_on_completion_call_events(session_id, sequence_no);
CREATE INDEX idx_prompt_hook_completion_response_session
ON prompt_hook_on_completion_response_events(session_id, sequence_no);
CREATE INDEX idx_prompt_hook_tool_call_session
ON prompt_hook_on_tool_call_events(session_id, sequence_no);
CREATE INDEX idx_prompt_hook_tool_result_session
ON prompt_hook_on_tool_result_events(session_id, sequence_no);
CREATE INDEX idx_prompt_hook_text_delta_session
ON prompt_hook_on_text_delta_events(session_id, sequence_no);
CREATE INDEX idx_prompt_hook_tool_delta_session
ON prompt_hook_on_tool_call_delta_events(session_id, sequence_no);
CREATE INDEX idx_prompt_hook_stream_finish_session
ON prompt_hook_on_stream_completion_response_finish_events(session_id, sequence_no);