added upstash

pinecone_keep_alive_service.py

@@ -31,12 +31,22 @@ keys = os.getenv("PINECONE_API_KEYS", "").split(",")
 
 # ---------- Pinecone Ping ----------
 def ping_all_pinecone_indexes():
+    results = []  # Collect all results here
+    
     for key in filter(None, (k.strip() for k in keys)):
         pc = Pinecone(api_key=key)
         try:
             indexes = pc.list_indexes()  # This returns IndexModel objects
         except Exception as e:
             logger.error(f"[{key[:5]}…] failed to list indexes: {e}")
+            # Create a response for the failed API key
+            result = PineconePingResponse(
+                idx_name=f"API_KEY_{key[:5]}…",
+                success=False,
+                error=f"Failed to list indexes: {e}",
+                time=datetime.now().isoformat(),
+            )
+            results.append(result)
             continue
 
         for index in indexes:
@@ -49,8 +59,7 @@ def ping_all_pinecone_indexes():
                     error=None,
                     time=datetime.now().isoformat(),
                 )
-                final_response = PineconeAllPingResponse(pinecone_services=[result])
-                return final_response
+                results.append(result)
             except Exception as e:
                 logger.error(f"[{key[:5]}…] ping {index.name} ✗: {e}")
                 result = PineconePingResponse(
@@ -59,6 +68,7 @@ def ping_all_pinecone_indexes():
                     error=str(e),
                     time=datetime.now().isoformat(),
                 )
-                final_response = PineconeAllPingResponse(pinecone_services=[result])
-                return final_response
-
+                results.append(result)
+    
+    # Return all collected results at the end
+    return PineconeAllPingResponse(pinecone_services=results)