Added persona_tool for generating responses based on persona descriptions via OpenAI API.
This commit is contained in:
+14
-20
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user