Migrate from Vercel Redis to Redis standalone client

ai_ta_backend/service/project_service.py

@@ -1,6 +1,7 @@
 import json
 import os
 
+import redis
 import requests
 from injector import inject
 
@@ -23,6 +24,8 @@ def __init__(self, sql_db: SQLDatabase, posthog_service: PosthogService, sentry_
     self.posthog = posthog_service
     self.sentry = sentry_service
 
+    self.redis_client = redis.Redis.from_url(os.environ['REDIS_URL'], db=0)
+
   def generate_json_schema(self, project_name: str, project_description: str | None) -> None:
     # Generate metadata schema using project_name and project_description
     json_schema = generate_schema_from_project_description(project_name, project_description)
@@ -93,8 +96,7 @@ def create_project(self, project_name: str, project_description: str | None, pro
             llm_val[row['providerName']] = row['providerBodyNoModels']
 
           # Insert the pre-assigned API keys into Redis
-          set_llm_url = str(os.environ['KV_REST_API_URL']) + f"/set/{redis_key}"
-          set_response = requests.post(set_llm_url, headers=headers, data=json.dumps(llm_val))
+          self.redis_client.set(redis_key, json.dumps(llm_val))
 
           # Check the response status
           if set_response.status_code == 200:

requirements.txt

@@ -38,6 +38,7 @@ boto3==1.28.79
 qdrant-client==1.7.3
 supabase==2.5.3
 minio==7.2.12
+"redis[hiredis]"
 
 # Logging 
 posthog==3.1.0