feat: workflow和chat分离

1，增加了创建workflow的页面 2.删除了event
2026-05-14 15:51:28 +00:00
parent c0e4fd34ae
commit 78bd6adc48
30 changed files with 1196 additions and 760 deletions
@@ -98,7 +98,7 @@ async def workflow_error_handler(request: Request, exc: WorkflowError):


 base_dir = os.path.dirname(
-    os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+    os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 )
 frontend_dir = os.path.join(base_dir, "frontend", "dist")

@@ -1,52 +0,0 @@
-#  Copyright 2026 zhaoxi826
-#
-#  Licensed under the Apache License, Version 2.0 (the "License");
-#  you may not use this file except in compliance with the License.
-#  You may obtain a copy of the License at
-#
-#       http://www.apache.org/licenses/LICENSE-2.0
-#
-#  Unless required by applicable law or agreed to in writing, software
-#  distributed under the License is distributed on an "AS IS" BASIS,
-#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-#  See the License for the specific language governing permissions and
-#  limitations under the License.
-
-import datetime
-from pydantic import BaseModel, Field, ConfigDict
-from ulid import ULID
-from typing import Any, Dict
-import asyncio
-from kilostar.core.work.workflow.workflow import KiloStarWorkflow
-
-
-class kilostarEvent(BaseModel):
-    """kilostarEvent 核心组件类。
-    这是一个领域数据模型或功能封装类，承载了 kilostarEvent 相关的内聚属性定义与状态维护。它的存在隔离了局部的业务复杂性，并对外提供了类型安全的访问接口。"""
-
-    model_config = ConfigDict(arbitrary_types_allowed=True)
-    trace_id: str = Field(
-        default_factory=lambda: str(ULID()), description="事件的唯一标识符"
-    )
-    platform: str = Field(description="消息来源的平台")
-    user_id: str = Field(description="用户id")
-    user_name: str = Field(description="用户名")
-    create_time: str = Field(
-        default_factory=lambda: str(
-            datetime.datetime.now(datetime.timezone.utc).isoformat()
-        ),
-        description="事件创建时间",
-    )
-    message: str = Field(description="用户发来的消息")
-    attachment: Dict[str, str] | None = Field(default=None, description="附件")
-    # --------------------------------------------------------------------------------------------------------------
-    context: Dict[str, Any] = Field(
-        default_factory=dict, description="事件上下文内容，可包含工作流模板等信息"
-    )
-    workflow: KiloStarWorkflow | None = Field(default=None, description="工作流")
-    pending_queue: asyncio.Queue[str] | None = Field(
-        default=None, description="待处理队列"
-    )
-    receive_queue: asyncio.Queue[str] | None = Field(
-        default=None, description="待接收队列"
-    )
@@ -12,10 +12,9 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.

-from fastapi import APIRouter, Depends, HTTPException, status, UploadFile, File
+from fastapi import APIRouter, Depends, HTTPException, UploadFile, File
 from pydantic import BaseModel
 from kilostar.utils.access import Accessor, TokenData
-from kilostar.api.platform.event import kilostarEvent
 from kilostar.utils.ray_hook import ray_actor_hook
 import os
 import anyio
@@ -43,22 +42,13 @@ async def create_message(
    Returns:        : 序列化后的标准网络响应模型（如包含业务状态码、成功标志及对应的数据载荷 Data）。"""
    logger.info("收到消息，来源：客户端")
    logger.debug(f"消息内容：{message.message}")
-    event = kilostarEvent(
-        platform="client",
-        user_id=str(token_data.user_id),
+    regulatory_node = ray_actor_hook("regulatory_node").regulatory_node
+    reply = await regulatory_node.handle_client_message.remote(
+        user_id=token_data.user_id,
        user_name=token_data.username,
        message=message.message,
    )
-    regulatory_node = ray_actor_hook("regulatory_node").regulatory_node
-    message = await regulatory_node.working.remote(event)
-    if message.startswith("任务已创建"):
-        return {"message": f"{event.trace_id}\n\n{message}"}
-    elif message == "未知相应类型":
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail="模型回复错误"
-        )
-    else:
-        return {"message": message}
+    return {"message": reply}


@client_router.post("/upload")
@@ -42,9 +42,12 @@ async def create_workflow(
        command=request.command,
    )

-    # 将需求发送给意识节点去处理构建
+    global_workflow_manager = ray_actor_hook(
+        "global_workflow_manager"
+    ).global_workflow_manager
+    await global_workflow_manager.create_trace.remote(trace_id)
+
    consciousness_node = ray_actor_hook("consciousness_node").consciousness_node
-    # 可以异步通知意识节点开始与用户在特定 Trace ID 下对话
    consciousness_node.start_workflow_design.remote(trace_id, request.command)

    return {"trace_id": trace_id, "status": "creating"}
@@ -59,6 +62,39 @@ async def get_workflow_list(token_data: TokenData = Depends(Accessor.get_current
    return {"workflows": workflows}


+@workflow_router.get("/sse/{trace_id}")
+async def get_workflow_sse(trace_id: str, request: Request):
+    global_workflow_manager = ray_actor_hook(
+        "global_workflow_manager"
+    ).global_workflow_manager
+
+    async def event_generator():
+        try:
+            while True:
+                if await request.is_disconnected():
+                    break
+                message = await global_workflow_manager.get_pending.remote(trace_id)
+                if message:
+                    yield f"data: {message}\n\n"
+                else:
+                    await asyncio.sleep(0.5)
+        except asyncio.CancelledError:
+            pass
+
+    return StreamingResponse(event_generator(), media_type="text/event-stream")
+
+
+@workflow_router.post("/reply/{trace_id}")
+async def post_workflow_reply(trace_id: str, request: Request):
+    data = await request.json()
+    reply_msg = data.get("message", "")
+    global_workflow_manager = ray_actor_hook(
+        "global_workflow_manager"
+    ).global_workflow_manager
+    await global_workflow_manager.put_received.remote(trace_id, reply_msg)
+    return {"status": "ok"}
+
+
@workflow_router.get("/{trace_id}")
 async def get_workflow_detail(
    trace_id: str, token_data: TokenData = Depends(Accessor.get_current_user)
@@ -80,42 +116,3 @@ async def get_workflow_detail(
        "steps": steps,
        "context_blackboard": context.blackboard if context else {},
    }
-
-
-@workflow_router.get("/sse/{trace_id}")
-async def get_workflow_sse(trace_id: str, request: Request):
-    """
-    用于与意识节点交互，获取工作流状态或设计阶段的问答消息
-    """
-    global_workflow_manager = ray_actor_hook(
-        "global_workflow_manager"
-    ).global_workflow_manager
-
-    async def event_generator():
-        try:
-            while True:
-                if await request.is_disconnected():
-                    break
-                message = await global_workflow_manager.get_pending.remote(trace_id)
-                if message:
-                    yield f"data: {message}\n\n"
-                else:
-                    await asyncio.sleep(0.5)
-        except asyncio.CancelledError:
-            pass
-
-    return StreamingResponse(event_generator(), media_type="text/event-stream")
-
-
-@workflow_router.post("/reply/{trace_id}")
-async def post_workflow_reply(trace_id: str, request: Request):
-    """
-    用于用户回复意识节点的提问（设计阶段或运行中的中断确认）
-    """
-    data = await request.json()
-    reply_msg = data.get("message", "")
-    global_workflow_manager = ray_actor_hook(
-        "global_workflow_manager"
-    ).global_workflow_manager
-    await global_workflow_manager.put_received.remote(trace_id, reply_msg)
-    return {"status": "ok"}