update1

mindsearch/app.py

@@ -1,3 +1,23 @@
+import os
+
+import subprocess
+
+
+
+try:
+
+    from class_registry import AutoRegister, ClassRegistry
+
+except ImportError:
+
+    # 如果导入失败，使用 pip 安装或升级 class_registry
+
+    subprocess.check_call([os.sys.executable, "-m", "pip", "install", "--upgrade", "class_registry"])
+
+    # 再次尝试导入
+
+    from class_registry import AutoRegister, ClassRegistry
+
 import asyncio
 import json
 import logging
@@ -8,18 +28,12 @@ from typing import Dict, List, Union
 import janus
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-import subprocess
 from lagent.schema import AgentStatusCode
 from pydantic import BaseModel
 from sse_starlette.sse import EventSourceResponse
 
 from mindsearch.agent import init_agent
 
-def install_library(library_name):
-    subprocess.run(["pip", "install", library_name])
-
-install_library("class_registry")
-
 def parse_arguments():
     import argparse
     parser = argparse.ArgumentParser(description='MindSearch API')