Use native system instructions for Gemini 1.5 models

2025-05-14 01:20:48 +00:00 · 2024-08-02 16:45:46 +03:00 · 2024-08-02 16:45:46 +03:00 · a94512dad4
commit a94512dad4
parent 342566765f
1 changed files with 8 additions and 5 deletions
--- a/examples/pipelines/providers/google_manifold_pipeline.py
+++ b/examples/pipelines/providers/google_manifold_pipeline.py
@ -126,6 +126,9 @@ class Pipeline:
                            "parts": [{"text": message["content"]}]
                        })
            
+            if "gemini-1.5" in model_id:
+                model = genai.GenerativeModel(model_name=model_id, system_instruction=system_message)
+            else:
                if system_message:
                    contents.insert(0, {"role": "user", "parts": [{"text": f"System: {system_message}"}]})