添加 Copilot 聊天流式响应接口及测试

app/api/v1/endpoints/copilot.py

@@ -8,9 +8,8 @@ from fastapi import APIRouter, Depends, Request, status
 from fastapi.responses import StreamingResponse
 from pydantic import BaseModel, Field
 
-from app.auth.dependencies import get_current_active_user
+from app.auth.keycloak_dependencies import get_current_keycloak_username
 from app.core.config import settings
-from app.domain.schemas.user import UserInDB
 
 router = APIRouter()
 
@@ -32,7 +31,7 @@ def _sse_event(event: str, data: dict) -> str:
 async def copilot_chat_stream(
     payload: CopilotChatStreamRequest,
     request: Request,
-    current_user: UserInDB = Depends(get_current_active_user),
+    username: str = Depends(get_current_keycloak_username),
 ):
     timeout = httpx.Timeout(
         connect=10.0,
@@ -55,7 +54,7 @@ async def copilot_chat_stream(
         body = {
             "message": payload.message,
             "conversationId": payload.conversation_id,
-            "userId": current_user.username,
+            "userId": username,
         }
 
         try:

app/api/v1/endpoints/simulation.py

@@ -312,23 +312,23 @@ async def valve_isolation_endpoint(
     - affected_nodes: 受影响的节点列表
     - isolatable: 是否可以有效隔离
     """
-    result = {
-        "accident_element": "P461309",
-        "accident_elements": ["P461309"],
-        "affected_nodes": [
-            "J316629_A",
-            "J317037_B",
-            "J317060_B",
-            "J408189_B",
-            "J499996",
-            "J524940",
-            "J535933",
-            "J58841",
-        ],
-        "isolatable": True,
-        "must_close_valves": ["210521658", "V12974", "V12986", "V12993"],
-        "optional_valves": [],
-    }
+    # result = {
+    #     "accident_element": "P461309",
+    #     "accident_elements": ["P461309"],
+    #     "affected_nodes": [
+    #         "J316629_A",
+    #         "J317037_B",
+    #         "J317060_B",
+    #         "J408189_B",
+    #         "J499996",
+    #         "J524940",
+    #         "J535933",
+    #         "J58841",
+    #     ],
+    #     "isolatable": True,
+    #     "must_close_valves": ["210521658", "V12974", "V12986", "V12993"],
+    #     "optional_valves": [],
+    # }
     result = analyze_valve_isolation(network, accident_element, disabled_valves)
     return result