minor cleanup

eclipse-theia · Feb 14, 2025 · be72949 · be72949
1 parent 8565a63
commit be72949
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/ai-openai/src/node/openai-streaming-iterator.spec.ts b/packages/ai-openai/src/node/openai-streaming-iterator.spec.ts
@@ -18,7 +18,7 @@ import { expect } from 'chai';
 import * as sinon from 'sinon';
 import { StreamingAsyncIterator } from './openai-streaming-iterator';
 import { ChatCompletionStream } from 'openai/lib/ChatCompletionStream';
-import { CancellationTokenSource, CancellationError, Disposable } from '@theia/core';
+import { CancellationTokenSource, CancellationError } from '@theia/core';
 import { LanguageModelStreamResponsePart } from '@theia/ai-core';
 import { EventEmitter } from 'events';
 
@@ -36,8 +36,8 @@ describe('StreamingAsyncIterator', () => {
     });
 
     afterEach(() => {
-        if (iterator && (iterator as Disposable).dispose) {
-            (iterator as Disposable).dispose();
+        if (iterator) {
+            iterator.dispose();
         }
         cts.dispose();
         console.error = consoleError;