返回提交历史
Modified
g4f/Provider/GeminiPro.py
+8
-0
XFEstudio/gpt4free
feat(GeminiPro): add support for system prompt
c2f72489
代码差异
1 个文件
+8
-0
@@ -54,6 +54,7 @@ class GeminiPro(AsyncGeneratorProvider, ProviderModelMixin):
54
54
"parts": [{"text": message["content"]}]
55
55
}
56
56
for message in messages
57
if message["role"] != "system"
57
58
]
58
59
if image is not None:
59
60
image = to_bytes(image)
@@ -73,6 +74,13 @@ class GeminiPro(AsyncGeneratorProvider, ProviderModelMixin):
73
74
"topK": kwargs.get("top_k"),
74
75
}
75
76
}
77
system_prompt = "\n".join(
78
message["content"]
79
for message in messages
80
if message["role"] == "system"
81
)
82
if system_prompt:
83
data["system_instruction"] = {"parts": {"text": system_prompt}}
76
84
async with session.post(url, params=params, json=data) as response:
77
85
if not response.ok:
78
86
data = await response.json()