Add display capabilities to python objects

mistralrs-pyo3/Cargo.toml

@@ -30,6 +30,7 @@ tokio.workspace = true
 image.workspace = true
 reqwest.workspace = true
 base64.workspace = true
+pyo3_special_method_derive = "0.4.1"
 
 [build-dependencies]
 pyo3-build-config = "0.22"

mistralrs-pyo3/src/anymoe.rs

@@ -1,7 +1,8 @@
 use pyo3::{pyclass, pymethods};
+use pyo3_special_method_derive::{Dict, Dir, Getattr, Repr, Str};
 
 #[pyclass]
-#[derive(Clone, Debug)]
+#[derive(Clone, Debug, Str, Repr, Dir, Dict, Getattr)]
 pub enum AnyMoeExpertType {
  FineTuned {},
  LoraAdapter {
@@ -28,7 +29,7 @@
  }
 }
 
-#[derive(Clone)]
+#[derive(Clone, Str, Repr, Dir, Dict, Getattr)]
 #[pyclass]
 pub struct AnyMoeConfig {
  pub(crate) hidden_size: usize,

mistralrs-pyo3/src/lib.rs

@@ -5,6 +5,7 @@ use base64::{engine::general_purpose, Engine};
 use candle_core::{quantized::GgmlDType, Result};
 use either::Either;
 use indexmap::IndexMap;
+use pyo3_special_method_derive::{Dict, Dir, Getattr, Repr, Str};
 use std::{
  cell::RefCell,
  collections::HashMap,
@@ -83,6 +84,7 @@ fn parse_isq(s: &str) -> std::result::Result<GgmlDType, String> {
 }
 
 #[pyclass]
+#[derive(Str, Repr, Dir, Dict, Getattr)]
 /// An object wrapping the underlying Rust system to handle requests and process conversations.
 struct Runner {
  runner: Arc<MistralRs>,
@@ -890,7 +892,7 @@ impl Runner {
 }
 
 #[pyclass]
-#[derive(Debug)]
+#[derive(Debug, Str, Repr, Dir, Dict, Getattr)]
 /// An OpenAI API compatible completion request.
 struct CompletionRequest {
  _model: String,
@@ -976,7 +978,7 @@ impl CompletionRequest {
 }
 
 #[pyclass]
-#[derive(Debug)]
+#[derive(Debug, Str, Repr, Dir, Dict, Getattr)]
 /// An OpenAI API compatible chat completion request.
 struct ChatCompletionRequest {
  #[allow(clippy::type_complexity)]

mistralrs-pyo3/src/which.rs

@@ -1,8 +1,9 @@
 use mistralrs_core::{NormalLoaderType, VisionLoaderType};
 use pyo3::pyclass;
+use pyo3_special_method_derive::{Dict, Dir, Getattr, Repr, Str};
 
 #[pyclass(eq, eq_int)]
-#[derive(Debug, Clone, PartialEq)]
+#[derive(Debug, PartialEq, Clone, Str, Repr, Dir, Dict, Getattr)]
 pub enum Architecture {
  Mistral,
  Gemma,
@@ -32,7 +33,7 @@ impl From<Architecture> for NormalLoaderType {
 }
 
 #[pyclass(eq, eq_int)]
-#[derive(Debug, Clone, PartialEq)]
+#[derive(Debug, PartialEq, Clone, Str, Repr, Dir, Dict, Getattr)]
 pub enum VisionArchitecture {
  Phi3V,
  Idefics2,
@@ -52,7 +53,7 @@ impl From<VisionArchitecture> for VisionLoaderType {
 }
 
 #[pyclass]
-#[derive(Clone)]
+#[derive(Clone, Str, Repr, Dir, Dict, Getattr)]
 pub enum Which {
  #[pyo3(constructor = (
  model_id,