Merge pull request #30 from anilaltuner/main

o1-openrouter added
andthattoo · Dec 18, 2024 · f622133 · f622133
2 parents 46dc2c5 + f34abe4
commit f622133
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/program/models.rs b/src/program/models.rs
@@ -194,6 +194,9 @@ pub enum Model {
 
     #[serde(rename = "nvidia/llama-3.1-nemotron-70b-instruct")]
     ORNemotron70B,
+
+    #[serde(rename = "openai/o1")]
+    OROpenAIO1,
 }
 
 impl Model {
@@ -327,6 +330,7 @@ impl From<Model> for ModelProvider {
             Model::ORQwenQwq => ModelProvider::OpenRouter,
             Model::ORNemotron70B => ModelProvider::OpenRouter,
             Model::ORNousHermes405B => ModelProvider::OpenRouter,
+            Model::OROpenAIO1 => ModelProvider::OpenRouter,
         }
     }
 }