Skip to content

Commit

Permalink
chore (provider/perplexity): fix processor legacy name in test (#4537)
Browse files Browse the repository at this point in the history
  • Loading branch information
shaper authored Jan 27, 2025
1 parent 706058c commit 9c3536e
Showing 1 changed file with 18 additions and 18 deletions.
36 changes: 18 additions & 18 deletions packages/perplexity/src/perplexity-metadata-extractor.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -87,26 +87,26 @@ describe('buildMetadataFromResponse', () => {
});
});

describe('streaming metadata processor', () => {
describe('streaming metadata extractor', () => {
it('should process streaming chunks and build final metadata', () => {
const processor = perplexityMetadataExtractor.createStreamExtractor();
const extractor = perplexityMetadataExtractor.createStreamExtractor();

// Process chunk with citations
processor.processChunk({
extractor.processChunk({
choices: [{ delta: { role: 'assistant', content: 'content' } }],
citations: ['source1', 'source2'],
});

// Process chunk with usage
processor.processChunk({
extractor.processChunk({
choices: [{ delta: { role: 'assistant', content: 'content' } }],
usage: {
citation_tokens: 100,
num_search_queries: 5,
},
});

const finalMetadata = processor.buildMetadata();
const finalMetadata = extractor.buildMetadata();

expect(finalMetadata).toEqual({
perplexity: {
Expand All @@ -120,10 +120,10 @@ describe('streaming metadata processor', () => {
});

it('should update metadata with latest chunk data', () => {
const processor = perplexityMetadataExtractor.createStreamExtractor();
const extractor = perplexityMetadataExtractor.createStreamExtractor();

// Process initial chunk
processor.processChunk({
extractor.processChunk({
citations: ['source1'],
usage: {
citation_tokens: 50,
Expand All @@ -132,15 +132,15 @@ describe('streaming metadata processor', () => {
});

// Process chunk with updated data
processor.processChunk({
extractor.processChunk({
citations: ['source1', 'source2'],
usage: {
citation_tokens: 100,
num_search_queries: 5,
},
});

const finalMetadata = processor.buildMetadata();
const finalMetadata = extractor.buildMetadata();

expect(finalMetadata).toEqual({
perplexity: {
Expand All @@ -154,38 +154,38 @@ describe('streaming metadata processor', () => {
});

it('should handle streaming chunks without metadata', () => {
const processor = perplexityMetadataExtractor.createStreamExtractor();
const extractor = perplexityMetadataExtractor.createStreamExtractor();

processor.processChunk({
extractor.processChunk({
choices: [{ delta: { role: 'assistant', content: 'content' } }],
});

const finalMetadata = processor.buildMetadata();
const finalMetadata = extractor.buildMetadata();

expect(finalMetadata).toBeUndefined();
});

it('should handle invalid streaming chunks', () => {
const processor = perplexityMetadataExtractor.createStreamExtractor();
const extractor = perplexityMetadataExtractor.createStreamExtractor();

processor.processChunk('invalid chunk');
extractor.processChunk('invalid chunk');

const finalMetadata = processor.buildMetadata();
const finalMetadata = extractor.buildMetadata();

expect(finalMetadata).toBeUndefined();
});

it('should handle null values in usage data', () => {
const processor = perplexityMetadataExtractor.createStreamExtractor();
const extractor = perplexityMetadataExtractor.createStreamExtractor();

processor.processChunk({
extractor.processChunk({
usage: {
citation_tokens: null,
num_search_queries: null,
},
});

const finalMetadata = processor.buildMetadata();
const finalMetadata = extractor.buildMetadata();

expect(finalMetadata).toEqual({
perplexity: {
Expand Down

0 comments on commit 9c3536e

Please sign in to comment.