from pydantic import BaseModel
class theBook(BaseModel):name: strwriter: str
class theFour(BaseModel):books: list[theBook]
completion = client.beta.chat.completions.parse(model="gpt-4o-2024-08-06",messages=[{"role": "system", "content": "Extract the event information."},{"role": "user", "content": "告诉我四大名著分别是什么，以及他们的作者是谁"},],response_format = theFour,)
response = completion.choices[0].message.parsed

"""Install the Google AI Python SDK
$ pip install google-generativeai$ pip install google.ai.generativelanguage"""
import osimport google.generativeai as genaifrom google.ai.generativelanguage_v1beta.types import content
genai.configure(api_key=os.environ["GEMINI_API_KEY"])
# Create the modelgeneration_config = {"temperature": 1,"top_p": 0.95,"top_k": 64,"max_output_tokens": 8192,"response_schema": content.Schema(type = content.Type.OBJECT,enum = "[]",required = "["books"]",properties = {"books": content.Schema(type = content.Type.ARRAY,items = content.Schema(type = content.Type.OBJECT,properties = {"name": content.Schema(type = content.Type.STRING,),"writer": content.Schema(type = content.Type.STRING,),},),),},),"response_mime_type": "application/json",}
model = genai.GenerativeModel(model_name="gemini-1.5-flash",generation_config=generation_config,# safety_settings = Adjust safety settings# See https://ai.google.dev/gemini-api/docs/safety-settingssystem_instruction="Extract the event information.",)
chat_session = model.start_chat(history=[{"role": "user","parts": ["告诉我四大名著分别是什么，以及他们的作者是谁",],},{"role": "model","parts": ["```json\n{\"books\": [{\"name\": \"红楼梦\", \"writer\": \"曹雪芹\"}, {\"name\": \"三国演义\", \"writer\": \"罗贯中\"}, {\"name\": \"水浒传\", \"writer\": \"施耐庵\"}, {\"name\": \"西游记\", \"writer\": \"吴承恩\"}]} \n```",],},])
response = chat_session.send_message("INSERT_INPUT_HERE")
print(response.text)