From f2f0fd2f759ed3fb12c4dea1e83d30f9dd6eb1fe Mon Sep 17 00:00:00 2001
From: udara <me@udara.io>
Date: Sat, 6 Jan 2024 17:38:28 -0500
Subject: [PATCH] Fix renderModelOptions provider

---
 src/renderer/pages/Pile/Editor/index.jsx   |  8 +++++---
 src/renderer/pages/Pile/Settings/index.jsx | 17 +++++++++++------
 2 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/src/renderer/pages/Pile/Editor/index.jsx b/src/renderer/pages/Pile/Editor/index.jsx
index 15d0009..19dc901 100644
--- a/src/renderer/pages/Pile/Editor/index.jsx
+++ b/src/renderer/pages/Pile/Editor/index.jsx
@@ -46,7 +46,7 @@ const Editor = memo(
       deletePost,
     } = usePost(postPath, { isReply, parentPostPath, reloadParentPost, isAI });
     const { getThread } = useThread();
-    const { ai, model, type, prompt } = useAIContext();
+    const { ai, provider, model, prompt } = useAIContext();
     const { addNotification, removeNotification } = useToastsContext();
 
     const isNew = !postPath;
@@ -249,7 +249,7 @@ const Editor = memo(
         if (context.length === 0) return;
 
         const stream = await ai.chat.completions.create({
-          model: model,
+          model: 'gpt-4',
           stream: true,
           max_tokens: 200,
           messages: context,
@@ -257,7 +257,9 @@ const Editor = memo(
 
         for await (const part of stream) {
           let token = '';
-          if (type == 'openai') {
+          console.log('model', model);
+          if (provider == 'openai') {
+            console.log('part.choices[0]', part);
             token = part.choices[0].delta.content;
           } else {
             token = part.message.content;
diff --git a/src/renderer/pages/Pile/Settings/index.jsx b/src/renderer/pages/Pile/Settings/index.jsx
index 009ddc0..551d808 100644
--- a/src/renderer/pages/Pile/Settings/index.jsx
+++ b/src/renderer/pages/Pile/Settings/index.jsx
@@ -77,21 +77,26 @@ export default function Settings() {
   };
 
   const renderModelOptions = () => {
-    return Object.keys(availableModelsProviders).map((model, index) => {
+    return Object.keys(availableModelsProviders).map((_provider, index) => {
       return (
         <button
-          key={`model-${model}`}
-          className={`${styles.model} ${provider == model && styles.current}`}
+          key={`model-${_provider}`}
+          className={`${styles.model} ${
+            provider == _provider && styles.current
+          }`}
           onClick={() => {
-            setProvider(model);
+            setProvider(_provider);
+            if (_provider == 'openai') {
+              setModel('gpt-4');
+            }
           }}
         >
-          {model === 'openai' ? (
+          {_provider === 'openai' ? (
             <OpenAIIcon className={styles.icon} />
           ) : (
             <OllamaIcon className={`${styles.icon} ${styles.ollama}`} />
           )}
-          {availableModelsProviders[model]}
+          {availableModelsProviders[_provider]}
         </button>
       );
     });