| """ |
| Chat analysis module for generating response suggestions. |
| |
| This module provides functionality to generate response suggestions |
| using OpenAI's GPT models. |
| """ |
|
|
| from openai import OpenAI |
|
|
|
|
| def get_suggestion(user_input): |
| """ |
| Generate a response suggestion based on user input. |
| |
| This function uses OpenAI's GPT model to generate a response suggestion |
| for a given user input. The response is formatted in XML with suggestion |
| and example tags. |
| |
| Args: |
| user_input (str): The user's input message |
| |
| Returns: |
| str: The generated response suggestion |
| """ |
| client = OpenAI() |
| |
| completion = client.chat.completions.create( |
| model="gpt-4o", |
| messages=[ |
| { |
| "role": "system", |
| "content": "你是一個可以提供尊重對方且讓提問者開心的回覆的助手，" |
| "使用者將會輸入對方的問題，回覆時請撇除預設立場且一定要給出回復不要轉移話題，" |
| "可以偏愛提問者，要強調出他特殊的魅力。請使用XML格式，" |
| "生成針對使用者的應對建議，以及實際舉例來回覆使用者訊息，" |
| "使用<suggestion>和<example>標記來格式化你的回覆，<example>不超過30字" |
| }, |
| {"role": "user", "content": user_input} |
| ] |
| ) |
| |
| return completion.choices[0].message.content |
|
|
|
|
| if __name__ == "__main__": |
| |
| dialogue = "你喜歡過除了我以外的異性嗎" |
| response = get_suggestion(dialogue) |
| print(response) |
|
|
