fix:this commit should replace he webhook_url

edenartlab · Jan 13, 2025 · f6a5bc5 · f6a5bc5
1 parent d37ce91
commit f6a5bc5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/eve/tools/replicate_tool.py b/eve/tools/replicate_tool.py
@@ -162,8 +162,8 @@ async def _create_prediction(self, args: dict, webhook=True):
         return prediction
 
 def get_webhook_url():
-    env = "api-prod" if os.getenv("ENV") == "PROD" else "api-stage"
-    dev = "-dev" if os.getenv("ENV") == "STAGE" and os.getenv("MODAL_SERVE") == "1" else ""
+    env = "api-web3-prod" if os.getenv("DB") == "WEB3-PROD" else "api-web3-stage"
+    dev = "-dev" if os.getenv("DB") == "WEB3-STAGE" and os.getenv("MODAL_SERVE") == "1" else ""
     webhook_url = f"https://edenartlab--{env}-fastapi-app{dev}.modal.run/update"
     return webhook_url