48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
"""
|
|
At the command line, only need to run once to install the package via pip:
|
|
|
|
$ pip install google-generativeai
|
|
"""
|
|
|
|
import google.generativeai as genai
|
|
|
|
genai.configure(api_key="YOUR_API_KEY")
|
|
|
|
# Set up the model
|
|
generation_config = {
|
|
"temperature": 1,
|
|
"top_p": 1,
|
|
"top_k": 1,
|
|
"max_output_tokens": 2048,
|
|
}
|
|
|
|
safety_settings = [
|
|
{
|
|
"category": "HARM_CATEGORY_HARASSMENT",
|
|
"threshold": "BLOCK_MEDIUM_AND_ABOVE"
|
|
},
|
|
{
|
|
"category": "HARM_CATEGORY_HATE_SPEECH",
|
|
"threshold": "BLOCK_MEDIUM_AND_ABOVE"
|
|
},
|
|
{
|
|
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
|
|
"threshold": "BLOCK_MEDIUM_AND_ABOVE"
|
|
},
|
|
{
|
|
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
|
|
"threshold": "BLOCK_MEDIUM_AND_ABOVE"
|
|
}
|
|
]
|
|
|
|
model = genai.GenerativeModel(model_name="gemini-pro",
|
|
generation_config=generation_config,
|
|
safety_settings=safety_settings)
|
|
|
|
prompt_parts = [
|
|
"As an expert programmer and web researcher, I will provide you with blog content. Your task is to understand the blog content and do web research around the main keywords. Check if the blog content is about programming then provide me with original code examples, relevant to the blog content. The provided code example should be of high coding standards, include docstring and follow pep8 standards. Do not provide explanations for your response.\nBlog content: \"\"\" {blog_content} \"\"\"\n ",
|
|
]
|
|
|
|
response = model.generate_content(prompt_parts)
|
|
print(response.text)
|