From: ibidyouadu <angel.d.umana@gmail.com>
Date: Mon, 21 Aug 2023 00:12:03 +0000 (-0500)
Subject: added json.dumps to correctly handle dictionary data passed to redis push
X-Git-Url: http://git.angelumana.com/?a=commitdiff_plain;h=b08d02f56d2580fb9e8b33c876aa3b8cce207584;p=diane%2F.git

added json.dumps to correctly handle dictionary data passed to redis push
---

diff --git a/main.py b/main.py
index 34866db..6a54e5f 100644
--- a/main.py
+++ b/main.py
@@ -5,6 +5,7 @@ import random
 import os
 import openai
 import redis
+import json
 from settings import OPENAI_API_KEY, OPENWEATHER_API_KEY
 
 openai.api_key = OPENAI_API_KEY
@@ -21,10 +22,11 @@ prompt = {
     "role": "system",
     "content": "You are a helpful assistant named Diane."
 }
-r.rpush(conversation_key, prompt)
+init_payload = json.dumps(prompt)
+r.rpush(conversation_key, init_payload)
 
-@app.route("/", methods=["POST"])
 
+@app.route("/", methods=["POST"])
 def main():
     # Get user input
     input_text = request.values.get("Body", "")
@@ -43,8 +45,8 @@ def main():
     conversation = r.lrange(conversation_key, 0, -1)
     messages = [eval(message) for message in conversation]
     messages.append(input_message)
-    print(messages)
-    r.rpush(conversation_key, input_message)
+    input_payload = json.dumps(input_message)
+    r.rpush(conversation_key, input_payload)
 
     # Call OpenAI API
     openai_res = openai.ChatCompletion.create(
@@ -60,7 +62,8 @@ def main():
         "role": "assistant",
         "content": output_text
     }
-    r.rpush(conversation_key, output_message)
+    output_payload = json.dumps(output_message)
+    r.rpush(conversation_key, output_payload)
 
     # Return OpenAI message
     msg.body(output_text)