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

enable engine for all models (#434)

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