Merge pull request #34 from SkywardAI/chatgpt

invoke chatgpt & fix bugs

package.json

@@ -1,8 +1,8 @@
 {
   "name": "shibuya",
   "author": {
-      "name": "Bohan Cheng",
-      "email": "[email protected]"
+    "name": "Bohan Cheng",
+    "email": "[email protected]"
   },
   "version": "0.1.11",
   "main": "electron.js",
@@ -20,6 +20,7 @@
     "@aws-sdk/credential-providers": "^3.650.0",
     "@huggingface/jinja": "^0.3.0",
     "@wllama/wllama": "^1.16.0",
+    "openai": "^4.61.0",
     "react": "^18.3.1",
     "react-bootstrap-icons": "^1.11.4",
     "react-dom": "^18.3.1",

src/components/chat/Conversation.jsx

@@ -4,9 +4,10 @@ import { FileImageFill, FileTextFill, Paperclip, Send, StopCircleFill } from 're
 import useIDB from "../../utils/idb";
 import { isModelLoaded, loadModel } from '../../utils/workers/worker'
 import { getCompletionFunctions } from "../../utils/workers";
-import { setClient } from "../../utils/workers/aws-worker";
+import { setClient as setAwsClient } from "../../utils/workers/aws-worker";
+import { setClient as setOpenaiClient } from "../../utils/workers/openai-worker";
 
-export default function Conversation({ uid, client }) {
+export default function Conversation({ uid, client, updateClient }) {
 
     const [conversation, setConversation] = useState([]);
     const [message, setMessage] = useState('');
@@ -130,15 +131,21 @@ export default function Conversation({ uid, client }) {
     useEffect(()=>{
         if(!chat_functions.current) return;
 
-        if(chat_functions.current.platform === 'AWS') {
+        const platform = chat_functions.current.platform
+        if(platform) {
             (async function() {
-                if(await setClient(client)) {
-                    await idb.updateOne('chat-history', {client}, [{uid}])
+                let set_result = 
+                    platform === "AWS" ? await setAwsClient(client) :
+                    platform === "OpenAI" ? await setOpenaiClient(client) :
+                    null;
+
+                if(set_result) {
+                    updateClient(set_result);
                 }
             })()
         }
     // eslint-disable-next-line
-    }, [chat_functions, client])
+    }, [client])
 
     return (
         <div className="conversation-main">