Skip to content

Commit 9ac710e

Browse files
committed
use baseURL always in upload module
1 parent a5b0ee4 commit 9ac710e

File tree

2 files changed

+11
-5
lines changed

2 files changed

+11
-5
lines changed

examples/uploading-files.ts

+10-3
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,25 @@
11
#!/usr/bin/env -S npm run tsn -T
22

33
// Example of uploading a file
4-
import { upload } from 'together-ai/lib/upload';
4+
import { upload, ErrorResponse, FileResponse } from 'together-ai/lib/upload';
55
import fetch from 'node-fetch';
6+
import * as core from 'together-ai/core';
67

78
async function main() {
89
// Upload a file
9-
const file = await upload('./examples/coqa.jsonl');
10+
const file: ErrorResponse | FileResponse = await upload('./examples/coqa.jsonl');
1011
console.log('Uploaded file');
1112
console.log(file);
1213

14+
if ('message' in file) {
15+
console.error(file.message);
16+
return;
17+
}
18+
1319
// Check if the file has any contents (https://docs.together.ai/reference/get_files-id-content)
1420
// @ts-ignore
15-
const url = `https://api.together.xyz/v1/files/${file.id}/content`;
21+
const baseUrl = core.readEnv('TOGETHER_API_BASE_URL') || 'https://api.together.ai/v1';
22+
const url = `${baseUrl}/files/${file.id}/content`;
1623
const options = {
1724
method: 'GET',
1825
headers: {

src/lib/upload.ts

+1-2
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ const failedUploadMessage = {
2929
message: 'failed to upload file',
3030
};
3131

32-
const baseURL = 'https://api.together.xyz/v1';
32+
const baseURL = core.readEnv('TOGETHER_API_BASE_URL') || 'https://api.together.xyz/v1';
3333
const MAX_FILE_SIZE = 4.8; // GB
3434
const BYTES_PER_GB = 1024 * 1024 * 1024;
3535
const MIN_SAMPLES = 1;
@@ -203,7 +203,6 @@ export async function upload(fileName: string, check: boolean = true): Promise<F
203203
// 1. check if file exists
204204
// 2. get signed upload url
205205
// 3. upload file
206-
const baseUrl = core.readEnv('TOGETHER_API_BASE_URL') || 'https://api.together.ai/v1';
207206
const apiKey = core.readEnv('TOGETHER_API_KEY');
208207

209208
if (!apiKey) {

0 commit comments

Comments
 (0)