Merge pull request #86 from camaze/issue_fastapi_reload

【CodeFuse】support fastapi hot reload

Author: lightislost

docker-compose.yaml

@@ -170,6 +170,7 @@ services:
       USER: root
       TZ:   "${TZ}"
       OLLAMA_HOST: http://ollama:11434
+      PYTHONPATH: /home/user/muagent/examples/ekg_examples
     ports:
       - 3737:3737
     volumes:
@@ -179,7 +180,7 @@ services:
     restart: on-failure
     networks:
       - ekg-net
-    command: ["python", "/home/user/muagent/examples/ekg_examples/start.py"]  # 指定要执行的脚本
+    command: sh -c "uvicorn start:app --reload --reload-dir /home/user/muagent/examples --reload-dir /home/user/muagent/muagent --port=3737 --host=0.0.0.0"
 
   ekgfrontend:
     build:

examples/ekg_examples/start.py

@@ -320,7 +320,7 @@ def embed_query(self, text: str) -> List[float]:
 
 
 from muagent.httpapis.ekg_construct import create_api
-create_api(
+app = create_api(
     llm, 
     llm_config,
     embeddings, 

muagent/httpapis/ekg_construct/api.py

@@ -456,5 +456,4 @@ def ekg_migration_reasoning(request: EKGFeaturesRequest):
 
 
 def create_api(llm, llm_config, embeddings,ekg_construct_service: EKGConstructService, memory_manager, geabase_handler, intention_router):
-    app = init_app(llm, llm_config, embeddings,ekg_construct_service, memory_manager, geabase_handler, intention_router)
-    uvicorn.run(app, host="0.0.0.0", port=3737)
+    return init_app(llm, llm_config, embeddings,ekg_construct_service, memory_manager, geabase_handler, intention_router)