Agent example uses DeepSeek.

.github/workflows/rust.yml

@@ -357,6 +357,7 @@ jobs:
         cargo clippy --all-targets --all-features --target x86_64-unknown-linux-gnu --locked
         cd agent
         cargo fmt -- --check
+        cargo clippy --all-targets --all-features --locked
 
   lint-cargo-clippy:
     runs-on: ubuntu-latest

examples/agent/Cargo.toml

@@ -10,7 +10,7 @@ clap = { version = "4.5.26", features = ["derive"] }
 reqwest = { version = "0.11.24", default-features = false, features = [
     "rustls-tls",
 ] }
-rig-core = "0.6.1"
+rig-core = "0.7.0"
 serde = { version = "1.0.152", features = ["derive"] }
 serde_json = "1.0.93"
 tokio = { version = "1.25.0", features = ["macros", "rt-multi-thread"] }

examples/agent/src/main.rs

@@ -6,10 +6,10 @@
 use anyhow::Result;
 use clap::Parser;
 use reqwest::Client;
 use rig::{
     agent::AgentBuilder,
     completion::{Chat, Message, ToolDefinition},
-    providers::openai,
+    providers::{openai, deepseek},
     tool::Tool,
 };
 use serde::{Deserialize, Serialize};
@@ -146,8 +146,7 @@
 async fn main() {
     let opt = Opt::parse();
 
-    let openai = openai::Client::from_env();
-    let model = openai.completion_model(&opt.model);
+    // let model = agent_from_opts(&opt.model);
     let node_service = LineraNodeService::new(opt.node_service_url.parse().unwrap()).unwrap();
 
     let graphql_def = node_service.get_graphql_definition().await.unwrap();
@@ -156,15 +155,7 @@
         graphql_def
     );
 
-    // Configure the agent
-    let agent = AgentBuilder::new(model)
-        .preamble(PREAMBLE)
-        .context(LINERA_CONTEXT)
-        .context(&graphql_context)
-        .tool(node_service)
-        .build();
-
-    chat(agent).await;
+    start_chat(&opt.model, &graphql_context, node_service).await;
 }
 
 #[derive(Deserialize)]
@@ -239,6 +230,36 @@
     }
 }
 
+async fn start_chat(model_name: &str, graphql_context: &str, node_service: LineraNodeService) {
+    match model_name {
+        _ if model_name.starts_with("gpt") => {
+            let openai = openai::Client::from_env();
+            let model = openai.completion_model(model_name);
+            let agent = AgentBuilder::new(model)
+                .preamble(PREAMBLE)
+                .context(LINERA_CONTEXT)
+                .context(graphql_context)
+                .tool(node_service)
+                .build();
+
+            chat(agent).await;
+        }
+        _ if model_name.starts_with("deepseek") => {
+            let deepseek = deepseek::Client::from_env();
+            let model = deepseek.completion_model(model_name);
+            let agent = AgentBuilder::new(model)
+                .preamble(PREAMBLE)
+                .context(LINERA_CONTEXT)
+                .context(graphql_context)
+                .tool(node_service)
+                .build();
+
+            chat(agent).await;
+        }
+        _ => panic!("Model {model_name} not recoginised. Try `gpt-*` or `deepseek-*`"),
+    }
+}
+
 pub async fn chat(chatbot: impl Chat) {
     let stdin = io::stdin();
     let mut stdout = io::stdout();