Minor Improvements

genomehubs · Sep 19, 2024 · 62e21dc · 62e21dc
1 parent 4b137f5
commit 62e21dc
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 16 deletions.
diff --git a/INSTALL.md b/INSTALL.md
@@ -51,7 +51,7 @@ curl -fsSL https://ollama.com/install.sh | sh
 Run the Ollama application:
 
 ```bash
-ollama run lama3.1:8b-instruct-q4_0
+ollama run llama3.1:8b-instruct-q4_0
 ```
 
 ## Step 7: Start the Flask Application

diff --git a/src/agent/test_query_pipeline.py b/src/agent/test_query_pipeline.py
@@ -20,7 +20,7 @@
 load_dotenv()
 
 Settings.llm = Ollama(
-    model=os.getenv("OLLAMA_MODEL_NAME", "llama3.1:8b-instruct-q4_0"),
+    model=os.getenv("OLLAMA_MODEL_NAME", "lllama3.1:8b-instruct-q4_0"),
     base_url=os.getenv("OLLAMA_HOST_URL", "http://127.0.0.1:11434"),
     request_timeout=36000.0,
 )

diff --git a/src/app.py b/src/app.py
@@ -20,7 +20,7 @@
 from agent.query_pipeline import qp
 
 Settings.llm = Ollama(
-    model=os.getenv("OLLAMA_MODEL_NAME", "llama3.1:8b-instruct-q4_0"),
+    model=os.getenv("OLLAMA_MODEL_NAME", "lllama3.1:8b-instruct-q4_0"),
     base_url=os.getenv("OLLAMA_HOST_URL", "http://127.0.0.1:11434"),
     request_timeout=36000.0,
 )

diff --git a/ui/src/app/[id]/page.tsx b/ui/src/app/[id]/page.tsx
@@ -147,9 +147,9 @@ export default function Page({ params }: { params: { id: string } }) {
             // break;
           }
           else if (currentState['done']) {
-            addMessage({ role: "assistant", content: '<a style="font-size: 1.5rem; text-decoration: underline; font-style: italic;" href="' + currentState['url'] + '" target="_blank">GoaT Link!</a>', id: chatId });
-            addMessage({ role: "assistant", content: "Webpage Details:\nHere is the JSON curated after parsing the above webpage:\n\n" + currentState['api_response'], id: chatId });
             addMessage({ role: "assistant", content: currentState['markdown'], id: chatId });
+            addMessage({ role: "assistant", content: '<a style="font-size: 1.5rem; text-decoration: underline; font-style: italic;" href="' + currentState['url'] + '" target="_blank">GoaT Link!</a>', id: chatId });
+            addMessage({ role: "user", content: "Webpage Details:\nHere is the JSON curated after parsing the above webpage:\n\n" + currentState['api_response'], id: chatId });
             setMessages([...messages]);
             setLoadingSubmit(false);
             break;

diff --git a/ui/src/app/page.tsx b/ui/src/app/page.tsx
@@ -123,9 +123,9 @@ export default function Home() {
             // break;
           }
           else if (currentState['done']) {
-            addMessage({ role: "assistant", content: '<a style="font-size: 1.5rem; text-decoration: underline; font-style: italic;" href="' + currentState['url'] + '" target="_blank">GoaT Link!</a>', id: chatId });
-            addMessage({ role: "assistant", content: "Webpage Details:\nHere is the JSON curated after parsing the above webpage:\n\n" + JSON.stringify(currentState['api_response']), id: chatId });
             addMessage({ role: "assistant", content: currentState['markdown'], id: chatId });
+            addMessage({ role: "assistant", content: '<a style="font-size: 1.5rem; text-decoration: underline; font-style: italic;" href="' + currentState['url'] + '" target="_blank">GoaT Link!</a>', id: chatId });
+            addMessage({ role: "user", content: "Webpage Details:\nHere is the JSON curated after parsing the above webpage:\n\n" + JSON.stringify(currentState['api_response']), id: chatId });
             setMessages([...messages]);
             setLoadingSubmit(false);
             break;
@@ -249,7 +249,7 @@ export default function Home() {
         />
         <DialogContent className="flex flex-col space-y-4">
           <DialogHeader className="space-y-2">
-            <DialogTitle>Welcome to Ollama!</DialogTitle>
+            <DialogTitle>Welcome to Goat NLP!</DialogTitle>
             <DialogDescription>
               Enter your name to get started. This is just to personalize your
               experience.

diff --git a/ui/src/lib/model-helper.ts b/ui/src/lib/model-helper.ts
@@ -1,9 +1,9 @@
 export function getSelectedModel(): string {
-    if (typeof window !== 'undefined') {
-      const storedModel = localStorage.getItem('selectedModel');
-      return storedModel || 'gemma:2b';
-    } else {
-      // Default model
-      return 'gemma:2b';
-    }
-  }
+  if (typeof window !== 'undefined') {
+    const storedModel = localStorage.getItem('selectedModel');
+    return storedModel || 'llama3.1:8b-instruct-q4_0';
+  } else {
+    // Default model
+    return 'llama3.1:8b-instruct-q4_0';
+  }
+}