test: add test command to include examples, add test config in ex

.github/workflows/ci.yaml

@@ -78,7 +78,7 @@ jobs:
         uses: actions-rs/cargo@v1
         with:
           command: nextest
-          args: run --all-features
+          args: run --all-features --tests --examples
         env:
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
           ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}

rig-core/examples/agent.rs

@@ -21,3 +21,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/agent_with_context.rs

@@ -26,3 +26,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/agent_with_loaders.rs

@@ -36,3 +36,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/agent_with_tools.rs

@@ -111,3 +111,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/anthropic_agent.rs

@@ -29,3 +29,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/calculator_chatbot.rs

@@ -260,10 +260,10 @@ async fn main() -> Result<(), anyhow::Error> {
         .agent("gpt-4")
         .preamble(
             "You are an assistant here to help the user select which tool is most appropriate to perform arithmetic operations.
-            Follow these instructions closely. 
+            Follow these instructions closely.
             1. Consider the user's request carefully and identify the core elements of the request.
-            2. Select which tool among those made available to you is appropriate given the context. 
-            3. This is very important: never perform the operation yourself and never give me the direct result. 
+            2. Select which tool among those made available to you is appropriate given the context.
+            3. This is very important: never perform the operation yourself and never give me the direct result.
             Always respond with the name of the tool that should be used and the appropriate inputs
             in the following format:
             Tool: <tool name>
@@ -281,3 +281,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/cohere_connector.rs

@@ -48,3 +48,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/debate.rs

@@ -98,3 +98,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/extractor.rs

@@ -29,3 +29,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/gemini_agent.rs

@@ -2,10 +2,9 @@ use rig::{
     completion::Prompt,
     providers::gemini::{self, completion::gemini_api_types::GenerationConfig},
 };
-#[tracing::instrument(ret)]
-#[tokio::main]
 
-async fn main() -> Result<(), anyhow::Error> {
+#[tokio::main]
+pub async fn main() -> Result<(), anyhow::Error> {
     tracing_subscriber::fmt()
         .with_max_level(tracing::Level::DEBUG)
         .with_target(false)
@@ -47,3 +46,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/gemini_embeddings.rs

@@ -18,3 +18,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/loaders.rs

@@ -12,3 +12,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/multi_agent.rs

@@ -68,3 +68,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/perplexity_agent.rs

@@ -32,3 +32,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/rag.rs

@@ -45,3 +45,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/rag_dynamic_tools.rs

@@ -183,3 +183,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/sentiment_classifier.rs

@@ -17,7 +17,7 @@ struct DocumentSentiment {
 }
 
 #[tokio::main]
-async fn main() {
+async fn main() -> Result<(), anyhow::Error> {
     // Create OpenAI client
     let openai_client = openai::Client::from_env();
 
@@ -32,4 +32,16 @@ async fn main() {
         .expect("Failed to extract sentiment");
 
     println!("GPT-4: {:?}", sentiment);
+
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
 }

rig-core/examples/simple_model.rs

@@ -1,7 +1,7 @@
 use rig::{completion::Prompt, providers::openai};
 
 #[tokio::main]
-async fn main() {
+async fn main() -> Result<(), anyhow::Error> {
     // Create OpenAI client and model
     let openai_client = openai::Client::from_env();
 
@@ -14,4 +14,16 @@ async fn main() {
         .expect("Failed to prompt GPT-4");
 
     println!("GPT-4: {response}");
+
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
 }

rig-core/examples/vector_search.rs

@@ -42,3 +42,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[tokio::test]
+    async fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}

rig-core/examples/vector_search_cohere.rs

@@ -39,3 +39,13 @@ async fn main() -> Result<(), anyhow::Error> {
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[tokio::test]
+    async fn example() -> Result<(), anyhow::Error> {
+        main()
+    }
+}