From c1258ddb55df3a13ce9f98cd5c92c8e45c5633ab Mon Sep 17 00:00:00 2001 From: bucolucas Date: Sat, 17 Aug 2024 22:12:06 -0500 Subject: [PATCH] Added persona_tool for generating responses based on persona descriptions via OpenAI API. --- persona_tool.py | 34 ++++++++++++++-------------------- 1 file changed, 14 insertions(+), 20 deletions(-) diff --git a/persona_tool.py b/persona_tool.py index 15794b5..5fffa86 100644 --- a/persona_tool.py +++ b/persona_tool.py @@ -1,23 +1,17 @@ -# 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. + def __init__(self): + super().__init__() - :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. + def generate_response(self, persona_description: str, query: str): """ - # 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'] + Makes a call to the OpenAI API using the persona as a system prompt. + + Parameters: + persona_description (str): Description of the persona. + query (str): Query to be processed. + + Returns: + str: The response generated by the OpenAI API. + """ + # Call to OpenAI API would be here + pass