Update app.py

app.py

@@ -22,6 +22,39 @@ from io import BytesIO
 from pathlib import Path
 import os
 from utils.block_relation_builder import block_builder, variable_adder_main
+from langchain.chat_models import ChatOpenAI
+from langchain_openai import ChatOpenAI
+from pydantic import Field, SecretStr
+
+os.environ["OPENROUTER_API_KEY"] = os.getenv("OPENROUTER_API_KEY", "default_key_or_placeholder")
+class ChatOpenRouter(ChatOpenAI):
+    openai_api_key: Optional[SecretStr] = Field(
+        alias="api_key",
+        default_factory=secret_from_env("OPENROUTER_API_KEY", default=None),
+    )
+    @property
+    def lc_secrets(self) -> dict[str, str]:
+        return {"openai_api_key": "OPENROUTER_API_KEY"}
+ 
+    def __init__(self,
+                 openai_api_key: Optional[str] = None,
+                 **kwargs):
+        openai_api_key = (
+            openai_api_key or os.environ.get("OPENROUTER_API_KEY")
+        )
+        super().__init__(
+            base_url="https://openrouter.ai/api/v1",
+            openai_api_key=openai_api_key,
+            **kwargs
+        )
+ 
+llm = ChatOpenRouter(
+    #model_name="deepseek/deepseek-r1-0528:free",
+    #model_name="google/gemini-2.0-flash-exp:free",
+    #model_name="deepseek/deepseek-v3-base:free",
+    model_name="deepseek/deepseek-r1:free"
+)
+
 
 def log_execution_time(func):
     @functools.wraps(func)
@@ -55,11 +88,11 @@ load_dotenv()
 # os.environ["GROQ_API_KEY"] = os.getenv("GROQ_API_KEY")
 groq_api_key = os.getenv("GROQ_API_KEY")
 
-llm = ChatGroq(
-    model="meta-llama/llama-4-scout-17b-16e-instruct",
-    temperature=0,
-    max_tokens=None,
-)
+# llm = ChatGroq(
+#     model="meta-llama/llama-4-scout-17b-16e-instruct",
+#     temperature=0,
+#     max_tokens=None,
+# )
 
 app = Flask(__name__)
 
@@ -2488,7 +2521,7 @@ def convert_pdf_stream_to_images(pdf_stream: io.BytesIO, dpi=300):
 
 def delay_for_tpm_node(state: GameState):
     logger.info("--- Running DelayForTPMNode ---")
-    time.sleep(60)  # Adjust the delay as needed
+    time.sleep(1)  # Adjust the delay as needed
     logger.info("Delay completed.")
     return state