review-pr-comments

Author: Vrajesh-Babu

contributing/samples/mcp_postgres_agent/README.md

@@ -54,7 +54,7 @@ Once the agent is running, try these queries:
 The agent uses:
 - **Model**: Gemini 2.0 Flash
 - **MCP Server**: `postgres-mcp` (via `uvx`)
-- **Access Mode**: Unrestricted (allows read/write operations)
+- **Access Mode**: Unrestricted (allows read/write operations). **Warning**: Using unrestricted mode in a production environment can pose significant security risks. It is recommended to use a more restrictive access mode or configure database user permissions appropriately for production use.
 - **Connection**: StdioConnectionParams with 60-second timeout
 - **Environment Variable**: `DATABASE_URI` (mapped from `POSTGRES_CONNECTION_STRING`)
 

contributing/samples/mcp_postgres_agent/agent.py

@@ -24,6 +24,11 @@
 load_dotenv()
 
 POSTGRES_CONNECTION_STRING = os.getenv("POSTGRES_CONNECTION_STRING")
+if not POSTGRES_CONNECTION_STRING:
+    raise ValueError(
+        "POSTGRES_CONNECTION_STRING environment variable not set. "
+        "Please create a .env file with this variable."
+    )
 
 root_agent = LlmAgent(
     model="gemini-2.0-flash",
@@ -46,7 +51,7 @@
         )
     ],
     generate_content_config=GenerateContentConfig(
-        temperature=1.0,
+        temperature=0.2,
         top_p=0.95,
     ),
 )