Google AI

pip install risicare google-generativeai

import risicare
import google.generativeai as genai
 
risicare.init()
 
genai.configure(api_key="your-api-key")
model = genai.GenerativeModel("gemini-1.5-pro")
 
# Automatically traced
response = model.generate_content("What is the capital of France?")

response = model.generate_content(
    "Write a poem about coding",
    stream=True
)
 
for chunk in response:
    print(chunk.text, end="")

chat = model.start_chat()
 
response1 = chat.send_message("Hello!")
response2 = chat.send_message("What did I just say?")

def get_weather(location: str) -> str:
    return f"Weather in {location}: Sunny"
 
model = genai.GenerativeModel(
    "gemini-1.5-pro",
    tools=[get_weather]
)
 
response = model.generate_content("What's the weather in Paris?")