Fixes issue 259 preventing the usage of other models not set in stati…

…cmodels
stackblitz-labs · Nov 17, 2024 · 736ce67 · 736ce67
1 parent 87064f7
commit 736ce67
Showing 1 changed file with 1 addition and 11 deletions.
diff --git a/app/lib/.server/llm/stream-text.ts b/app/lib/.server/llm/stream-text.ts
@@ -54,22 +54,12 @@ export function streamText(
   const processedMessages = messages.map((message) => {
     if (message.role === 'user') {
       const { model, provider, content } = extractPropertiesFromMessage(message);
-
-      if (MODEL_LIST.find((m) => m.name === model)) {
-        currentModel = model;
-      }
-
+      currentModel = model;
       currentProvider = provider;
-
       return { ...message, content };
     }
-
     return message; // No changes for non-user messages
   });
-  currentModel = 'llama3.1:8b';
-  console.log('Processed messages:', processedMessages);
-  console.log('Current model:', currentModel);
-  console.log('Current provider:', currentProvider);
   return _streamText({
     model: getModel(currentProvider, currentModel, env, apiKeys),
     system: getSystemPrompt(),