Update app.py

app.py

@@ -1285,6 +1285,13 @@ def parse_query(query: str) -> dict:
 
 def autogen_multi_document_analysis(query: str, docs: list, file_names: list) -> str:
     try:
+        import os
+        import tempfile
+        
+        # 創建臨時目錄作為緩存目錄
+        cache_dir = tempfile.mkdtemp()
+        os.environ['AUTOGEN_CACHE_DIR'] = cache_dir
+        
         # 準備文件上下文
         context = "\n\n".join(
             f"Document {name}:\n{doc[:2000]}..." 
@@ -1302,14 +1309,16 @@ def autogen_multi_document_analysis(query: str, docs: list, file_names: list) ->
             "config_list": config_list,
             "temperature": 0,
             "request_timeout": 120,
-            "seed": 42
+            "seed": 42,
+            "cache_dir": cache_dir  # 使用臨時目錄作為緩存
         }
 
         # 代碼執行配置
         code_execution_config = {
-            "use_docker": False,  # 禁用 Docker
+            "use_docker": False,
             "last_n_messages": 3,
-            "timeout": 60
+            "timeout": 60,
+            "work_dir": cache_dir  # 使用臨時目錄作為工作目錄
         }
 
         # 定義代理人
@@ -1317,7 +1326,7 @@ def autogen_multi_document_analysis(query: str, docs: list, file_names: list) ->
             name="User",
             system_message="A user seeking information from multiple documents.",
             human_input_mode="NEVER",
-            code_execution_config=code_execution_config  # 使用代碼執行配置
+            code_execution_config=code_execution_config
         )
 
         # 定義專家代理人們
@@ -1385,7 +1394,16 @@ def autogen_multi_document_analysis(query: str, docs: list, file_names: list) ->
 
         # 執行群組討論
         user_proxy.initiate_chat(manager, message=task_prompt)
-        return user_proxy.last_message()["content"]
+        result = user_proxy.last_message()["content"]
+        
+        # 清理臨時目錄
+        import shutil
+        try:
+            shutil.rmtree(cache_dir)
+        except:
+            pass
+            
+        return result
 
     except Exception as e:
         print(f"ERROR in AutoGen processing: {str(e)}")