diff --git a/examples/pipelines/providers/google_manifold_pipeline.py b/examples/pipelines/providers/google_manifold_pipeline.py index 8e71bddd..c313aa2e 100644 --- a/examples/pipelines/providers/google_manifold_pipeline.py +++ b/examples/pipelines/providers/google_manifold_pipeline.py @@ -128,13 +128,13 @@ def pipe( "parts": [{"text": message["content"]}] }) - if "gemini-1.5" in model_id: - model = genai.GenerativeModel(model_name=model_id, system_instruction=system_message) - else: + if model_id.startswith("gemini-1.0") or model_id.startswith("gemini-pro"): if system_message: contents.insert(0, {"role": "user", "parts": [{"text": f"System: {system_message}"}]}) - + model = genai.GenerativeModel(model_name=model_id) + else: + model = genai.GenerativeModel(model_name=model_id, system_instruction=system_message) generation_config = GenerationConfig( temperature=body.get("temperature", 0.7),