Merge pull request #47 from sususweet/fix/psycopg-connection-issue

fix: psycopg3 connection issues on windows debugging environment.

Author: beersoccer

main.py

@@ -36,6 +36,8 @@
 
 plugin = Plugin(DifyPluginEnv(MAX_REQUEST_TIMEOUT=MAX_REQUEST_TIMEOUT))
 
+if sys.platform == 'win32':
+    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
 
 def _graceful_shutdown() -> None:
     logger.info("Initiating graceful shutdown of Mem0 plugin")

pyproject.toml

@@ -9,8 +9,9 @@ dependencies = [
   "azure-identity",
   "langchain-neo4j",
   "rank-bm25",
-  "psycopg[binary,pool]",
+  "psycopg[binary,pool] ; sys_platform != 'win32'",
   "dify_plugin",
+  "psycopg2-binary ; sys_platform == 'win32'",
   "tiktoken",
   "cohere>=6.1.0",
   "ollama==0.6.1"

requirements.txt

@@ -3,7 +3,8 @@ openai
 azure-identity
 langchain-neo4j
 rank-bm25
-psycopg[binary,pool]  # psycopg3 with connection pool support
+psycopg[binary,pool] ; sys_platform != "win32"  # psycopg3 with connection pool support (non-Windows)
+psycopg2-binary      ; sys_platform == "win32"  # psycopg2 pool (Windows only)
 dify_plugin
 tiktoken  # Accurate token counting for conversation processing
 cohere>=6.1.0

utils/constants.py

@@ -53,7 +53,7 @@
 # Write operations: default timeout for all write operations (add/update/delete/delete_all)
 # NOTE: In async_mode, write tools enqueue work and return immediately, but the background
 # operation still uses this timeout as a safeguard.
-WRITE_OPERATION_TIMEOUT: int = 15
+WRITE_OPERATION_TIMEOUT: int = 45
 
 # Dify API pagination constraints
 # Maximum items (conversations or messages) that can be fetched in a single API request