feat(llms): default values for default_llm_factories

packages/ragbits-core/src/ragbits/core/config.py

@@ -13,10 +13,10 @@ class CoreConfig(BaseModel):
     prompt_path_pattern: str = "**/prompt_*.py"
 
     # Path to a functions that returns LLM objects, e.g. "my_project.llms.get_llm"
-    default_llm_factories: dict[LLMType, str | None] = {
-        LLMType.TEXT: None,
-        LLMType.VISION: None,
-        LLMType.STRUCTURED_OUTPUT: None,
+    default_llm_factories: dict[LLMType, str] = {
+        LLMType.TEXT: "ragbits.core.llms.factory.simple_litellm_factory",
+        LLMType.VISION: "ragbits.core.llms.factory.simple_litellm_vision_factory",
+        LLMType.STRUCTURED_OUTPUT: "ragbits.core.llms.factory.simple_litellm_structured_output_factory",
     }
 
 

packages/ragbits-core/src/ragbits/core/llms/factory.py

@@ -49,9 +49,6 @@ def get_default_llm(llm_type: LLMType = LLMType.TEXT) -> LLM:
     if llm_type not in core_config.default_llm_factories:
         raise ValueError(f"Default LLM of type {llm_type} is not defined in pyproject.toml config.")
     factory = core_config.default_llm_factories[llm_type]
-    if factory is None:
-        raise ValueError("Default LLM factory is not set")
-
     return get_llm_from_factory(factory)
 
 
@@ -75,3 +72,14 @@ def simple_litellm_vision_factory() -> LLM:
         LLM: An instance of the LiteLLM.
     """
     return LiteLLM(model_name="gpt-4o-mini")
+
+
+def simple_litellm_structured_output_factory() -> LLM:
+    """
+    A basic LLM factory that creates an LiteLLM instance with the structured output.
+    Functionality supported
+
+    Returns:
+        LLM: An instance of the LiteLLM.
+    """
+    return LiteLLM(model_name="gpt-4o-mini-2024-07-18", use_structured_output=True)