抽象统一定位方法，支持多种地理要素

2026-04-03 13:45:37 +08:00
parent d610a09c14
commit c484aad1d3
6 changed files with 252 additions and 48 deletions
@@ -789,15 +789,50 @@ export const GlobalChatbox: React.FC<Props> = ({ open, onClose }) => {
        }

        // Other frontend tools → dispatch to chatToolStore immediately
+        const buildLocateFeaturesAction = (
+          layer: string,
+          geometryKind: "point" | "line",
+        ): ChatToolAction => ({
+          type: "locate_features" as const,
+          ids: (params.ids as string[]) ?? [],
+          layer,
+          geometryKind,
+        });
+        const buildLocateByFeatureType = (): ChatToolAction | null => {
+          const rawType = params.feature_type;
+          const featureType =
+            typeof rawType === "string" ? rawType.trim().toLowerCase() : "";
+          const featureTypeMap: Record<
+            string,
+            { layer: string; geometryKind: "point" | "line" }
+          > = {
+            junction: { layer: "geo_junctions_mat", geometryKind: "point" },
+            junctions: { layer: "geo_junctions_mat", geometryKind: "point" },
+            pipe: { layer: "geo_pipes_mat", geometryKind: "line" },
+            pipes: { layer: "geo_pipes_mat", geometryKind: "line" },
+            valve: { layer: "geo_valves", geometryKind: "point" },
+            valves: { layer: "geo_valves", geometryKind: "point" },
+            reservoir: { layer: "geo_reservoirs", geometryKind: "point" },
+            reservoirs: { layer: "geo_reservoirs", geometryKind: "point" },
+            pump: { layer: "geo_pumps", geometryKind: "point" },
+            pumps: { layer: "geo_pumps", geometryKind: "point" },
+            tank: { layer: "geo_tanks", geometryKind: "point" },
+            tanks: { layer: "geo_tanks", geometryKind: "point" },
+          };
+          const config = featureTypeMap[featureType];
+          if (!config) return null;
+          return buildLocateFeaturesAction(config.layer, config.geometryKind);
+        };
        const actionMap: Record<string, () => ChatToolAction | null> = {
-          locate_nodes: () => ({
-            type: "locate_nodes" as const,
-            ids: (params.ids as string[]) ?? [],
-          }),
-          locate_pipes: () => ({
-            type: "locate_pipes" as const,
-            ids: (params.ids as string[]) ?? [],
-          }),
+          locate_features: buildLocateByFeatureType,
+          locate_pipes: () => buildLocateFeaturesAction("geo_pipes_mat", "line"),
+          locate_junctions: () =>
+            buildLocateFeaturesAction("geo_junctions_mat", "point"),
+          locate_valves: () => buildLocateFeaturesAction("geo_valves", "point"),
+          locate_reservoirs: () =>
+            buildLocateFeaturesAction("geo_reservoirs", "point"),
+          locate_pumps: () => buildLocateFeaturesAction("geo_pumps", "point"),
+          locate_tanks: () => buildLocateFeaturesAction("geo_tanks", "point"),
          view_history: () => ({
            type: "view_history" as const,
            featureInfos: (params.feature_infos as [string, string][]) ?? [],
@@ -837,6 +872,17 @@ export const GlobalChatbox: React.FC<Props> = ({ open, onClose }) => {
              );
            } else if (event.type === "done") {
              if (!conversationId && event.conversationId) setConversationId(event.conversationId);
+              setMessages((prev) =>
+                prev.map((m) =>
+                  m.id === assistantId && m.content.trim().length === 0
+                    ? {
+                        ...m,
+                        content: "⚠️ **错误：** Copilot 未返回内容，请稍后重试。",
+                        isError: true,
+                      }
+                    : m
+                )
+              );
              setIsStreaming(false);
            } else if (event.type === "error") {
              setMessages((prev) =>