Merge pull request #2219 from monosans/patch-01

feat(GeminiPro): add support for system prompt
xtekky · Sep 29, 2024 · 0e32e27 · 0e32e27
2 parents 0deb0f6 + c2f7248
commit 0e32e27
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/g4f/Provider/GeminiPro.py b/g4f/Provider/GeminiPro.py
@@ -54,6 +54,7 @@ async def create_async_generator(
                     "parts": [{"text": message["content"]}]
                 }
                 for message in messages
+                if message["role"] != "system"
             ]
             if image is not None:
                 image = to_bytes(image)
@@ -73,6 +74,13 @@ async def create_async_generator(
                     "topK": kwargs.get("top_k"),
                 }
             }
+            system_prompt = "\n".join(
+                message["content"]
+                for message in messages
+                if message["role"] == "system"
+            )
+            if system_prompt:
+                data["system_instruction"] = {"parts": {"text": system_prompt}}
             async with session.post(url, params=params, json=data) as response:
                 if not response.ok:
                     data = await response.json()