All files / kernel-test/src/vats lms-chat-vat.ts

0% Statements 0/6
0% Branches 0/4
0% Functions 0/2
0% Lines 0/6

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42                                                                                   
import type { ERef } from '@endo/eventual-send';
import { makeDefaultExo } from '@metamask/kernel-utils/exo';
import { makeChatClient } from '@ocap/kernel-language-model-service';
import type { ChatService } from '@ocap/kernel-language-model-service';
 
import { unwrapTestLogger } from '../test-powers.ts';
import type { TestPowers } from '../test-powers.ts';
 
/**
 * A vat that uses a language model service to generate text.
 *
 * @param vatPowers - The powers of the vat.
 * @param parameters - The parameters of the vat.
 * @param parameters.name - The name of the vat.
 * @returns A default Exo instance.
 */
// eslint-disable-next-line @typescript-eslint/explicit-function-return-type
export function buildRootObject(
  vatPowers: TestPowers,
  { name = 'anonymous' }: { name?: string } = {},
) {
  const tlog = unwrapTestLogger(vatPowers, name);
  const root = makeDefaultExo('root', {
    async bootstrap(
      _roots: unknown,
      { languageModelService }: { languageModelService: ERef<ChatService> },
    ) {
      const client = makeChatClient(languageModelService, 'test');
      const result = await client.chat.completions.create({
        messages: [
          {
            role: 'user',
            content: `Hello, my name is ${name}. What is your name?`,
          },
        ],
      });
      tlog(`response: ${result.choices[0]?.message.content ?? ''}`);
    },
  });
  return root;
}