Skip to content

Commit 62490ec

Browse files
committed
feat: Add CloudModel import and specify GOOGLE_GEMINI model
1 parent b5ad51c commit 62490ec

File tree

1 file changed

+2
-1
lines changed
  • examples/bedtime-story-teller/python

1 file changed

+2
-1
lines changed

examples/bedtime-story-teller/python/main.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,14 @@
33
# SPDX-License-Identifier: MPL-2.0
44

55
import os
6-
from arduino.app_bricks.cloud_llm import CloudLLM
6+
from arduino.app_bricks.cloud_llm import CloudLLM, CloudModel
77
from arduino.app_bricks.web_ui import WebUI
88
from arduino.app_utils import App
99

1010

1111
llm = CloudLLM(
1212
api_key=os.getenv("API_KEY"), # Make sure to set your API key in the environment variable
13+
model=CloudModel.GOOGLE_GEMINI,
1314
system_prompt="You are a bedtime story teller. Tell short, funny and relaxing stories that are suitable for children. Use a simple language and avoid long sentences.",
1415
)
1516
llm.with_memory()

0 commit comments

Comments
 (0)