diff --git a/persona_tool.py b/persona_tool.py new file mode 100644 index 0000000..15794b5 --- /dev/null +++ b/persona_tool.py @@ -0,0 +1,23 @@ +# persona_tool.py + +from base_tool import BaseTool +import openai + +class PersonaTool(BaseTool): + def generate_response(self, persona_description, query): + """ + Generates a response based on the persona description using OpenAI API. + + :param persona_description: Description of the persona to be used as a system prompt. + :param query: The query to be answered by the persona. + :return: Response from OpenAI API. + """ + # Call OpenAI API with the provided persona and query + response = openai.ChatCompletion.create( + model="gpt-3.5-turbo", + messages=[ + {"role": "system", "content": persona_description}, + {"role": "user", "content": query} + ] + ) + return response.choices[0].message['content']