Make all parameters optional (#80)

TilmanGriesel · Jan 18, 2025 · 969294d · 969294d
1 parent 331a0d0
commit 969294d
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/services/api/src/core/pipeline_config.py b/services/api/src/core/pipeline_config.py
@@ -15,11 +15,15 @@ def _default_none() -> None:
 class QueryPipelineConfig:
     """Base configuration for all pipelines."""
 
-    # Required parameters
-    ollama_url: str
-    embedding_model: str
-    model_name: str
-    system_prompt: str
+    # Common provider parameters
+    provider: str = field(default=ModelProvider.OLLAMA)
+    ollama_url: Optional[str] = None
+    hf_api_key: Optional[str] = field(default_factory=_default_none)
+
+    embedding_model: Optional[str] = None
+    model_name: Optional[str] = None
+    system_prompt: Optional[str] = None
+    allow_model_pull: bool = field(default=True)
 
     # Elasticsearch parameters
     es_url: str
@@ -53,11 +57,6 @@ class QueryPipelineConfig:
     tfs_z: Optional[float] = None
     stop_sequence: Optional[str] = None
 
-    # Common provider parameters
-    provider: str = field(default=ModelProvider.OLLAMA)
-    hf_api_key: Optional[str] = field(default_factory=_default_none)
-    allow_model_pull: bool = field(default=True)
-
     def __post_init__(self):
         """Validate configuration after initialization."""
         if self.provider == ModelProvider.HUGGINGFACE and not self.hf_api_key: