update workflows and related tests

Author: erhant

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "dkn-compute"
-version = "0.2.1"
+version = "0.2.2"
 edition = "2021"
 license = "Apache-2.0"
 readme = "README.md"
@@ -45,7 +45,7 @@ sha3 = "0.10.8"
 fastbloom-rs = "0.5.9"
 
 # workflows
-ollama-workflows = { git = "https://github.com/andthattoo/ollama-workflows", rev = "875f403" }
+ollama-workflows = { git = "https://github.com/andthattoo/ollama-workflows", rev = "31b6da6" }
 
 # peer-to-peer
 libp2p = { git = "https://github.com/anilaltuner/rust-libp2p.git", rev = "7ce9f9e", features = [

Cargo.toml

@@ -256,33 +256,15 @@ You will be asked to provide your choice of models within the CLI. You can also
 
 ```sh
 # macos or linux
-./dkn-compute-launcher -m=llama3.1:latest -m=gpt-3.5-turbo
+./dkn-compute-launcher -m=llama3.1:latest -m=gpt-4o-mini
 ```
 
 ```sh
 # windows
-.\dkn-compute-launcher.exe -m=llama3.1:latest -m=gpt-3.5-turbo
+.\dkn-compute-launcher.exe -m=llama3.1:latest -m=gpt-4o-mini
 ```
 
-[Available models](https://github.com/andthattoo/ollama-workflows/blob/main/src/program/models.rs) are given below:
-
-#### Ollama Models
-
-- `finalend/hermes-3-llama-3.1:8b-q8_0`
-- `phi3:14b-medium-4k-instruct-q4_1`
-- `phi3:14b-medium-128k-instruct-q4_1`
-- `phi3.5:3.8b`
-- `phi3.5:3.8b-mini-instruct-fp16`
-- `llama3.1:latest`
-- `llama3.1:8b-instruct-q8_0`
-- `gemma2:9b-instruct-q8_0`
-
-#### OpenAI Models
-
-- `gpt-3.5-turbo`
-- `gpt-4-turbo`
-- `gpt-4o`
-- `gpt-4o-mini`
+Available models can be seen within the launcher, or under [`ollama-workflows/models`](https://github.com/andthattoo/ollama-workflows/blob/main/src/program/models.rs).
 
 ### Additional Static Nodes
 

docs/NODE_GUIDE.md

@@ -96,11 +96,12 @@ impl DriaComputeNodeConfig {
         log::info!("Node Address:     0x{}", hex::encode(address));
 
         let model_config = ModelConfig::new_from_csv(env::var("DKN_MODELS").ok());
+        #[cfg(not(test))]
         if model_config.models.is_empty() {
             log::error!("No models were provided, make sure to restart with at least one model provided within DKN_MODELS.");
             panic!("No models provided.");
         }
-        log::info!("Models: {}", model_config);
+        log::info!("Models: {:?}", model_config.models);
 
         let p2p_listen_addr = env::var("DKN_P2P_LISTEN_ADDR")
             .map(|addr| addr.trim_matches('"').to_string())

src/config/mod.rs

@@ -143,10 +143,10 @@ mod tests {
 
     #[test]
     fn test_model_matching() {
-        let cfg = ModelConfig::new_from_csv(Some("gpt-3.5-turbo,llama3.1:latest".to_string()));
+        let cfg = ModelConfig::new_from_csv(Some("gpt-4o,llama3.1:latest".to_string()));
         assert_eq!(
             cfg.get_matching_model("openai".to_string()).unwrap().1,
-            Model::GPT3_5Turbo,
+            Model::GPT4o,
             "Should find existing model"
         );
 
@@ -159,7 +159,7 @@ mod tests {
         );
 
         assert!(
-            cfg.get_matching_model("gpt-4o".to_string()).is_err(),
+            cfg.get_matching_model("gpt-4o-mini".to_string()).is_err(),
             "Should not find anything for unsupported model"
         );