Platform guides
Gemini
Add Riza as a Python code interpreter to Gemini
Add Riza as a code interpreter to Google’s hosted Gemini model. Download and run the full example on GitHub.
import os
import rizaio
import google.generativeai as genai
# requires a valid Google Gemini API key in the GOOGLE_API_KEY env var
# requires a valid Riza API key in the RIZA_API_KEY env var
def execute_python(code:str):
""" Executes a Python script and returns whatever was printed to stdout.
The Python runtime does not have filesystem access, but does include the
entire standard library. Make HTTP requests with the httpx or requests
libraries. Read input from stdin and write output to stdout.
"""
riza = rizaio.Riza()
resp = riza.command.exec(
language="PYTHON",
code=code
)
return resp.stdout
def main():
genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
model = genai.GenerativeModel('gemini-1.0-pro-latest', tools=[execute_python])
chat = model.start_chat(enable_automatic_function_calling=True)
response = chat.send_message('Please base64 encode the last message I sent')
print(response.text)
for content in chat.history:
print(content.role, "->",
[type(part).to_dict(part) for part in content.parts])
print('-'*80)
if __name__ == "__main__":
main()