docs: Add extraction pipeline example

rig-core/examples/multi_extract.rs

@@ -0,0 +1,73 @@
+use rig::{parallel, pipeline::{self, agent_ops, Op}, providers::openai};
+use schemars::JsonSchema;
+use serde::{Deserialize, Serialize};
+
+#[derive(Debug, Deserialize, JsonSchema, Serialize)]
+/// A record containing extracted names
+pub struct Names {
+    /// The names extracted from the text
+    pub names: Vec<String>,
+}
+
+#[derive(Debug, Deserialize, JsonSchema, Serialize)]
+/// A record containing extracted topics
+pub struct Topics {
+    /// The topics extracted from the text
+    pub topics: Vec<String>,
+}
+
+#[derive(Debug, Deserialize, JsonSchema, Serialize)]
+/// A record containing extracted sentiment
+pub struct Sentiment {
+    /// The sentiment of the text (-1 being negative, 1 being positive)
+    pub sentiment: f64,
+    /// The confidence of the sentiment
+    pub confidence: f64,
+}
+
+
+#[tokio::main]
+async fn main() -> anyhow::Result<()> {
+    let openai = openai::Client::from_env();
+
+    let names_extractor = openai
+        .extractor::<Names>("gpt-4")
+        .preamble("Extract names (e.g.: of people, places) from the given text.")
+        .build();
+
+    let topics_extractor = openai
+        .extractor::<Topics>("gpt-4")
+        .preamble("Extract topics from the given text.")
+        .build();
+
+    let sentiment_extractor = openai
+        .extractor::<Sentiment>("gpt-4")
+        .preamble("Extract sentiment (and how confident you are of the sentiment) from the given text.")
+        .build();
+
+    let chain = pipeline::new()
+        .chain(parallel!(
+            agent_ops::extract(names_extractor),
+            agent_ops::extract(topics_extractor),
+            agent_ops::extract(sentiment_extractor),
+        ))
+        .map(|(names, topics, sentiment)| {
+            match (names, topics, sentiment) {
+                (Ok(names), Ok(topics), Ok(sentiment)) => {
+                    format!(
+                        "Extracted names: {names}\nExtracted topics: {topics}\nExtracted sentiment: {sentiment}",
+                        names = names.names.join(", "),
+                        topics = topics.topics.join(", "),
+                        sentiment = sentiment.sentiment,
+                    )
+                }
+                _ => "Failed to extract names, topics, or sentiment".to_string(),
+            }
+        });
+
+    let response = chain.call("Screw you Putin!").await;
+
+    println!("Text analysis:\n{response}");
+
+    Ok(())
+}