chore: move some tasks

Author: pr-Mais

extensions/firestore-huggingface-inference-api/functions/src/inference.ts

@@ -12,11 +12,15 @@ import {
   ZeroShotClassificationOutput,
   HfInferenceEndpoint,
   HfInference,
+  FeatureExtractionOutput,
+  ConversationalOutput,
 } from '@huggingface/inference';
 
 import config from './config';
 import { Task } from './tasks';
 import { FirestoreTableInput } from './types/table';
+import { conversational } from './tasks/conversational';
+import { featureExtraction } from './tasks/feature-extraction';
 
 /**
  * Validate inputs and create a task.
@@ -39,6 +43,8 @@ export async function runInference(
   | TokenClassificationOutput
   | TranslationOutput
   | ZeroShotClassificationOutput
+  | FeatureExtractionOutput
+  | ConversationalOutput
 > {
   switch (task) {
     case Task.fillMask: {
@@ -221,10 +227,10 @@ export async function runInference(
       return await inference.zeroShotClassification(options);
     }
     case Task.conversational: {
-      throw new Error('Conversational task is not supported yet');
+      return await conversational(snapshot, inference);
     }
     case Task.featureExtraction: {
-      throw new Error('Feature extraction task is not supported yet');
+      return await featureExtraction(snapshot, inference);
     }
 
     default: {

extensions/firestore-huggingface-inference-api/functions/src/tasks/conversational.ts

@@ -0,0 +1,13 @@
+import {
+  ConversationalOutput,
+  HfInference,
+  HfInferenceEndpoint,
+} from '@huggingface/inference';
+import { DocumentSnapshot } from 'firebase-admin/firestore';
+
+export function conversational(
+  data: DocumentSnapshot,
+  inference: HfInference | HfInferenceEndpoint,
+): Promise<ConversationalOutput> {
+  throw new Error('Conversational task is not supported yet');
+}

extensions/firestore-huggingface-inference-api/functions/src/tasks/feature-extraction.ts

@@ -0,0 +1,13 @@
+import {
+  FeatureExtractionOutput,
+  HfInference,
+  HfInferenceEndpoint,
+} from '@huggingface/inference';
+import { DocumentSnapshot } from 'firebase-admin/firestore';
+
+export function featureExtraction(
+  data: DocumentSnapshot,
+  inference: HfInference | HfInferenceEndpoint,
+): Promise<FeatureExtractionOutput> {
+  throw new Error('Feature Extraction task is not supported yet');
+}