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

add divider

Browse source
This commit is contained in:
Simon Knott 2025-02-13 14:41:50 +01:00
parent 7967e6ab96
commit 554ef49a3e
No known key found for this signature in database
GPG key ID: 8CEDC00028084AEC
3 changed files with 20 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
packages/trace-viewer/src/ui/aiConversation.css
View file

@ -22,6 +22,18 @@
color: #e0e0e0;
}
.chat-disclaimer {
text-align: center;
color: var(--vscode-editorBracketMatch-border);
margin: 0px;
}
.chat-container hr {
width: 100%;
border: none;
border-top: 1px solid var(--vscode-titleBar-inactiveBackground);
}
.messages-container {
flex: 1;
overflow-y: auto;

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

@ -31,6 +31,8 @@ export function AIConversation({ history, conversation }: { history: LLMMessage[
return (
<div className='chat-container'>
<p className='chat-disclaimer'>Chat based on {conversation.chat.api.name}. Check for mistakes.</p>
<hr/>
<div className='messages-container'>
{history.filter(({ role }) => role !== 'developer').map((message, index) => (
<div

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

@ -24,7 +24,8 @@ export type LLMMessage = {
displayContent?: string;
};
interface LLM {
interface LLM {
readonly name: string;
chatCompletion(messages: LLMMessage[], signal: AbortSignal): AsyncGenerator<string>;
}
@ -96,6 +97,8 @@ async function *parseSSE(body: NonNullable<Response['body']>): AsyncGenerator<{
class OpenAI implements LLM {
name = 'OpenAI';
constructor(private apiKey: string, private baseURL = 'https://api.openai.com') {}
async *chatCompletion(messages: LLMMessage[], signal: AbortSignal) {
@ -130,6 +133,7 @@ class OpenAI implements LLM {
}
class Anthropic implements LLM {
name = 'Anthropic';
constructor(private apiKey: string, private baseURL = 'https://api.anthropic.com') {}
async *chatCompletion(messages: LLMMessage[], signal: AbortSignal): AsyncGenerator<string> {
const response = await fetch(new URL('./v1/messages', this.baseURL), {
@ -180,7 +184,7 @@ export class Conversation {
onChange = new EventEmitter<void>();
private _abortControllers = new Set<AbortController>();
constructor(private chat: LLMChat, systemPrompt: string) {
constructor(public chat: LLMChat, systemPrompt: string) {
this.history = [{ role: 'developer', content: systemPrompt }];
}