From ca56bfc6f6310e6afa932416a5d2e4e45bd45933 Mon Sep 17 00:00:00 2001
From: bucolucas <bucolucas@gmail.com>
Date: Sat, 17 Aug 2024 22:12:01 -0500
Subject: [PATCH] Added persona_tool for generating responses based on persona
 descriptions via OpenAI API.

---
 persona_tool.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 persona_tool.py

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']