OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-26 18:15:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use native system instructions for Gemini 1.5 models

Browse Source
This commit is contained in:
Rotem Dan 2024-08-02 16:45:46 +03:00
parent 342566765f
commit a94512dad4
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/pipelines/providers/google_manifold_pipeline.py
View File

@ -126,6 +126,9 @@ class Pipeline:
"parts": [{"text": message["content"]}] "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: if system_message:
contents.insert(0, {"role": "user", "parts": [{"text": f"System: {system_message}"}]}) contents.insert(0, {"role": "user", "parts": [{"text": f"System: {system_message}"}]})