AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-15 12:35:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'feat/mcp' into deploy/dev

Browse Source
This commit is contained in:
Novice 2025-05-30 10:57:11 +08:00
commit a4dfb548c3
2 changed files with 41 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
api/migrations/versions/2025_05_30_1054-162a22ab34cf_mcp_tool_server_url_change_type.py Normal file
View File

@ -0,0 +1,39 @@
"""mcp tool server url change type
Revision ID: 162a22ab34cf
Revises: de71f8771550
Create Date: 2025-05-30 10:54:44.511650
"""
from alembic import op
import models as models
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision = '162a22ab34cf'
down_revision = 'de71f8771550'
branch_labels = None
depends_on = None
def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
with op.batch_alter_table('tool_mcp_providers', schema=None) as batch_op:
batch_op.drop_constraint(batch_op.f('unique_mcp_tool_provider'), type_='unique')
batch_op.drop_constraint(batch_op.f('unique_mcp_tool_provider_server_url_hash'), type_='unique')
batch_op.create_unique_constraint('unique_mcp_provider_name', ['tenant_id', 'name'])
batch_op.create_unique_constraint('unique_mcp_provider_server_url', ['tenant_id', 'server_url_hash'])
# ### end Alembic commands ###
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
with op.batch_alter_table('tool_mcp_providers', schema=None) as batch_op:
batch_op.drop_constraint('unique_mcp_provider_server_url', type_='unique')
batch_op.drop_constraint('unique_mcp_provider_name', type_='unique')
batch_op.create_unique_constraint(batch_op.f('unique_mcp_tool_provider_server_url_hash'), ['server_url_hash'])
batch_op.create_unique_constraint(batch_op.f('unique_mcp_tool_provider'), ['name', 'tenant_id'])
# ### end Alembic commands ###

4
api/models/tools.py
View File

@ -232,8 +232,8 @@ class MCPToolProvider(Base):
__tablename__ = "tool_mcp_providers"
__table_args__ = (
db.PrimaryKeyConstraint("id", name="tool_mcp_provider_pkey"),
db.UniqueConstraint("name", "tenant_id", name="unique_mcp_tool_provider"),
db.UniqueConstraint("server_url_hash", name="unique_mcp_tool_provider_server_url_hash"),
db.UniqueConstraint("tenant_id", "server_url_hash", name="unique_mcp_provider_server_url"),
db.UniqueConstraint("tenant_id", "name", name="unique_mcp_provider_name"),
)
id: Mapped[str] = mapped_column(StringUUID, server_default=db.text("uuid_generate_v4()"))