rename audit table

Arize-ai · Aug 8, 2024 · eea99d9 · eea99d9
1 parent 2660186
commit eea99d9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/phoenix/db/migrations/future_versions/cd164e83824f_users_and_tokens.py b/src/phoenix/db/migrations/future_versions/cd164e83824f_users_and_tokens.py
@@ -253,7 +253,7 @@ def upgrade() -> None:
         ),
     )
     op.create_table(
-        "api_keys_actions",
+        "audit_api_keys",
         sa.Column("id", sa.Integer, primary_key=True),
         sa.Column(
             "api_key_id",
@@ -286,7 +286,7 @@ def upgrade() -> None:
 
 
 def downgrade() -> None:
-    op.drop_table("api_keys_actions")
+    op.drop_table("audit_api_keys")
     op.drop_table("api_keys")
     op.drop_table("users")
     op.drop_table("user_roles")
diff --git a/src/phoenix/db/models.py b/src/phoenix/db/models.py
@@ -658,8 +658,8 @@ class APIKey(Base):
     expires_at: Mapped[Optional[datetime]] = mapped_column(UtcTimeStamp)
 
 
-class APIKeysAction(Base):
-    __tablename__ = "api_keys_actions"
+class AuditAPIKey(Base):
+    __tablename__ = "audit_api_keys"
     id: Mapped[int] = mapped_column(primary_key=True)
     api_key_id: Mapped[int] = mapped_column(
         ForeignKey("api_keys.id"),