Upload app.py

app.py

@@ -59,14 +59,15 @@ scheduler = CommitScheduler(
 
 ########### vLLM Service Management ###########
 VLLM_MODEL_NAME = "AgentFlow/agentflow-planner-7b"
-VLLM_PORT = 8000
-VLLM_HOST = "localhost"
+VLLM_PORT = "Lambda"
+VLLM_HOST = os.getenv("VLLM_HOST", "localhost")
 VLLM_PROCESS = None
 
 def check_vllm_service() -> bool:
     """Check if vLLM service is running"""
     try:
-        response = requests.get(f"http://{VLLM_HOST}:{VLLM_PORT}/v1/models", timeout=2)
+        # response = requests.get(f"http://{VLLM_HOST}:{VLLM_PORT}/v1/models", timeout=2)
+        response = requests.get(f"http://{VLLM_HOST}/v1/models", timeout=2)
         return response.status_code == 200
     except:
         return False