update

redis_service.py

@@ -1,4 +1,4 @@
-import os
+import os, requests
 import asyncio
 import redis.asyncio as redis
 from dotenv import load_dotenv
@@ -75,6 +75,55 @@ async def redis_subscriber(redis_client, channel_name):
                     # Extract necessary info from data_dict for sending reply
                     msg_id = data_dict.get('msg_id')
                     content = data_dict.get('content') # This should be the dictionary like {"text": "..."}
+                    print('content["text"]',content["text"])
+
+
+                    response = requests.get("https://tanbushi-adk-teams.hf.space/apps/ai_quant/users/tanbushi/sessions/ai_quant_tanbushi_session")
+                    if (response.status_code != 200):
+                        print("Failed to get session")
+                        response=requests.post("https://tanbushi-adk-teams.hf.space/apps/ai_quant/users/tanbushi/sessions/ai_quant_tanbushi_session")
+                        if (response.status_code==200):
+                            data = {
+                                "app_name": "ai_quant",
+                                "user_id": "user",
+                                "session_id": "232d4228-d3a2-41ee-99eb-3db5b307260e",
+                                "new_message": {
+                                    "parts": [
+                                        {
+                                            "text": content["text"]
+                                        }
+                                    ],
+                                    "role": "user"
+                                },
+                                "streaming": False
+                            }
+                            response1 = requests.post("https://tanbushi-adk-teams.hf.space/run", json=data)
+                            ret_text = response1.json()[0]['content']['parts'][0]['text']
+                            print(ret_text)
+                    else:
+                        print("Successfully get session")
+                        data = {
+                            "app_name": "ai_quant",
+                            "user_id": "tanbushi",
+                            "session_id": "ai_quant_tanbushi_session",
+                            "new_message": {
+                                "parts": [
+                                    {
+                                        "text": content["text"]
+                                    }
+                                ],
+                                "role": "user"
+                            },
+                            "streaming": False
+                        }
+                        response1 = requests.post("https://tanbushi-adk-teams.hf.space/run", json=data)
+                        ret_text = response1.json()[0]['content']['parts'][0]['text']
+                        print(ret_text)
+
+
+
+
+                    content = {"text": ret_text}
                     # Correctly extract msg_type from the original_msg_type field
                     msg_type = data_dict.get('original_msg_type')
 

requirements.txt

@@ -4,3 +4,4 @@ redis
 python-dotenv
 httpx
 supabase
+requests