Provider => hub mapping (#1806)

Author: julien-c

packages/inference/src/types.ts

@@ -76,6 +76,37 @@ export type InferenceProvider = (typeof INFERENCE_PROVIDERS)[number];
 
 export type InferenceProviderOrPolicy = (typeof PROVIDERS_OR_POLICIES)[number];
 
+/**
+ * The org namespace on the HF Hub i.e. hf.co/…
+ *
+ * Whenever possible, InferenceProvider should == org namespace
+ */
+export const PROVIDERS_HUB_ORGS: Record<InferenceProvider, string> = {
+	baseten: "baseten",
+	"black-forest-labs": "black-forest-labs",
+	cerebras: "cerebras",
+	clarifai: "clarifai",
+	cohere: "CohereLabs",
+	"fal-ai": "fal",
+	"featherless-ai": "featherless-ai",
+	"fireworks-ai": "fireworks-ai",
+	groq: "groq",
+	"hf-inference": "hf-inference",
+	hyperbolic: "Hyperbolic",
+	nebius: "nebius",
+	novita: "novita",
+	nscale: "nscale",
+	openai: "openai",
+	ovhcloud: "ovhcloud",
+	publicai: "publicai",
+	replicate: "replicate",
+	sambanova: "sambanovasystems",
+	scaleway: "scaleway",
+	together: "togethercomputer",
+	wavespeed: "wavespeed",
+	"zai-org": "zai-org",
+};
+
 export interface InferenceProviderMappingEntry {
 	adapter?: string;
 	adapterWeightsPath?: string;