Change docs theme

Author: YoanSallami

docs/stylesheets/extra.css

@@ -0,0 +1,38 @@
+/* Light mode */
+[data-md-color-scheme="default"] {
+    --md-default-bg-color: #fdf6e3;
+    --md-default-fg-color: #002c37;
+
+    --md-primary-bg-color: #fdf6e3;
+    --md-primary-fg-color: #002c37;
+
+    --md-accent-fg-color: #d56564;
+
+    --md-code-fg-color: #002c37;
+    --md-code-bg-color: #ffffff;
+
+    --md-footer-fg-color: #002c37;
+    --md-footer-bg-color: #002c37;
+    --md-footer-fg-color--light: #ffffff;
+    --md-footer-bg-color--dark: #002c37;
+}
+
+/* Dark mode */
+[data-md-color-scheme="slate"] {
+    --md-default-bg-color: #002c37;
+    --md-default-fg-color: #fdf6e3;
+
+    --md-primary-bg-color: #fdf6e3;
+    --md-primary-fg-color: #0a3642;
+
+    --md-accent-fg-color: #d56564;
+    --md-accent-bg-color: #002c37;
+
+    --md-code-fg-color: #ffffff;
+    --md-code-bg-color: #0a3642;
+
+    --md-footer-fg-color: #002c37;
+    --md-footer-bg-color: #002c37;
+    --md-footer-fg-color--light: #ffffff;
+    --md-footer-bg-color--dark: #0a3642;
+}

mkdocs.yml

@@ -8,10 +8,30 @@ theme:
   favicon: assets/synalinks_favicon.png
   features:
     - content.code.copy
-  palette:
-    primary: black
   icon:
     repo: fontawesome/brands/github
+  theme:
+  palette:
+    # Palette toggle for automatic mode
+    - media: "(prefers-color-scheme)"
+      toggle:
+        icon: material/brightness-auto
+        name: Switch to light mode
+    # Palette toggle for light mode
+    - media: "(prefers-color-scheme: light)"
+      scheme: default 
+      toggle:
+        icon: material/brightness-7
+        name: Switch to dark mode
+    # Palette toggle for dark mode
+    - media: "(prefers-color-scheme: dark)"
+      scheme: slate
+      toggle:
+        icon: material/brightness-4
+        name: Switch to system preference
+
+extra_css:
+  - stylesheets/extra.css
 
 markdown_extensions:
   - pymdownx.highlight:

synalinks/src/optimizers/random_few_shot_test.py

@@ -20,7 +20,7 @@ class RandomFewShotTest(testing.TestCase):
     @patch("litellm.embedding")
     @patch("litellm.completion")
     async def test_random_few_shot_training(self, mock_completion, mock_embedding):
-        language_model = LanguageModel(model="ollama/deepseek-r1")
+        language_model = LanguageModel(model="ollama/mistral")
 
         embedding_model = EmbeddingModel(model="ollama/all-minilm")