44 lines
1.4 KiB
PL/PgSQL
44 lines
1.4 KiB
PL/PgSQL
-- AI Usage Logs: Update log_ai_usage function to include prompt and response
|
|
-- Note: prompt and response columns already exist from previous migration
|
|
|
|
-- Update log_ai_usage function to accept prompt and response
|
|
CREATE OR REPLACE FUNCTION log_ai_usage(
|
|
p_user_id UUID,
|
|
p_org_id UUID,
|
|
p_tokens INTEGER,
|
|
p_input_tokens INTEGER,
|
|
p_output_tokens INTEGER,
|
|
p_endpoint VARCHAR,
|
|
p_model VARCHAR,
|
|
p_request_type VARCHAR,
|
|
p_metadata JSONB,
|
|
p_prompt TEXT DEFAULT NULL,
|
|
p_response TEXT DEFAULT NULL
|
|
)
|
|
RETURNS UUID AS $$
|
|
DECLARE
|
|
v_log_id UUID;
|
|
v_cost NUMERIC(10, 6);
|
|
BEGIN
|
|
-- Calculate estimated cost (OpenAI-like pricing)
|
|
v_cost := (p_input_tokens::NUMERIC * 0.000001) + (p_output_tokens::NUMERIC * 0.000003);
|
|
|
|
INSERT INTO ai_usage_logs (
|
|
user_id, organization_id, tokens_used, input_tokens, output_tokens,
|
|
endpoint, model, request_type, request_metadata, estimated_cost_usd,
|
|
prompt, response
|
|
)
|
|
VALUES (
|
|
p_user_id, p_org_id, p_tokens, p_input_tokens, p_output_tokens,
|
|
p_endpoint, p_model, p_request_type, p_metadata, v_cost,
|
|
p_prompt, p_response
|
|
)
|
|
RETURNING id INTO v_log_id;
|
|
|
|
RETURN v_log_id;
|
|
END;
|
|
$$ LANGUAGE plpgsql;
|
|
|
|
COMMENT ON COLUMN ai_usage_logs.prompt IS 'The actual prompt sent to the AI model';
|
|
COMMENT ON COLUMN ai_usage_logs.response IS 'The AI model response content';
|