Skip to content

Commit 5c6766e

Browse files
authored
Support system message for Gemini (#4869)
1 parent 265c621 commit 5c6766e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

marimo/_ai/llm/_impl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,6 @@ def __init__(
181181
self.system_message = system_message
182182
self.api_key = api_key
183183
self.base_url = base_url
184-
self.system_message = system_message
185184

186185
@property
187186
def _require_api_key(self) -> str:
@@ -313,6 +312,7 @@ def __call__(
313312
genai.configure(api_key=self._require_api_key)
314313
client = genai.GenerativeModel(
315314
model_name=self.model,
315+
system_instruction=self.system_message,
316316
generation_config=genai.GenerationConfig(
317317
max_output_tokens=config.max_tokens,
318318
temperature=config.temperature,

0 commit comments

Comments
 (0)