Skip to content

Commit 90eb449

Browse files
amirai21asafgardin
authored andcommitted
feat: disable examples for non node env
1 parent 29997e0 commit 90eb449

File tree

3 files changed

+18
-6
lines changed

3 files changed

+18
-6
lines changed

examples/studio/conversational-rag/rag-engine.ts

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
1-
import { AI21 } from 'ai21';
2-
import { FileResponse, UploadFileResponse } from '../../../src/types';
1+
import { AI21, FileResponse, UploadFileResponse, isNode } from 'ai21';
32
import path from 'path';
43

54
function sleep(ms) {
@@ -73,6 +72,12 @@ const fileContent = Buffer.from(
7372
'Opossums are members of the marsupial order Didelphimorphia endemic to the Americas.',
7473
);
7574
const dummyFile = new File([fileContent], 'example.txt', { type: 'text/plain' });
76-
uploadGetUpdateDelete(dummyFile, Date.now().toString()).catch(console.error);
7775

78-
listFiles().catch(console.error);
76+
if (isNode){
77+
uploadGetUpdateDelete(dummyFile, Date.now().toString()).catch(console.error);
78+
listFiles().catch(console.error);
79+
}
80+
else{
81+
// TODO - add node support for files
82+
console.log('Cannot run uploads in not Node environment');
83+
}

src/index.ts

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,9 +38,16 @@ export {
3838
type ConversationalRagResponse,
3939
type ConversationalRagSource,
4040
type RetrievalStrategy,
41+
type UploadFileRequest,
42+
type UploadFileResponse,
43+
type FileResponse,
44+
type ListFilesFilters,
45+
type UpdateFileRequest,
46+
type FilePathOrFileObject,
4147
} from './types';
4248
export { APIClient } from './APIClient';
4349
export { AI21Error, MissingAPIKeyError } from './errors';
4450
export { Stream } from './streaming';
4551
export { APIResource } from './APIResource';
46-
export { Chat, Completions, ConversationalRag } from './resources';
52+
export { Chat, Completions, ConversationalRag, Files } from './resources';
53+
export { isBrowser, isNode } from './runtime';

tests/unittests/resources/rag-engine.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ describe('RAGEngine', () => {
104104
'/library/files',
105105
{
106106
query: filters,
107-
headers: { 'Authorization': `Bearer ${dummyAPIKey}` }
107+
...options,
108108
}
109109
);
110110
expect(response).toEqual(expectedResponse);

0 commit comments

Comments
 (0)