nolo-cli 0.1.7 → 0.1.9

package/ai/agent/cliChatClient.ts

@@ -0,0 +1,119 @@
+import { selectCurrentServer } from "app/settings/settingSlice";
+import { selectCurrentToken } from "auth/authSlice";
+import type { RootState } from "app/store";
+import type { CliProvider } from "./cliExecutor";
+
+type CliChatAction = "start" | "turn" | "get" | "close";
+
+type CliChatRequestBase = {
+  cliProvider?: CliProvider;
+  model?: string;
+};
+
+function getCliChatRequestConfig(thunkApi: any) {
+  const state = thunkApi.getState() as RootState;
+  const currentServer = selectCurrentServer(state);
+  const token = selectCurrentToken(state);
+  if (!currentServer) throw new Error("无法获取当前服务器地址。");
+  return { currentServer, token };
+}
+
+async function postCliChat(
+  thunkApi: any,
+  body: Record<string, any>,
+  signal?: AbortSignal,
+): Promise<Response> {
+  const { currentServer, token } = getCliChatRequestConfig(thunkApi);
+  return fetch(`${currentServer}/api/cli/chat`, {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      ...(token ? { Authorization: `Bearer ${token}` } : {}),
+    },
+    body: JSON.stringify(body),
+    signal,
+  });
+}
+
+export async function startCliChatSession(
+  thunkApi: any,
+  args: CliChatRequestBase & { systemPrompt?: string; cliProvider?: CliProvider },
+) {
+  const response = await postCliChat(thunkApi, {
+    action: "start" satisfies CliChatAction,
+    cliProvider: args.cliProvider,
+    model: args.model,
+    systemPrompt: args.systemPrompt,
+  });
+  return response.json();
+}
+
+export async function getCliChatSession(
+  thunkApi: any,
+  args: { sessionId: string },
+) {
+  const response = await postCliChat(thunkApi, {
+    action: "get" satisfies CliChatAction,
+    sessionId: args.sessionId,
+  });
+  return response.json();
+}
+
+export async function closeCliChatSession(
+  thunkApi: any,
+  args: { sessionId: string },
+) {
+  const response = await postCliChat(thunkApi, {
+    action: "close" satisfies CliChatAction,
+    sessionId: args.sessionId,
+  });
+  return response.json();
+}
+
+export async function runCliChatTurnNonStreaming(
+  thunkApi: any,
+  args: CliChatRequestBase & {
+    prompt: string;
+    sessionId: string;
+  },
+  signal?: AbortSignal,
+) {
+  const response = await postCliChat(
+    thunkApi,
+    {
+      action: "turn" satisfies CliChatAction,
+      sessionId: args.sessionId,
+      prompt: args.prompt,
+      model: args.model,
+      stream: false,
+    },
+    signal,
+  );
+  return response.json();
+}
+
+export function createCliChatTurnStream(
+  thunkApi: any,
+  args: CliChatRequestBase & {
+    prompt: string;
+    sessionId?: string;
+  },
+  signal?: AbortSignal,
+) {
+  return postCliChat(
+    thunkApi,
+    args.sessionId
+      ? {
+          action: "turn" satisfies CliChatAction,
+          sessionId: args.sessionId,
+          prompt: args.prompt,
+          model: args.model,
+        }
+      : {
+          prompt: args.prompt,
+          model: args.model,
+          cliProvider: args.cliProvider,
+        },
+    signal,
+  );
+}