pass model directly

components/Playground/index.tsx

@@ -155,7 +155,7 @@ const Playground = () => {
                   className="border p-2 rounded-md shadow-sm w-full bg-transparent text-gray-700 mt-2"
                 >
                   <option value="openai">gpt-4</option>
-                  <option value="huggingface">meta-llama/Llama-2-70b-chat-hf</option>
+                  <option value="meta-llama/Llama-2-70b-chat-hf">meta-llama/Llama-2-70b-chat-hf</option>
                 </select>
               </div>
             )}

pages/api/chat/index.ts

@@ -35,10 +35,10 @@ export default async function(req: Request) {
 
         const stream = OpenAIStream(response);
         return new StreamingTextResponse(stream);
-    } else if (aiProvider === 'huggingface') {
+    } else {
         const response = Hf.textGenerationStream({
             // @ts-ignore
-            model: 'meta-llama/Llama-2-70b-chat-hf',
+            model: aiProvider,
             inputs: experimental_buildLlama2Prompt(messages),
             parameters: {
               max_new_tokens: 500,
@@ -50,7 +50,5 @@ export default async function(req: Request) {
 
         const stream = HuggingFaceStream(response);
         return new StreamingTextResponse(stream);
-    } else {
-        throw new Error(`Unsupported AI provider: ${aiProvider}`);
     }
 }