Upload app.py

app.py

@@ -4,6 +4,7 @@ from typing import List
 import logging
 import logging.handlers
 import json
+from datetime import datetime
 from langchain_openai import ChatOpenAI
 from langchain_core.tools import tool
 from langchain_core.messages import HumanMessage, AIMessage, ToolMessage
@@ -41,6 +42,36 @@ else:
 
 logger = logging.getLogger(__name__)
 
+def get_current_date() -> str:
+    """Get current date in YYYY-MM-DD format"""
+    return datetime.now().strftime("%Y-%m-%d")
+
+def create_system_prompt() -> str:
+    """Create dynamic system prompt with current date"""
+    current_date = get_current_date()
+    return f"""You are a helpful AI assistant with web search capabilities. 
+
+TODAY'S DATE: {current_date}
+
+IMPORTANT: You have access to a web_search tool. Consider your knowledge cutoff date and today's date to decide when to search:
+
+USE WEB SEARCH when users ask about:
+- Events after your knowledge cutoff date
+- Current events, breaking news, or recent developments
+- Today's date, current time, or "what's happening now"
+- Real-time data (weather, stock prices, sports scores)
+- Recent updates to ongoing situations
+- Information that changes frequently
+- When users explicitly ask you to search the web
+
+DO NOT use web search for:
+- Historical facts before your cutoff date
+- General knowledge that doesn't change (capitals, basic science, etc.)
+- Established facts and concepts
+- Personal advice or opinions
+
+When in doubt about whether information might be outdated, use web search to get the most current information."""
+
 # Configuration from environment variables
 llm_ip = os.environ.get('public_ip')
 llm_port = os.environ.get('port')
@@ -195,19 +226,14 @@ class ToolCallingAgentChat:
                 logger.info(f"Message: {message}")
                 logger.info(f"History length: {len(history)}")
             
-            # Convert history to messages for context with system message
+            # Convert history to messages for context with dynamic system message
             from langchain_core.messages import SystemMessage
             
-            messages = [SystemMessage(content="""You are a helpful AI assistant with web search capabilities. 
-
-IMPORTANT: You have access to a web_search tool. Use it when users ask about:
-- Current events, news, or recent information
-- Today's date or current time
-- Real-time data or recent updates
-- Specific facts that may have changed recently
-- When users explicitly ask you to search the web
-
-For general knowledge questions that don't require current information, answer directly without using tools.""")]
+            # Create dynamic system prompt with current date
+            system_prompt = create_system_prompt()
+            if ENABLE_DETAILED_LOGGING:
+                logger.info(f"System prompt includes today's date: {get_current_date()}")
+            messages = [SystemMessage(content=system_prompt)]
             
             for user_msg, assistant_msg in history:
                 messages.append(HumanMessage(content=user_msg))
@@ -318,14 +344,13 @@ For general knowledge questions that don't require current information, answer d
 # Global agent instance
 tool_calling_agent = ToolCallingAgentChat(llm_ip, llm_port, llm_key, llm_model)
 
-def generate_response(message: str, history: List[List[str]], system_prompt: str, 
-                     max_tokens: int, ip: str, port: str, api_key: str, model: str):
-    """Generate response using tool calling agent"""
+def generate_response(message: str, history: List[List[str]], max_tokens: int):
+    """Generate response using tool calling agent with dynamic system prompt"""
     global tool_calling_agent
     
     try:
-        # Update agent configuration if changed
-        tool_calling_agent.update_config(ip, port, api_key, model)
+        # Configuration is pre-loaded from environment variables
+        # No runtime config changes allowed for security
         
         # Reset conversation if history is empty (indicates clear button was pressed)
         if len(history) == 0:
@@ -396,21 +421,8 @@ chatbot = gr.ChatInterface(
         elem_id="chatbot"
     ),
     additional_inputs=[
-        gr.Textbox(
-            "You are a helpful AI assistant with web search capabilities. Use web search when you need current information, recent events, or real-time data.", 
-            label="System Prompt",
-            lines=2
-        ),
-        gr.Slider(50, 2048, label="Max Tokens", value=512, 
+        gr.Slider(50, 8192, label="Max Tokens", value=1024, 
                  info="Maximum number of tokens in the response"),
-        gr.Textbox(llm_ip, label="LLM IP Address", 
-                  info="IP address of the OpenAI-compatible LLM server"),
-        gr.Textbox(llm_port, label="LLM Port", 
-                  info="Port of the LLM server"),
-        gr.Textbox(llm_key, label="API Key", type="password",
-                  info="API key for the LLM server"),
-        gr.Textbox(llm_model, label="Model Name", 
-                  info="Name of the model to use"),
     ],
     title="🤖 DQ Micro Agent",
     description="DevQuasar self hosted Micro Agent with websearch capabilities",