67 lines
1.7 KiB
Rust
67 lines
1.7 KiB
Rust
use std::path::PathBuf;
|
|
|
|
use clap::{ArgGroup, Args, Subcommand};
|
|
|
|
/// Artifact-related subcommands.
|
|
#[derive(Debug, Clone, Subcommand)]
|
|
pub enum ArtifactCommand {
|
|
/// Upload an artifact file.
|
|
#[command(hide = true)]
|
|
Upload(UploadArtifactArgs),
|
|
/// Get artifact metadata.
|
|
Get(GetArtifactArgs),
|
|
/// Download an artifact file.
|
|
Download(DownloadArtifactArgs),
|
|
}
|
|
|
|
impl ArtifactCommand {
|
|
pub(crate) fn as_str_for_tracing(&self) -> &'static str {
|
|
match self {
|
|
ArtifactCommand::Upload(_) => "artifact upload",
|
|
ArtifactCommand::Get(_) => "artifact get",
|
|
ArtifactCommand::Download(_) => "artifact download",
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone, Args)]
|
|
#[command(
|
|
group(
|
|
ArgGroup::new("artifact_association")
|
|
.multiple(false)
|
|
.args(["run_id", "conversation_id"])
|
|
)
|
|
)]
|
|
pub struct UploadArtifactArgs {
|
|
/// Path to the artifact file to upload.
|
|
pub path: PathBuf,
|
|
|
|
/// Associate the uploaded artifact with a run.
|
|
#[arg(long = "run-id")]
|
|
pub run_id: Option<String>,
|
|
|
|
/// Associate the uploaded artifact with a conversation.
|
|
#[arg(long = "conversation-id")]
|
|
pub conversation_id: Option<String>,
|
|
|
|
/// Description for the uploaded artifact.
|
|
#[arg(long = "description")]
|
|
pub description: Option<String>,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Args)]
|
|
pub struct DownloadArtifactArgs {
|
|
/// UID of the artifact to download.
|
|
pub artifact_uid: String,
|
|
|
|
/// Write the downloaded artifact to a specific file path.
|
|
#[arg(long = "out", short = 'o')]
|
|
pub out: Option<PathBuf>,
|
|
}
|
|
|
|
#[derive(Debug, Clone, Args)]
|
|
pub struct GetArtifactArgs {
|
|
/// UID of the artifact to get.
|
|
pub artifact_uid: String,
|
|
}
|