bugfix min max

Author: thor314

src/main.rs

@@ -11,16 +11,14 @@ use ollama_rs::{
 #[derive(Parser, Debug)]
 #[command(author, version, about, long_about = None)]
 struct Args {
-    /// Maximum value (exclusive) for random number
-    #[arg(default_value = "10")]
-    max: Option<u128>,
+    /// First argument: either max (if only one provided) or min (if two provided)
+    first: Option<u128>,
 
-    /// Minimum value (inclusive) for random number
-    min: Option<u128>,
+    /// Second argument (optional): max value when min is provided
+    second: Option<u128>,
 }
 
 async fn get_random_number(min: u128, max: u128) -> Result<u128> {
-    // Initialize Ollama client
     let ollama = Ollama::default();
 
     let prompt = format!(
@@ -29,25 +27,19 @@ async fn get_random_number(min: u128, max: u128) -> Result<u128> {
         min, max
     );
 
-    // Create generation request with specific options to reduce randomness
-    let options = GenerationOptions::default()
-        .temperature(0.9)  // High temperature for more randomness
-        .top_p(0.9)       // Allow more variety in responses
-        .top_k(40); // Consider more tokens for more randomness
+    let options = GenerationOptions::default().temperature(0.9).top_p(0.9).top_k(40);
 
     let request = GenerationRequest::new("mistral".to_string(), prompt).options(options);
 
     let response =
         ollama.generate(request).await.context("Failed to generate response from Ollama")?;
 
-    // Parse the response string into a number
     let num = response
         .response
         .trim()
         .parse::<u128>()
         .context("Failed to parse number from Ollama response")?;
 
-    // Ensure the number is within bounds
     if num < min || num > max {
         anyhow::bail!("Generated number {} is out of bounds [{}, {}]", num, min, max);
     }
@@ -59,18 +51,21 @@ async fn get_random_number(min: u128, max: u128) -> Result<u128> {
 async fn main() {
     let args = Args::parse();
 
-    let (min, max) = match (args.min, args.max) {
-        (None, None) => (0, 10),
-        (None, Some(max)) => (0, max),
-        (Some(min), Some(max)) => (min, max),
-        (Some(_), None) => {
-            eprintln!("Error: If min is specified, max must also be specified");
+    let (min, max) = match (args.first, args.second) {
+        (None, None) => (0, 10),              // raindom
+        (Some(max), None) => (0, max),        // raindom [max]
+        (Some(min), Some(max)) => (min, max), // raindom [min] [max]
+        (None, Some(_)) => {
+            eprintln!("Invalid argument pattern. Usage:");
+            eprintln!("  raindom");
+            eprintln!("  raindom [max]");
+            eprintln!("  raindom [min] [max]");
             process::exit(1);
         },
     };
 
     if min >= max {
-        eprintln!("Error: min must be less than max");
+        eprintln!("Error: min ({}) must be less than max ({})", min, max);
         process::exit(1);
     }