Added persona_tool for generating responses based on persona descriptions via OpenAI API.

This commit is contained in:
2024-08-17 22:12:06 -05:00
parent ca56bfc6f6
commit c1258ddb55
+14 -20
View File
@@ -1,23 +1,17 @@
# persona_tool.py
from base_tool import BaseTool
import openai
class PersonaTool(BaseTool): class PersonaTool(BaseTool):
def generate_response(self, persona_description, query): def __init__(self):
""" super().__init__()
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. def generate_response(self, persona_description: str, query: str):
:param query: The query to be answered by the persona.
:return: Response from OpenAI API.
""" """
# Call OpenAI API with the provided persona and query Makes a call to the OpenAI API using the persona as a system prompt.
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo", Parameters:
messages=[ persona_description (str): Description of the persona.
{"role": "system", "content": persona_description}, query (str): Query to be processed.
{"role": "user", "content": query}
] Returns:
) str: The response generated by the OpenAI API.
return response.choices[0].message['content'] """
# Call to OpenAI API would be here
pass