import google.generativeai as genai

API_KEY = "AIzaSyCkqv9dWrlbRjv9fHO_O8jBORGfYVPJTnY"

def call_gemini(prompt="", given_text=None, given_image=None, generation_config=None, safety_settings=None):
    genai.configure(api_key=API_KEY)
    generation_config = {
            "temperature": 0.8,
            "top_p": 1,
            "top_k": 32,
            "max_output_tokens": 8192,
    }

    safety_settings = [
            {
                "category": "HARM_CATEGORY_HARASSMENT",
                "threshold": "BLOCK_ONLY_HIGH"
            },
            {
                "category": "HARM_CATEGORY_HATE_SPEECH",
                "threshold": "BLOCK_ONLY_HIGH"
            },
            {
                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
                "threshold": "BLOCK_ONLY_HIGH"
            },
            {
                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
                "threshold": "BLOCK_ONLY_HIGH"
            },
        ]
    
    model = genai.GenerativeModel(model_name='gemini-pro',
                                generation_config=generation_config,
                                safety_settings=safety_settings)
    
    prompt_parts = "如何学习rust语言?"

    response = model.generate_content(prompt_parts)
    print(response.text)

if __name__ == "__main__":
    call_gemini()