import google.generativeai as palm
import os
palm.configure(api_key=os.environ.get("openai_key"))

defaults = {
  'model': 'models/chat-bison-001',
  'temperature': 0.25,
  'candidate_count': 1,
  'top_k': 40,
  'top_p': 0.95,
}
context = "Your a computer failure assitant"
examples = [
  [
    "Hey my computer is broken",
    "Hey what is the issue with your computer?"
  ]
]
messages = []
messages.append("NEXT REQUEST")
response = palm.chat(
  **defaults,
  context=context,
  examples=examples,
  messages=messages
)
print(response.last) # Response of the AI to your most recent request