18 lines
511 B
Python
18 lines
511 B
Python
|
def migrate(cr, version):
|
||
|
|
# Add num_predict column if it doesn't exist
|
||
|
|
cr.execute("""
|
||
|
|
DO $$
|
||
|
|
BEGIN
|
||
|
|
IF NOT EXISTS (
|
||
|
|
SELECT 1
|
||
|
|
FROM information_schema.columns
|
||
|
|
WHERE table_name='ai_provider_instance'
|
||
|
|
AND column_name='num_predict'
|
||
|
|
) THEN
|
||
|
|
ALTER TABLE ai_provider_instance
|
||
|
|
ADD COLUMN num_predict integer DEFAULT 1024;
|
||
|
|
END IF;
|
||
|
|
END
|
||
|
|
$$;
|
||
|
|
""")
|