|
""" |
|
At the command line, only need to run once to install the package via pip: |
|
$ pip install google-generativeai |
|
""" |
|
import os |
|
from dotenv import load_dotenv, find_dotenv |
|
load_dotenv(find_dotenv()) |
|
|
|
import google.generativeai as palm |
|
|
|
palm.configure(api_key=os.environ["PALM_API_KEY"]) |
|
|
|
defaults = { |
|
'model': 'models/chat-bison-001', |
|
'temperature': 0.25, |
|
'candidate_count': 1, |
|
'top_k': 40, |
|
'top_p': 0.95, |
|
} |
|
context = "" |
|
examples = [] |
|
messages = [] |
|
messages.append("what is 1 + 3 ?") |
|
response = palm.chat( |
|
**defaults, |
|
context=context, |
|
examples=examples, |
|
messages=messages |
|
) |
|
|
|
print(response.last) |
