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

move into cookies

Browse source
This commit is contained in:
Simon Knott 2025-02-13 10:15:30 +01:00
parent 47086720c6
commit 9bc2c28847
No known key found for this signature in database
GPG key ID: 8CEDC00028084AEC
3 changed files with 23 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -131,10 +131,6 @@ export async function installRootRedirect(server: HttpServer, traceUrls: string[
params.append('project', project);
for (const reporter of options.reporter || [])
params.append('reporter', reporter);
if (process.env.OPENAI_API_KEY)
params.append('openai_api_key', process.env.OPENAI_API_KEY);
if (process.env.ANTHROPIC_API_KEY)
params.append('anthropic_api_key', process.env.ANTHROPIC_API_KEY);
let baseUrl = '.';
if (process.env.PW_HMR) {
@ -146,6 +142,13 @@ export async function installRootRedirect(server: HttpServer, traceUrls: string[
server.routePath('/', (_, response) => {
response.statusCode = 302;
response.setHeader('Location', urlPath);
if (process.env.OPENAI_API_KEY)
response.setHeader('Set-Cookie', `openai_api_key=${process.env.OPENAI_API_KEY}`);
if (process.env.ANTHROPIC_API_KEY)
response.setHeader('Set-Cookie', `anthropic_api_key=${process.env.ANTHROPIC_API_KEY}`);
response.end();
return true;
});

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

@ -198,16 +198,22 @@ class Conversation {
const llmContext = React.createContext<LLMChat | undefined>(undefined);
export function LLMProvider({ openai, anthropic, children }: React.PropsWithChildren<{ openai?: string, anthropic?: string }>) {
const chat = React.useMemo(() => {
let llm: LLM | undefined;
if (openai)
llm = new OpenAI(openai);
if (anthropic)
llm = new Anthropic(anthropic);
if (llm)
return new LLMChat(llm);
}, [openai, anthropic]);
function parseCookie(cookie: string): [name: string, value: string][] {
return cookie.split(";").filter(v => v.includes("=")).map(kv => {
const separator = kv.indexOf("=");
return [kv.substring(0, separator), kv.substring(separator + 1)];
})
}
export function LLMProvider({ children }: React.PropsWithChildren<{}>) {
const chat = React.useMemo(() => {
for (const [name, value] of parseCookie(document.cookie)) {
if (name === 'openai_api_key')
return new LLMChat(new OpenAI(value));
if (name === 'anthropic_api_key')
return new LLMChat(new Anthropic(value))
}
}, []);
return <llmContext.Provider value={chat}>{children}</llmContext.Provider>;
};

2
packages/trace-viewer/src/ui/uiModeView.tsx
View file

@ -62,8 +62,6 @@ const queryParams = {
updateSnapshots: (searchParams.get('updateSnapshots') as 'all' | 'none' | 'missing' | undefined) || undefined,
reporters: searchParams.has('reporter') ? searchParams.getAll('reporter') : undefined,
pathSeparator: searchParams.get('pathSeparator') || '/',
openai_api_key: searchParams.get('openai_api_key') || undefined,
anthropic_api_key: searchParams.get('anthropic_api_key') || undefined,
};
if (queryParams.updateSnapshots && !['all', 'none', 'missing'].includes(queryParams.updateSnapshots))
queryParams.updateSnapshots = undefined;