mirror of
https://github.com/langgenius/dify.git
synced 2026-07-25 05:28:35 +08:00
52c106b532
Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: yyh <yuanyouhuilyz@gmail.com> Co-authored-by: Yansong Zhang <916125788@qq.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
"""Contract tests for the dify.drive declaration layer (ENG-623)."""
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from dify_agent.layers.drive import (
|
|
DIFY_DRIVE_LAYER_TYPE_ID,
|
|
DifyDriveLayerConfig,
|
|
DifyDriveSkillConfig,
|
|
)
|
|
from dify_agent.layers.drive.layer import DifyDriveLayer
|
|
|
|
|
|
def test_type_id_is_frozen_contract() -> None:
|
|
assert DIFY_DRIVE_LAYER_TYPE_ID == "dify.drive"
|
|
assert DifyDriveLayer.type_id == DIFY_DRIVE_LAYER_TYPE_ID
|
|
|
|
|
|
def test_layer_config_round_trips_manifest_entries() -> None:
|
|
config = DifyDriveLayerConfig.model_validate(
|
|
{
|
|
"drive_ref": "agent-019e9112",
|
|
"skills": [
|
|
{
|
|
"path": "tender-analyzer",
|
|
"name": "Tender Analyzer",
|
|
"description": "Parses RFP documents step by step.",
|
|
"skill_md_key": "tender-analyzer/SKILL.md",
|
|
"archive_key": "tender-analyzer/.DIFY-SKILL-FULL.zip",
|
|
}
|
|
],
|
|
"mentioned_skill_keys": ["tender-analyzer/SKILL.md"],
|
|
"mentioned_file_keys": ["files/sample.pdf"],
|
|
}
|
|
)
|
|
|
|
dumped = config.model_dump(mode="json")
|
|
assert dumped["drive_ref"] == "agent-019e9112"
|
|
assert "drive_base" not in dumped
|
|
assert dumped["skills"][0]["skill_md_key"] == "tender-analyzer/SKILL.md"
|
|
assert dumped["mentioned_file_keys"] == ["files/sample.pdf"]
|
|
assert "content" not in DifyDriveSkillConfig.model_fields
|
|
|
|
|
|
def test_layer_config_rejects_unknown_fields() -> None:
|
|
with pytest.raises(ValidationError):
|
|
DifyDriveLayerConfig.model_validate({"drive_ref": "agent-1", "skill_md_body": "# inline content"})
|
|
|
|
|
|
def test_drive_layer_is_registered_and_constructible_from_config() -> None:
|
|
layer = DifyDriveLayer.from_config(
|
|
DifyDriveLayerConfig(drive_ref="agent-1", skills=[], mentioned_skill_keys=[], mentioned_file_keys=[]),
|
|
)
|
|
|
|
assert isinstance(layer, DifyDriveLayer)
|
|
assert layer.config.drive_ref == "agent-1"
|
|
assert not hasattr(layer, "local_drive_base")
|