增加模型选择功能,支持不同 Agent 模型
This commit is contained in:
@@ -51,6 +51,7 @@ describe("streamAgentChat", () => {
|
||||
|
||||
await streamAgentChat({
|
||||
message: "hi",
|
||||
model: "deepseek/deepseek-v4-pro",
|
||||
onEvent: (event) => events.push(event),
|
||||
});
|
||||
|
||||
@@ -60,6 +61,11 @@ describe("streamAgentChat", () => {
|
||||
method: "POST",
|
||||
projectHeaderMode: "include",
|
||||
skipAuthRedirect: true,
|
||||
body: JSON.stringify({
|
||||
message: "hi",
|
||||
session_id: undefined,
|
||||
model: "deepseek/deepseek-v4-pro",
|
||||
}),
|
||||
}),
|
||||
);
|
||||
|
||||
|
||||
@@ -1,6 +1,10 @@
|
||||
import { apiFetch } from "@/lib/apiFetch";
|
||||
import { config } from "@config/config";
|
||||
|
||||
export type AgentModel =
|
||||
| "deepseek/deepseek-v4-flash"
|
||||
| "deepseek/deepseek-v4-pro";
|
||||
|
||||
export type StreamEvent =
|
||||
| { type: "token"; sessionId: string; content: string }
|
||||
| { type: "done"; sessionId: string; totalDurationMs?: number }
|
||||
@@ -35,6 +39,7 @@ export type StreamEvent =
|
||||
type StreamOptions = {
|
||||
message: string;
|
||||
sessionId?: string;
|
||||
model?: AgentModel;
|
||||
signal?: AbortSignal;
|
||||
onEvent: (event: StreamEvent) => void;
|
||||
};
|
||||
@@ -85,6 +90,7 @@ const resolveToolParams = (
|
||||
export const streamAgentChat = async ({
|
||||
message,
|
||||
sessionId,
|
||||
model,
|
||||
signal,
|
||||
onEvent,
|
||||
}: StreamOptions) => {
|
||||
@@ -102,6 +108,7 @@ export const streamAgentChat = async ({
|
||||
body: JSON.stringify({
|
||||
message,
|
||||
session_id: sessionId,
|
||||
model,
|
||||
}),
|
||||
projectHeaderMode: "include",
|
||||
userHeaderMode: "include",
|
||||
|
||||
Reference in New Issue
Block a user