bot / app.py
Antoine245's picture
Create app.py
b40d855
raw
history blame
591 Bytes
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