diff --git a/alembic/versions/2024_10_18_2003-2873c5c8c41e_add_actions_confidence_float.py b/alembic/versions/2024_10_18_2003-2873c5c8c41e_add_actions_confidence_float.py new file mode 100644 index 00000000..50b0f956 --- /dev/null +++ b/alembic/versions/2024_10_18_2003-2873c5c8c41e_add_actions_confidence_float.py @@ -0,0 +1,31 @@ +"""add actions.confidence_float + +Revision ID: 2873c5c8c41e +Revises: 137eee1d3b3e +Create Date: 2024-10-18 20:03:10.612242+00:00 + +""" + +from typing import Sequence, Union + +import sqlalchemy as sa + +from alembic import op + +# revision identifiers, used by Alembic. +revision: str = "2873c5c8c41e" +down_revision: Union[str, None] = "137eee1d3b3e" +branch_labels: Union[str, Sequence[str], None] = None +depends_on: Union[str, Sequence[str], None] = None + + +def upgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.add_column("actions", sa.Column("confidence_float", sa.Numeric(), nullable=True)) + # ### end Alembic commands ### + + +def downgrade() -> None: + # ### commands auto generated by Alembic - please adjust! ### + op.drop_column("actions", "confidence_float") + # ### end Alembic commands ### diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index 7c8438ab..4f86dddc 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -1613,6 +1613,7 @@ class AgentDB: skyvern_element_hash=action.skyvern_element_hash, skyvern_element_data=action.skyvern_element_data, action_json=action.model_dump(), + confidence_float=action.confidence_float, ) session.add(new_action) await session.commit() diff --git a/skyvern/forge/sdk/db/models.py b/skyvern/forge/sdk/db/models.py index 6d02bc84..a6b47945 100644 --- a/skyvern/forge/sdk/db/models.py +++ b/skyvern/forge/sdk/db/models.py @@ -461,6 +461,7 @@ class ActionModel(Base): skyvern_element_hash = Column(String, nullable=True) skyvern_element_data = Column(JSON, nullable=True) action_json = Column(JSON, nullable=True) + confidence_float = Column(Numeric, nullable=True) created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False) modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)