Filip/playwright
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

rewrite first

Browse source
This commit is contained in:
Simon Knott 2025-02-13 14:24:16 +01:00
parent 30d32916e0
commit 3a8ba54e9b
No known key found for this signature in database
GPG key ID: 8CEDC00028084AEC
4 changed files with 34 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
packages/playwright-core/src/server/trace/viewer/traceViewer.ts
View file

@ -144,10 +144,13 @@ export async function installRootRedirect(server: HttpServer, traceUrls: string[
response.setHeader('Location', urlPath); response.setHeader('Location', urlPath);
if (process.env.OPENAI_API_KEY) if (process.env.OPENAI_API_KEY)
response.setHeader('Set-Cookie', `openai_api_key=${process.env.OPENAI_API_KEY}`); response.appendHeader('Set-Cookie', `openai_api_key=${process.env.OPENAI_API_KEY}`);
if (process.env.OPENAI_BASE_URL)
response.appendHeader('Set-Cookie', `openai_base_url=${process.env.OPENAI_BASE_URL}`);
if (process.env.ANTHROPIC_API_KEY) if (process.env.ANTHROPIC_API_KEY)
response.setHeader('Set-Cookie', `anthropic_api_key=${process.env.ANTHROPIC_API_KEY}`); response.appendHeader('Set-Cookie', `anthropic_api_key=${process.env.ANTHROPIC_API_KEY}`);
if (process.env.ANTHROPIC_BASE_URL)
response.appendHeader('Set-Cookie', `anthropic_base_url=${process.env.ANTHROPIC_BASE_URL}`);
response.end(); response.end();
return true; return true;

16
packages/trace-viewer/src/ui/llm.tsx
View file

@ -221,15 +221,15 @@ export class Conversation {
const llmContext = React.createContext<LLMChat | undefined>(undefined); const llmContext = React.createContext<LLMChat | undefined>(undefined);
export function LLMProvider({ children }: React.PropsWithChildren<{}>) { export function LLMProvider({ children }: React.PropsWithChildren<{}>) {
const cookies = useCookies(); const cookiePairs = useCookies();
const chat = React.useMemo(() => { const chat = React.useMemo(() => {
for (const [name, value] of cookies) { const cookies = Object.fromEntries(cookiePairs);
if (name === 'openai_api_key') console.log({ cookies })
return new LLMChat(new OpenAI(value)); if (cookies.openai_api_key)
if (name === 'anthropic_api_key') return new LLMChat(new OpenAI(cookies.openai_api_key, cookies.openai_base_url));
return new LLMChat(new Anthropic(value)) if (cookies.anthropic_api_key)
} return new LLMChat(new Anthropic(cookies.anthropic_api_key, cookies.anthropic_base_url));
}, [cookies]); }, [cookiePairs]);
return <llmContext.Provider value={chat}>{children}</llmContext.Provider>; return <llmContext.Provider value={chat}>{children}</llmContext.Provider>;
}; };

2
packages/web/src/uiUtils.ts
View file

@ -250,7 +250,7 @@ export function useFlash(): [boolean, EffectCallback] {
} }
export function useCookies() { export function useCookies() {
return document.cookie.split(";").filter(v => v.includes("=")).map(kv => { return document.cookie.split("; ").filter(v => v.includes("=")).map(kv => {
const separator = kv.indexOf("="); const separator = kv.indexOf("=");
return [kv.substring(0, separator), kv.substring(separator + 1)]; return [kv.substring(0, separator), kv.substring(separator + 1)];
}) })

33
tests/playwright-test/ui-mode-llm.spec.ts
View file

@ -21,7 +21,23 @@ test.describe.configure({ mode: 'parallel', retries });
test.beforeAll(() => process.env.PW_EXPERIMENTAL_SERVICE_WORKER_NETWORK_EVENTS = '1'); test.beforeAll(() => process.env.PW_EXPERIMENTAL_SERVICE_WORKER_NETWORK_EVENTS = '1');
test.afterAll(() => delete process.env.PW_EXPERIMENTAL_SERVICE_WORKER_NETWORK_EVENTS); test.afterAll(() => delete process.env.PW_EXPERIMENTAL_SERVICE_WORKER_NETWORK_EVENTS);
test('openai', async ({ runUITest }) => { test('openai', async ({ runUITest, server }) => {
server.setRoute('/v1/chat/completions', async (req, res) => {
res.setHeader('Access-Control-Allow-Origin', '*');
res.setHeader('Access-Control-Allow-Headers', '*');
if (req.method === 'OPTIONS')
return res.end();
expect(req.headers.authorization).toBe('Bearer fake-key');
expect((await req.postBody).toString()).toContain(`- button \\"Submit\\"`);
const event = {
object: 'chat.completion.chunk',
choices: [{ delta: { content: 'This is a mock response' } }]
};
res.setHeader('Content-Type', 'text/event-stream');
res.write(`data: ${JSON.stringify(event)}\n\n`);
});
const { page } = await runUITest({ const { page } = await runUITest({
'a.test.ts': ` 'a.test.ts': `
import { test, expect } from '@playwright/test'; import { test, expect } from '@playwright/test';
@ -31,19 +47,8 @@ test('openai', async ({ runUITest }) => {
}); });
`, `,
}, { }, {
OPENAI_API_KEY: 'fake-key' OPENAI_API_KEY: 'fake-key',
}); OPENAI_BASE_URL: server.PREFIX,
await page.context().route('https://api.openai.com/**', async (route, request) => {
expect(await request.headerValue('authorization')).toBe('Bearer fake-key');
expect(request.postData()).toContain(`- button \\"Submit\\"`);
const event = {
object: 'chat.completion.chunk',
choices: [{ delta: { content: 'This is a mock response' } }]
};
await route.fulfill({
body: `\n\ndata: ${JSON.stringify(event)}\n\n`
});
}); });
await page.getByTitle('Run all').click(); await page.getByTitle('Run all').click();