File tree Expand file tree Collapse file tree 4 files changed +5
-6
lines changed
extensions/firestore-huggingface-inference-api/functions/src Expand file tree Collapse file tree 4 files changed +5
-6
lines changed Original file line number Diff line number Diff line change 7
7
HfInferenceEndpoint ,
8
8
HfInference ,
9
9
} from '@huggingface/inference' ;
10
- import { runInference } from './run_inference ' ;
10
+ import { runInference } from './inference ' ;
11
11
12
12
/**
13
13
* Trigger inference on Firestore document creation.
Original file line number Diff line number Diff line change 1
1
import { describe , expect , test } from '@jest/globals' ;
2
- import { runInference } from './run_inference ' ;
2
+ import { runInference } from './inference ' ;
3
3
import { Task } from './tasks' ;
4
4
import { HfInference , HfInferenceEndpoint } from '@huggingface/inference' ;
5
- import { DocumentSnapshot } from 'firebase-admin/firestore' ;
6
5
7
6
const inference = new HfInference ( 'test' ) ;
8
7
Original file line number Diff line number Diff line change @@ -16,7 +16,7 @@ import {
16
16
17
17
import config from './config' ;
18
18
import { Task } from './tasks' ;
19
- import { FirestoreInput } from './types/table' ;
19
+ import { FirestoreTableInput } from './types/table' ;
20
20
21
21
/**
22
22
* Validate inputs and create a task.
@@ -254,7 +254,7 @@ function checkListInputs(inputs: any) {
254
254
}
255
255
}
256
256
257
- function validateFirestoreInput ( data : any ) : data is FirestoreInput {
257
+ function validateFirestoreInput ( data : any ) : data is FirestoreTableInput {
258
258
if ( ! data || typeof data !== 'object' || ! data . inputs ) {
259
259
return false ;
260
260
}
Original file line number Diff line number Diff line change @@ -7,6 +7,6 @@ interface InferenceInput {
7
7
table : TableData ;
8
8
}
9
9
10
- export interface FirestoreInput {
10
+ export interface FirestoreTableInput {
11
11
inputs : InferenceInput ;
12
12
}
You can’t perform that action at this time.
0 commit comments