Fix

council/llm/__init__.py

@@ -72,5 +72,7 @@ def _build_llm(llm_config: LLMConfigObject) -> LLMBase:
         return OpenAILLM.from_config(llm_config)
     elif provider.is_of_kind(LLMProviders.Anthropic):
         return AnthropicLLM.from_config(llm_config)
+    elif provider.is_of_kind(LLMProviders.Gemini):
+        return GeminiLLM.from_config(llm_config)
 
     raise ValueError(f"Provider `{provider.kind}` not supported by Council")

requirements.txt

@@ -21,5 +21,4 @@ beautifulsoup4~=4.12.2
 
 toml~=0.10.2
 nbformat~=5.9.2
-nbconvert~=7.11.0
 PyYAML~=6.0.1

tests/unit/llm/test_llm_config_object.py

@@ -1,4 +1,4 @@
-from council import OpenAILLM, AzureLLM, AnthropicLLM
+from council import OpenAILLM, AzureLLM, AnthropicLLM, GeminiLLM
 from council.llm import get_llm_from_config, LLMFallback, OpenAIChatGPTConfiguration
 from council.llm.llm_config_object import LLMConfigObject
 from council.utils import OsEnviron
@@ -52,6 +52,19 @@ def test_anthropic_from_yaml():
         assert isinstance(llm, AnthropicLLM)
 
 
+def test_gemini_from_yaml():
+    filename = get_data_filename(LLModels.Gemini)
+
+    with OsEnviron("GEMINI_API_KEY", "a-key"):
+        actual = LLMConfigObject.from_yaml(filename)
+        llm = GeminiLLM.from_config(actual)
+        assert isinstance(llm, GeminiLLM)
+        assert llm.configuration.top_k.value == 8
+
+        llm = get_llm_from_config(filename)
+        assert isinstance(llm, GeminiLLM)
+
+
 def test_azure_with_openai_fallback_from_yaml():
     filename = get_data_filename(LLModels.AzureWithFallback)