Unverified 提交 897d2e52 authored 作者: Will Chen's avatar Will Chen 提交者: GitHub

enable engine for all models (#434)

上级 e326f147
......@@ -34,10 +34,9 @@ testSkipIfWindows(
},
);
// auto (defaults to Gemini 2.5 Flash)
testSkipIfWindows("auto should send message to gateway", async ({ po }) => {
testSkipIfWindows("auto should send message to engine", async ({ po }) => {
await po.setUpDyadPro();
await po.sendPrompt("[dump] tc=gateway-simple");
await po.sendPrompt("[dump] tc=turbo-edits");
await po.snapshotServerDump("request");
await po.snapshotMessages({ replaceDumpPath: true });
......
import { testSkipIfWindows } from "./helpers/test_helper";
testSkipIfWindows("gemini 2.5 flash", async ({ po }) => {
// Note: we do not need to disable pro modes because 2.5 flash doesn't
// use engine.
await po.setUpDyadPro();
await po.selectModel({ provider: "Google", model: "Gemini 2.5 Flash" });
await po.sendPrompt("[dump] tc=gateway-simple");
await po.snapshotServerDump("request");
await po.snapshotMessages({ replaceDumpPath: true });
});
testSkipIfWindows("claude 4 sonnet", async ({ po }) => {
await po.setUpDyadPro();
// Disable the pro modes so it routes to gateway.
......@@ -26,4 +15,5 @@ testSkipIfWindows("claude 4 sonnet", async ({ po }) => {
await po.sendPrompt("[dump] tc=gateway-simple");
await po.snapshotServerDump("request");
await po.snapshotMessages({ replaceDumpPath: true });
});
- paragraph: "[dump] tc=gateway-simple"
- paragraph: "[dump] tc=turbo-edits"
- paragraph: "[[dyad-dump-path=*]]"
- button "Retry":
- img
\ No newline at end of file
......@@ -168,7 +168,6 @@ export type LanguageModel =
tag?: string;
maxOutputTokens?: number;
contextWindow?: number;
supportsTurboEdits?: boolean;
type: "local" | "cloud";
};
......
......@@ -17,7 +17,6 @@ export interface ModelOption {
tag?: string;
maxOutputTokens?: number;
contextWindow?: number;
supportsTurboEdits?: boolean;
}
export const MODEL_OPTIONS: Record<string, ModelOption[]> = {
......@@ -29,7 +28,6 @@ export const MODEL_OPTIONS: Record<string, ModelOption[]> = {
description: "OpenAI's flagship model",
maxOutputTokens: 32_768,
contextWindow: 1_047_576,
supportsTurboEdits: true,
},
// https://platform.openai.com/docs/models/gpt-4.1-mini
{
......@@ -69,7 +67,6 @@ export const MODEL_OPTIONS: Record<string, ModelOption[]> = {
// See comment below for Claude 3.7 Sonnet for why we set this to 16k
maxOutputTokens: 16_000,
contextWindow: 200_000,
supportsTurboEdits: true,
},
{
name: "claude-3-7-sonnet-latest",
......@@ -81,7 +78,6 @@ export const MODEL_OPTIONS: Record<string, ModelOption[]> = {
// https://docs.anthropic.com/en/docs/build-with-claude/extended-thinking#max-tokens-and-context-window-size-with-extended-thinking
maxOutputTokens: 16_000,
contextWindow: 200_000,
supportsTurboEdits: true,
},
{
name: "claude-3-5-sonnet-20241022",
......@@ -89,7 +85,6 @@ export const MODEL_OPTIONS: Record<string, ModelOption[]> = {
description: "Good coder, excellent at following instructions",
maxOutputTokens: 8_000,
contextWindow: 200_000,
supportsTurboEdits: true,
},
{
name: "claude-3-5-haiku-20241022",
......@@ -109,7 +104,6 @@ export const MODEL_OPTIONS: Record<string, ModelOption[]> = {
maxOutputTokens: 65_536 - 1,
// Gemini context window = input token + output token
contextWindow: 1_048_576,
supportsTurboEdits: true,
},
// https://ai.google.dev/gemini-api/docs/models#gemini-2.5-flash-preview
{
......
......@@ -108,9 +108,7 @@ export async function getModelClient(
settings.enableProLazyEditsMode;
// Currently engine is only used for turbo edits.
const isEngineEnabled = Boolean(
engineProMode &&
languageModel?.type === "cloud" &&
languageModel?.supportsTurboEdits,
engineProMode && languageModel?.type === "cloud",
);
const provider = isEngineEnabled
? createDyadEngine({
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论