|
@@ -18,7 +18,9 @@ def test_convert_with_vision():
|
|
number_limits=5,
|
|
number_limits=5,
|
|
transfer_methods=[FileTransferMethod.REMOTE_URL],
|
|
transfer_methods=[FileTransferMethod.REMOTE_URL],
|
|
detail=ImagePromptMessageContent.DETAIL.HIGH,
|
|
detail=ImagePromptMessageContent.DETAIL.HIGH,
|
|
- )
|
|
|
|
|
|
+ ),
|
|
|
|
+ allowed_file_upload_methods=[FileTransferMethod.REMOTE_URL],
|
|
|
|
+ number_limits=5,
|
|
)
|
|
)
|
|
assert result == expected
|
|
assert result == expected
|
|
|
|
|
|
@@ -33,7 +35,9 @@ def test_convert_without_vision():
|
|
}
|
|
}
|
|
result = FileUploadConfigManager.convert(config, is_vision=False)
|
|
result = FileUploadConfigManager.convert(config, is_vision=False)
|
|
expected = FileUploadConfig(
|
|
expected = FileUploadConfig(
|
|
- image_config=ImageConfig(number_limits=5, transfer_methods=[FileTransferMethod.REMOTE_URL])
|
|
|
|
|
|
+ image_config=ImageConfig(number_limits=5, transfer_methods=[FileTransferMethod.REMOTE_URL]),
|
|
|
|
+ allowed_file_upload_methods=[FileTransferMethod.REMOTE_URL],
|
|
|
|
+ number_limits=5,
|
|
)
|
|
)
|
|
assert result == expected
|
|
assert result == expected
|
|
|
|
|