TradingAgents/docker/db/seed.sql

58 lines
2.0 KiB
SQL

-- TimescaleDB initialization script for TradingAgents
-- This script sets up the main database and test database with required extensions
-- First, create extensions in the default postgres database
CREATE EXTENSION IF NOT EXISTS timescaledb CASCADE;
CREATE EXTENSION IF NOT EXISTS vector;
-- vectorscale extension - try to install but don't fail if not available
DO $$
BEGIN
CREATE EXTENSION IF NOT EXISTS vectorscale CASCADE;
EXCEPTION WHEN OTHERS THEN
RAISE NOTICE 'vectorscale extension not available, continuing without it';
END $$;
CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
-- Create test database (main database 'tradingagents' is created by POSTGRES_DB env var)
CREATE DATABASE tradingagents_test;
-- Setup extensions in main database
\c tradingagents
-- Install extensions
CREATE EXTENSION IF NOT EXISTS timescaledb CASCADE;
CREATE EXTENSION IF NOT EXISTS vector;
-- vectorscale extension - try to install but don't fail if not available
DO $$
BEGIN
CREATE EXTENSION IF NOT EXISTS vectorscale CASCADE;
EXCEPTION WHEN OTHERS THEN
RAISE NOTICE 'vectorscale extension not available, continuing without it';
END $$;
CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
-- Verify extensions are installed
SELECT extname FROM pg_extension WHERE extname IN ('timescaledb', 'vector', 'uuid-ossp');
-- Setup extensions in test database
\c tradingagents_test
-- Same extensions in test database
CREATE EXTENSION IF NOT EXISTS timescaledb CASCADE;
CREATE EXTENSION IF NOT EXISTS vector;
-- vectorscale extension - try to install but don't fail if not available
DO $$
BEGIN
CREATE EXTENSION IF NOT EXISTS vectorscale CASCADE;
EXCEPTION WHEN OTHERS THEN
RAISE NOTICE 'vectorscale extension not available, continuing without it';
END $$;
CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
-- Verify extensions are installed in test database
SELECT extname FROM pg_extension WHERE extname IN ('timescaledb', 'vector', 'uuid-ossp');
-- Output confirmation message
\c tradingagents
SELECT 'TradingAgents TimescaleDB setup complete with TimescaleDB, vector, and test database' AS status;