feat(2658): introduce Adapter

src/cli/llm/adapter.rs

@@ -0,0 +1,25 @@
+use std::fmt::{Display, Formatter};
+
+#[derive(Clone)]
+pub enum Adapter {
+    Groq(GroqModel),
+}
+
+#[derive(Clone)]
+pub enum GroqModel {
+    Llama38b8192,
+}
+
+impl Display for Adapter {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        let str = match self { Adapter::Groq(g) => g.to_string() };
+        write!(f, "{}", str)
+    }
+}
+
+impl Display for GroqModel {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        let str = match self { GroqModel::Llama38b8192 => "Llama3_8b_8192" };
+        write!(f, "{}", str)
+    }
+}

src/cli/llm/mod.rs

@@ -1,7 +1,9 @@
 mod error;
 pub mod infer_type_name;
 pub use error::Error;
 use error::Result;
 pub use infer_type_name::InferTypeName;
 mod wizard;
+mod adapter;
+
 pub use wizard::Wizard;

src/cli/llm/wizard.rs

@@ -1,20 +1,19 @@
 use derive_setters::Setters;
 use genai::chat::{ChatOptions, ChatRequest, ChatResponse};
 use genai::Client;
-
+use crate::cli::llm::adapter::Adapter;
 use super::Result;
 
 #[derive(Setters, Clone)]
 pub struct Wizard<Q, A> {
     client: Client,
-    // TODO: change model to enum
-    model: String,
+    model: Adapter,
     _q: std::marker::PhantomData<Q>,
     _a: std::marker::PhantomData<A>,
 }
 
 impl<Q, A> Wizard<Q, A> {
-    pub fn new(model: String) -> Self {
+    pub fn new(model: Adapter) -> Self {
         Self {
             client: Client::builder()
                 .with_chat_options(ChatOptions::default().with_json_mode(true))
@@ -32,7 +31,7 @@
     {
         let response = self
             .client
-            .exec_chat(self.model.as_str(), q.try_into()?, None)
+            .exec_chat(self.model.to_string().as_str(), q.try_into()?, None)
             .await?;
         A::try_from(response)
     }