From f61ce71d064bedfc7c198a20ffcc567cc94c917f Mon Sep 17 00:00:00 2001
From: hkjang <gagagiga@naver.com>
Date: Wed, 6 Nov 2024 14:32:36 +0900
Subject: [PATCH 01/27] add Koraen localization

---
 src/assets/locale/ko/chrome.json     |  13 +
 src/assets/locale/ko/common.json     | 115 +++++++++
 src/assets/locale/ko/knowledge.json  |  40 ++++
 src/assets/locale/ko/openai.json     |  90 +++++++
 src/assets/locale/ko/option.json     |  12 +
 src/assets/locale/ko/playground.json |  29 +++
 src/assets/locale/ko/settings.json   | 345 +++++++++++++++++++++++++++
 src/assets/locale/ko/sidepanel.json  |   5 +
 8 files changed, 649 insertions(+)
 create mode 100644 src/assets/locale/ko/chrome.json
 create mode 100644 src/assets/locale/ko/common.json
 create mode 100644 src/assets/locale/ko/knowledge.json
 create mode 100644 src/assets/locale/ko/openai.json
 create mode 100644 src/assets/locale/ko/option.json
 create mode 100644 src/assets/locale/ko/playground.json
 create mode 100644 src/assets/locale/ko/settings.json
 create mode 100644 src/assets/locale/ko/sidepanel.json

diff --git a/src/assets/locale/ko/chrome.json b/src/assets/locale/ko/chrome.json
new file mode 100644
index 0000000..8f89fc1
--- /dev/null
+++ b/src/assets/locale/ko/chrome.json
@@ -0,0 +1,13 @@
+{
+    "heading": "Chrome AI 설정",
+    "status": {
+        "label": "Page Assist에서 Chrome AI 지원을 활성화하거나 비활성화하기"
+    },
+    "error": {
+        "browser_not_supported": "이 Chrome 버전은 Gemini Nano 모델을 지원하지 않습니다. 버전을 127 이상으로 업데이트해 주세요.",
+        "ai_not_supported": "설정 `chrome://flags/#prompt-api-for-gemini-nano`가 활성화되지 않았습니다. 활성화해 주세요.",
+        "ai_not_ready": "Gemini Nano가 아직 준비되지 않았습니다. Chrome 설정을 다시 확인해 주세요.",
+        "internal_error": "내부 오류가 발생했습니다. 나중에 다시 시도해 주세요."
+    },
+    "errorDescription": "Chrome AI를 사용하려면 현재 Dev 및 Canary 채널에 있는 127 이상의 브라우저 버전이 필요합니다. 지원되는 버전을 다운로드한 후, 다음 단계를 따르세요:\n\n1. `chrome://flags/#prompt-api-for-gemini-nano`에 접속하여 '활성화'를 선택합니다.\n2. `chrome://flags/#optimization-guide-on-device-model`에 접속하여 'EnabledBypassPrefRequirement'를 선택합니다.\n3. `chrome://components`에 접속하여 'Optimization Guide On Device Model'을 검색한 후 '업데이트 확인'을 클릭합니다. 이를 통해 모델이 다운로드됩니다. 설정이 표시되지 않는 경우, 단계 1과 2를 반복하고 브라우저를 재시작해 주세요."
+}
diff --git a/src/assets/locale/ko/common.json b/src/assets/locale/ko/common.json
new file mode 100644
index 0000000..908bb0d
--- /dev/null
+++ b/src/assets/locale/ko/common.json
@@ -0,0 +1,115 @@
+{
+  "pageAssist": "페이지 어시스트",
+  "selectAModel": "모델 선택",
+  "save": "저장",
+  "saved": "저장됨",
+  "cancel": "취소",
+  "retry": "재시도",
+  "share": {
+    "tooltip": {
+      "share": "공유"
+    },
+    "modal": {
+      "title": "채팅 링크 공유"
+    },
+    "form": {
+      "defaultValue": {
+        "name": "익명",
+        "title": "제목 없는 채팅"
+      },
+      "title": {
+        "label": "채팅 제목",
+        "placeholder": "채팅 제목을 입력하세요",
+        "required": "채팅 제목은 필수 항목입니다"
+      },
+      "name": {
+        "label": "이름",
+        "placeholder": "이름을 입력하세요",
+        "required": "이름은 필수 항목입니다"
+      },
+      "btn": {
+        "save": "링크 생성",
+        "saving": "링크 생성 중..."
+      }
+    },
+    "notification": {
+      "successGenerate": "링크가 클립보드에 복사되었습니다",
+      "failGenerate": "링크 생성에 실패했습니다"
+    }
+  },
+  "copyToClipboard": "클립보드에 복사",
+  "webSearch": "웹 검색 중",
+  "regenerate": "재생성",
+  "edit": "편집",
+  "delete": "삭제",
+  "saveAndSubmit": "저장하고 제출",
+  "editMessage": {
+    "placeholder": "메시지를 입력하세요..."
+  },
+  "submit": "제출",
+  "noData": "데이터가 없습니다",
+  "noHistory": "채팅 기록이 없습니다",
+  "chatWithCurrentPage": "현재 페이지에서 채팅",
+  "beta": "베타",
+  "tts": "TTS",
+  "currentChatModelSettings": "현재 채팅 모델 설정",
+  "modelSettings": {
+    "label": "모델 설정",
+    "description": "모든 채팅에 대해 글로벌 모델 옵션을 설정합니다",
+    "form": {
+      "keepAlive": {
+        "label": "Keep Alive",
+        "help": "요청 후 모델이 메모리에 유지되는 시간을 설정합니다 (기본값: 5분)",
+        "placeholder": "Keep Alive 기간을 입력하세요 (예: 5분, 10분, 1시간)"
+      },
+      "temperature": {
+        "label": "온도",
+        "placeholder": "온도 값을 입력하세요 (예: 0.7, 1.0)"
+      },
+      "numCtx": {
+        "label": "컨텍스트 수",
+        "placeholder": "컨텍스트 수를 입력하세요 (기본값: 2048)"
+      },
+      "seed": {
+        "label": "시드",
+        "placeholder": "시드 값을 입력하세요 (예: 1234)",
+        "help": "모델 출력의 재현성"
+      },
+      "topK": {
+        "label": "Top K",
+        "placeholder": "Top K 값을 입력하세요 (예: 40, 100)"
+      },
+      "topP": {
+        "label": "Top P",
+        "placeholder": "Top P 값을 입력하세요 (예: 0.9, 0.95)"
+      },
+      "numGpu": {
+        "label": "GPU 수",
+        "placeholder": "GPU에 할당할 레이어 수를 입력하세요"
+      },
+      "systemPrompt": {
+        "label": "임시 시스템 프롬프트",
+        "placeholder": "시스템 프롬프트를 입력하세요",
+        "help": "현재 채팅에서 시스템 프롬프트를 빠르게 설정하는 방법이며, 선택된 시스템 프롬프트가 있을 경우 이를 덮어씁니다."
+      }
+    },
+    "advanced": "기타 모델 설정"
+  },
+  "copilot": {
+    "summary": "요약",
+    "explain": "설명",
+    "rephrase": "다르게 표현",
+    "translate": "번역"
+  },
+  "citations": "인용",
+  "downloadCode": "코드 다운로드",
+  "date": {
+    "pinned": "고정됨",
+    "today": "오늘",
+    "yesterday": "어제",
+    "last7Days": "지난 7일",
+    "older": "그 이전"
+  },
+  "pin": "고정",
+  "unpin": "고정 해제"
+}
diff --git a/src/assets/locale/ko/knowledge.json b/src/assets/locale/ko/knowledge.json
new file mode 100644
index 0000000..1bf8536
--- /dev/null
+++ b/src/assets/locale/ko/knowledge.json
@@ -0,0 +1,40 @@
+{
+    "addBtn": "새 지식 추가",
+    "columns": {
+        "title": "제목",
+        "status": "상태",
+        "embeddings": "임베딩 모델",
+        "createdAt": "생성일",
+        "action": "작업"
+    },
+    "expandedColumns": {
+        "name": "이름"
+    },
+    "confirm": {
+        "delete": "이 지식을 삭제하시겠습니까?"
+    },
+    "deleteSuccess": "지식이 정상적으로 삭제되었습니다",
+    "status": {
+        "pending": "대기 중",
+        "finished": "완료",
+        "processing": "처리 중",
+        "failed": "실패"
+    },
+    "addKnowledge": "지식 추가",
+    "form": {
+        "title": {
+            "label": "지식 제목",
+            "placeholder": "지식 제목을 입력하세요",
+            "required": "지식 제목은 필수 항목입니다"
+        },
+        "uploadFile": {
+            "label": "파일 업로드",
+            "uploadText": "파일을 여기에 드래그 앤 드롭하거나 클릭하여 업로드하세요",
+            "uploadHint": "지원되는 파일 형식: .pdf, .csv, .txt",
+            "required": "파일은 필수 항목입니다"
+        },
+        "submit": "제출",
+        "success": "지식이 정상적으로 추가되었습니다"
+    },
+    "noEmbeddingModel": "먼저 RAG 설정 페이지에서 임베딩 모델을 추가해 주세요"
+}
diff --git a/src/assets/locale/ko/openai.json b/src/assets/locale/ko/openai.json
new file mode 100644
index 0000000..13bc431
--- /dev/null
+++ b/src/assets/locale/ko/openai.json
@@ -0,0 +1,90 @@
+{
+    "settings": "OpenAI 호환 API",
+    "heading": "OpenAI 호환 API",
+    "subheading": "여기에서 OpenAI API 호환 공급자를 관리하고 설정할 수 있습니다.",
+    "addBtn": "공급자 추가",
+    "table": {
+        "name": "공급자 이름",
+        "baseUrl": "기본 URL",
+        "actions": "작업"
+    },
+    "modal": {
+        "titleAdd": "새 공급자 추가",
+        "name": {
+            "label": "공급자 이름",
+            "required": "공급자 이름은 필수 항목입니다.",
+            "placeholder": "공급자 이름 입력"
+        },
+        "baseUrl": {
+            "label": "기본 URL",
+            "help": "OpenAI API 공급자의 기본 URL 예시: (http://localhost:1234/v1)",
+            "required": "기본 URL은 필수 항목입니다.",
+            "placeholder": "기본 URL 입력"
+        },
+        "apiKey": {
+            "label": "API 키",
+            "required": "API 키는 필수 항목입니다.",
+            "placeholder": "API 키 입력"
+        },
+        "submit": "저장",
+        "update": "업데이트",
+        "deleteConfirm": "이 공급자를 삭제하시겠습니까?",
+        "model": {
+            "title": "모델 목록",
+            "subheading": "이 공급자에서 사용하고자 하는 챗 모델을 선택하세요.",
+            "success": "새로운 모델이 정상적으로 추가되었습니다."
+        },
+        "tipLMStudio": "Page Assist는 LM Studio에 로드된 모델을 자동으로 가져옵니다. 수동 추가가 필요하지 않습니다."
+    },
+    "addSuccess": "공급자가 정상적으로 추가되었습니다.",
+    "deleteSuccess": "공급자가 정상적으로 삭제되었습니다.",
+    "updateSuccess": "공급자가 정상적으로 업데이트되었습니다.",
+    "delete": "삭제",
+    "edit": "편집",
+    "newModel": "공급자에 모델 추가",
+    "noNewModel": "LMStudio의 경우 동적으로 가져옵니다. 수동 추가는 필요하지 않습니다.",
+    "searchModel": "모델 검색",
+    "selectAll": "모두 선택",
+    "save": "저장",
+    "saving": "저장 중...",
+    "manageModels": {
+        "columns": {
+            "name": "모델 이름",
+            "model_type": "모델 타입",
+            "model_id": "모델 ID",
+            "provider": "공급자 이름",
+            "actions": "작업"
+        },
+        "tooltip": {
+            "delete": "삭제"
+        },
+        "confirm": {
+            "delete": "이 모델을 삭제하시겠습니까?"
+        },
+        "modal": {
+            "title": "사용자 정의 모델 추가",
+            "form": {
+                "name": {
+                    "label": "모델 ID",
+                    "placeholder": "llama3.2",
+                    "required": "모델 ID는 필수 항목입니다."
+                },
+                "provider": {
+                    "label": "공급자",
+                    "placeholder": "공급자 선택",
+                    "required": "공급자는 필수 항목입니다."
+                },
+                "type": {
+                    "label": "모델 타입"
+                }
+            }
+        }
+    },
+    "noModelFound": "모델을 찾을 수 없습니다. 올바른 기본 URL과 API 키를 가진 공급자가 추가되었는지 확인하세요.",
+    "radio": {
+        "chat": "챗 모델",
+        "embedding": "임베딩 모델",
+        "chatInfo": "는 챗 완료 및 대화 생성에 사용됩니다",
+        "embeddingInfo": "는 RAG 및 기타 의미 검색 관련 작업에 사용됩니다."
+    }
+}
diff --git a/src/assets/locale/ko/option.json b/src/assets/locale/ko/option.json
new file mode 100644
index 0000000..21a5048
--- /dev/null
+++ b/src/assets/locale/ko/option.json
@@ -0,0 +1,12 @@
+{
+    "newChat": "새 채팅",
+    "selectAPrompt": "프롬프트 선택",
+    "githubRepository": "GitHub 리포지토리",
+    "settings": "설정",
+    "sidebarTitle": "채팅 기록",
+    "error": "오류",
+    "somethingWentWrong": "문제가 발생했습니다",
+    "validationSelectModel": "계속하려면 모델을 선택하세요",
+    "deleteHistoryConfirmation": "이 기록을 삭제하시겠습니까?",
+    "editHistoryTitle": "새 제목 입력"
+}
diff --git a/src/assets/locale/ko/playground.json b/src/assets/locale/ko/playground.json
new file mode 100644
index 0000000..082ad5d
--- /dev/null
+++ b/src/assets/locale/ko/playground.json
@@ -0,0 +1,29 @@
+{
+    "ollamaState": {
+        "searching": "Ollama 검색 중 🦙",
+        "running": "Ollama 실행 중 🦙",
+        "notRunning": "Ollama에 연결할 수 없습니다 🦙",
+        "connectionError": "연결 오류가 발생한 것 같습니다. 문제 해결에 대한 자세한 내용은 <anchor>문서</anchor>를 참조하세요."
+    },
+    "formError": {
+        "noModel": "모델을 선택하세요",
+        "noEmbeddingModel": "설정 > RAG 페이지에서 임베딩 모델을 설정하세요"
+    },
+    "form": {
+        "textarea": {
+            "placeholder": "메시지를 입력하세요..."
+        },
+        "webSearch": {
+            "on": "켜짐",
+            "off": "꺼짐"
+        }
+    },
+    "tooltip": {
+        "searchInternet": "인터넷 검색",
+        "speechToText": "음성 입력",
+        "uploadImage": "이미지 업로드",
+        "stopStreaming": "스트리밍 중지",
+        "knowledge": "지식"
+    },
+    "sendWhenEnter": "Enter 키를 누르면 전송"
+}
diff --git a/src/assets/locale/ko/settings.json b/src/assets/locale/ko/settings.json
new file mode 100644
index 0000000..072c7f6
--- /dev/null
+++ b/src/assets/locale/ko/settings.json
@@ -0,0 +1,345 @@
+{
+  "generalSettings": {
+    "title": "일반 설정",
+    "settings": {
+      "heading": "웹 UI 설정",
+      "speechRecognitionLang": {
+        "label": "음성 인식 언어",
+        "placeholder": "언어 선택"
+      },
+      "language": {
+        "label": "언어",
+        "placeholder": "언어 선택"
+      },
+      "darkMode": {
+        "label": "테마 변경",
+        "options": {
+          "light": "라이트",
+          "dark": "다크"
+        }
+      },
+      "searchMode": {
+        "label": "간편 인터넷 검색 실행"
+      },
+      "copilotResumeLastChat": {
+        "label": "사이드 패널을 열 때 마지막 채팅 재개 (Copilot)"
+      },
+      "hideCurrentChatModelSettings": {
+        "label": "현재 채팅 모델 설정 숨기기"
+      },
+      "restoreLastChatModel": {
+        "label": "이전 채팅에서 마지막 사용한 모델 복원"
+      },
+      "sendNotificationAfterIndexing": {
+        "label": "지식 베이스 처리 완료 후 알림 전송"
+      },
+      "generateTitle": {
+        "label": "AI로 제목 생성"
+      }
+    },
+    "sidepanelRag": {
+      "heading": "웹사이트와의 Copilot 채팅 설정",
+      "ragEnabled": {
+        "label": "벡터 임베딩을 사용하여 웹사이트와 채팅"
+      },
+      "maxWebsiteContext": {
+        "label": "일반 모드 웹사이트 콘텐츠 크기",
+        "placeholder": "콘텐츠 크기 (기본값 4028)"
+      }
+    },
+    "webSearch": {
+      "heading": "웹 검색 관리",
+      "searchMode": {
+        "label": "간편한 인터넷 검색 실행"
+      },
+      "provider": {
+        "label": "검색 엔진",
+        "placeholder": "검색 엔진 선택"
+      },
+      "totalSearchResults": {
+        "label": "총 검색 결과",
+        "placeholder": "총 검색 결과 입력"
+      },
+      "visitSpecificWebsite": {
+        "label": "메시지에 언급된 웹사이트 방문"
+      }
+    },
+    "system": {
+      "heading": "시스템 설정",
+      "deleteChatHistory": {
+        "label": "채팅 기록 삭제",
+        "button": "삭제",
+        "confirm": "채팅 기록을 삭제하시겠습니까? 이 작업은 되돌릴 수 없습니다."
+      },
+      "export": {
+        "label": "채팅 기록, 지식 베이스, 프롬프트 내보내기",
+        "button": "데이터 내보내기",
+        "success": "내보내기 성공"
+      },
+      "import": {
+        "label": "채팅 기록, 지식 베이스, 프롬프트 가져오기",
+        "button": "데이터 가져오기",
+        "success": "가져오기 성공",
+        "error": "가져오기 오류"
+      }
+    },
+    "tts": {
+      "heading": "텍스트 음성 변환 설정",
+      "ttsEnabled": {
+        "label": "텍스트 음성 변환 활성화"
+      },
+      "ttsProvider": {
+        "label": "텍스트 음성 변환 제공자",
+        "placeholder": "제공자 선택"
+      },
+      "ttsVoice": {
+        "label": "텍스트 음성 변환 음성",
+        "placeholder": "음성 선택"
+      },
+      "ssmlEnabled": {
+        "label": "SSML (Speech Synthesis Markup Language) 활성화"
+      }
+    }
+  },
+  "manageModels": {
+    "title": "모델 관리",
+    "addBtn": "새 모델 추가",
+    "columns": {
+      "name": "이름",
+      "digest": "다이제스트",
+      "modifiedAt": "수정 일시",
+      "size": "크기",
+      "actions": "동작"
+    },
+    "expandedColumns": {
+      "parentModel": "상위 모델",
+      "format": "형식",
+      "family": "패밀리",
+      "parameterSize": "파라미터 크기",
+      "quantizationLevel": "양자화 수준"
+    },
+    "tooltip": {
+      "delete": "모델 삭제",
+      "repull": "모델 다시 가져오기"
+    },
+    "confirm": {
+      "delete": "이 모델을 정말 삭제하시겠습니까?",
+      "repull": "이 모델을 정말 다시 가져오시겠습니까?"
+    },
+    "modal": {
+      "title": "새 모델 추가",
+      "placeholder": "모델 이름 입력",
+      "pull": "모델 가져오기"
+    },
+    "notification": {
+      "pullModel": "모델 가져오는 중",
+      "pullModelDescription": "{{modelName}} 모델을 가져오는 중입니다. 자세한 내용은 확장 기능 아이콘을 확인하세요.",
+      "success": "성공",
+      "error": "오류",
+      "successDescription": "모델 가져오기가 완료되었습니다",
+      "successDeleteDescription": "모델 삭제가 완료되었습니다",
+      "someError": "문제가 발생했습니다. 나중에 다시 시도해 주세요."
+    }
+  },
+  "managePrompts": {
+    "title": "프롬프트 관리",
+    "addBtn": "새 프롬프트 추가",
+    "option1": "일반",
+    "option2": "RAG",
+    "questionPrompt": "질문 프롬프트",
+    "columns": {
+      "title": "제목",
+      "prompt": "프롬프트",
+      "type": "프롬프트 유형",
+      "actions": "동작"
+    },
+    "systemPrompt": "시스템 프롬프트",
+    "quickPrompt": "퀵 프롬프트",
+    "tooltip": {
+      "delete": "프롬프트 삭제",
+      "edit": "프롬프트 수정"
+    },
+    "confirm": {
+      "delete": "이 프롬프트를 정말 삭제하시겠습니까? 이 작업은 되돌릴 수 없습니다."
+    },
+    "modal": {
+      "addTitle": "새 프롬프트 추가",
+      "editTitle": "프롬프트 수정"
+    },
+    "segmented": {
+      "custom": "커스텀 프롬프트",
+      "copilot": "Copilot 프롬프트"
+    },
+    "form": {
+      "title": {
+        "label": "제목",
+        "placeholder": "훌륭한 프롬프트",
+        "required": "제목을 입력하세요"
+      },
+      "prompt": {
+        "label": "프롬프트",
+        "placeholder": "프롬프트 입력",
+        "required": "프롬프트를 입력하세요",
+        "help": "프롬프트 내에서 {key}를 변수로 사용할 수 있습니다.",
+        "missingTextPlaceholder": "프롬프트에 {text} 변수가 없습니다. 추가해 주세요."
+      },
+      "isSystem": {
+        "label": "시스템 프롬프트"
+      },
+      "btnSave": {
+        "saving": "프롬프트 추가 중...",
+        "save": "프롬프트 추가"
+      },
+      "btnEdit": {
+        "saving": "프롬프트 업데이트 중...",
+        "save": "프롬프트 업데이트"
+      }
+    },
+    "notification": {
+      "addSuccess": "프롬프트가 추가되었습니다",
+      "addSuccessDesc": "프롬프트가 정상적으로 추가되었습니다",
+      "error": "오류",
+      "someError": "문제가 발생했습니다. 나중에 다시 시도해 주세요.",
+      "updatedSuccess": "프롬프트가 업데이트되었습니다",
+      "updatedSuccessDesc": "프롬프트가 정상적으로 업데이트되었습니다",
+      "deletedSuccess": "프롬프트가 삭제되었습니다",
+      "deletedSuccessDesc": "프롬프트가 정상적으로 삭제되었습니다"
+    }
+  },
+  "manageShare": {
+    "title": "공유 관리",
+    "heading": "페이지 공유 URL 설정",
+    "form": {
+      "url": {
+        "label": "페이지 공유 URL",
+        "placeholder": "페이지 공유 URL 입력",
+        "required": "페이지 공유 URL을 입력해 주세요!",
+        "help": "개인정보 보호를 위해 페이지 공유를 자체 호스팅하고, 해당 URL을 여기에 입력할 수 있습니다. <anchor>자세히 보기</anchor>"
+      }
+    },
+    "webshare": {
+      "heading": "웹 공유",
+      "columns": {
+        "title": "제목",
+        "url": "URL",
+        "actions": "동작"
+      },
+      "tooltip": {
+        "delete": "공유 삭제"
+      },
+      "confirm": {
+        "delete": "이 공유를 정말 삭제하시겠습니까? 이 작업은 되돌릴 수 없습니다."
+      },
+      "label": "페이지 공유 관리",
+      "description": "페이지 공유 기능을 활성화 또는 비활성화"
+    },
+    "notification": {
+      "pageShareSuccess": "페이지 공유 URL이 정상적으로 업데이트되었습니다",
+      "someError": "문제가 발생했습니다. 나중에 다시 시도해 주세요.",
+      "webShareDeleteSuccess": "웹 공유가 정상적으로 삭제되었습니다"
+    }
+  },
+  "ollamaSettings": {
+    "title": "Ollama 설정",
+    "heading": "Ollama 설정하기",
+    "settings": {
+      "ollamaUrl": {
+        "label": "Ollama URL",
+        "placeholder": "Ollama URL 입력"
+      },
+      "advanced": {
+        "label": "Ollama URL 고급 설정",
+        "urlRewriteEnabled": {
+          "label": "사용자 지정 Origin URL 활성화 또는 비활성화"
+        },
+        "rewriteUrl": {
+          "label": "사용자 지정 Origin URL",
+          "placeholder": "사용자 지정 Origin URL 입력"
+        },
+        "headers": {
+          "label": "사용자 지정 헤더",
+          "add": "헤더 추가",
+          "key": {
+            "label": "헤더 키",
+            "placeholder": "인증"
+          },
+          "value": {
+            "label": "헤더 값",
+            "placeholder": "베어러 토큰"
+          }
+        },
+        "help": "Page Assist에서 Ollama 연결에 문제가 있는 경우 사용자 지정 Origin URL을 설정할 수 있습니다. 설정에 대한 자세한 내용은 <anchor>여기를 클릭</anchor>하세요."
+      }
+    }
+  },
+  "manageSearch": {
+    "title": "웹 검색 관리",
+    "heading": "웹 검색 설정하기"
+  },
+  "about": {
+    "title": "소개",
+    "heading": "소개",
+    "chromeVersion": "Page Assist 버전",
+    "ollamaVersion": "Ollama 버전",
+    "support": "Page Assist 프로젝트는 다음 플랫폼에서 기부나 후원을 통해 지원할 수 있습니다:",
+    "koFi": "Ko-fi로 후원하기",
+    "githubSponsor": "GitHub에서 후원하기",
+    "githubRepo": "GitHub 저장소"
+  },
+  "manageKnowledge": {
+    "title": "지식 관리",
+    "heading": "지식 베이스 구성하기"
+  },
+  "rag": {
+    "title": "RAG 설정",
+    "ragSettings": {
+      "label": "RAG 설정",
+      "model": {
+        "label": "임베딩 모델",
+        "required": "모델을 선택해주세요",
+        "help": "`nomic-embed-text`와 같은 임베딩 모델 사용을 강력히 권장합니다.",
+        "placeholder": "모델 선택"
+      },
+      "chunkSize": {
+        "label": "청크 크기",
+        "placeholder": "청크 크기 입력",
+        "required": "청크 크기를 입력해주세요"
+      },
+      "chunkOverlap": {
+        "label": "청크 오버랩",
+        "placeholder": "청크 오버랩 입력",
+        "required": "청크 오버랩을 입력해주세요"
+      },
+      "totalFilePerKB": {
+        "label": "지식 베이스 기본 파일 업로드 제한",
+        "placeholder": "기본 파일 업로드 제한 입력 (예: 10)",
+        "required": "기본 파일 업로드 제한을 입력해주세요"
+      },
+      "noOfRetrievedDocs": {
+        "label": "검색 문서 수",
+        "placeholder": "검색 문서 수 입력",
+        "required": "검색 문서 수를 입력해주세요"
+      }
+    },
+    "prompt": {
+      "label": "RAG 프롬프트 설정",
+      "option1": "일반",
+      "option2": "웹",
+      "alert": "여기서 시스템 프롬프트를 설정하는 것은 더 이상 권장되지 않습니다. 프롬프트 추가 및 편집은 '프롬프트 관리' 섹션을 이용해주세요. 이 섹션은 향후 릴리스에서 제거될 예정입니다.",
+      "systemPrompt": "시스템 프롬프트",
+      "systemPromptPlaceholder": "시스템 프롬프트 입력",
+      "webSearchPrompt": "웹 검색 프롬프트",
+      "webSearchPromptHelp": "프롬프트에서 `{search_results}`를 제거하지 마세요.",
+      "webSearchPromptError": "웹 검색 프롬프트를 입력해주세요",
+      "webSearchPromptPlaceholder": "웹 검색 프롬프트 입력",
+      "webSearchFollowUpPrompt": "웹 검색 후속 프롬프트",
+      "webSearchFollowUpPromptHelp": "프롬프트에서 `{chat_history}`와 `{question}`를 제거하지 마세요.",
+      "webSearchFollowUpPromptError": "웹 검색 후속 프롬프트를 입력해주세요!",
+      "webSearchFollowUpPromptPlaceholder": "웹 검색 후속 프롬프트"
+    }
+  },
+  "chromeAiSettings": {
+    "title": "Chrome AI 설정"
+  }
+}
+
diff --git a/src/assets/locale/ko/sidepanel.json b/src/assets/locale/ko/sidepanel.json
new file mode 100644
index 0000000..b7a5eea
--- /dev/null
+++ b/src/assets/locale/ko/sidepanel.json
@@ -0,0 +1,5 @@
+{
+  "tooltip": {
+    "embed": "페이지를 임베드하는 데 몇 분이 걸릴 수 있습니다. 잠시만 기다려 주세요..."
+  }
+}

From 9f383a81b64c4faf9b5bf5b66a28621ffed89818 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 9 Nov 2024 15:17:59 +0530
Subject: [PATCH 02/27] feat: Add generation info to messages

This commit introduces a new feature that displays generation information for each message in the chat.

The generation info is displayed in a popover and includes details about the model used, the prompt, and other relevant information. This helps users understand how their messages were generated and troubleshoot any issues that may arise.

The generation info is retrieved from the LLM response and is stored in the database alongside other message details.

This commit also includes translations for the generation info label in all supported languages.
---
 src/assets/locale/da/common.json              |  3 +-
 src/assets/locale/de/common.json              |  3 +-
 src/assets/locale/en/common.json              |  3 +-
 src/assets/locale/es/common.json              |  3 +-
 src/assets/locale/fa/common.json              |  3 +-
 src/assets/locale/fr/common.json              |  3 +-
 src/assets/locale/it/common.json              |  3 +-
 src/assets/locale/ja-JP/common.json           |  3 +-
 src/assets/locale/ko/common.json              |  3 +-
 src/assets/locale/ml/common.json              |  4 +-
 src/assets/locale/no/common.json              |  3 +-
 src/assets/locale/pt-BR/common.json           |  3 +-
 src/assets/locale/ru/common.json              |  3 +-
 src/assets/locale/sv/common.json              |  3 +-
 src/assets/locale/zh/common.json              |  3 +-
 .../Common/Playground/GenerationInfo.tsx      | 65 +++++++++++++
 src/components/Common/Playground/Message.tsx  | 17 +++-
 .../Option/Playground/PlaygroundChat.tsx      |  1 +
 src/components/Sidepanel/Chat/body.tsx        |  1 +
 src/db/index.ts                               |  7 +-
 src/hooks/chat-helper/index.ts                | 15 ++-
 src/hooks/useMessage.tsx                      | 91 ++++++++++++++++---
 src/hooks/useMessageOption.tsx                | 63 +++++++++++--
 src/models/index.ts                           |  2 +-
 src/types/message.ts                          | 37 ++++----
 wxt.config.ts                                 |  2 +-
 26 files changed, 283 insertions(+), 64 deletions(-)
 create mode 100644 src/components/Common/Playground/GenerationInfo.tsx

diff --git a/src/assets/locale/da/common.json b/src/assets/locale/da/common.json
index 4d0de2d..881dab7 100644
--- a/src/assets/locale/da/common.json
+++ b/src/assets/locale/da/common.json
@@ -112,5 +112,6 @@
         "older": "Ældre"
     },
     "pin": "Fastgør",
-    "unpin": "Frigør"
+    "unpin": "Frigør",
+    "generationInfo": "Genererings Info"
 }
\ No newline at end of file
diff --git a/src/assets/locale/de/common.json b/src/assets/locale/de/common.json
index 739e0dd..9a69b22 100644
--- a/src/assets/locale/de/common.json
+++ b/src/assets/locale/de/common.json
@@ -112,5 +112,6 @@
         "older": "Älter"
     },
     "pin": "Anheften",
-    "unpin": "Losheften"
+    "unpin": "Losheften",
+    "generationInfo": "Generierungsinformationen"
 }
\ No newline at end of file
diff --git a/src/assets/locale/en/common.json b/src/assets/locale/en/common.json
index f5b9e39..a7a3ecc 100644
--- a/src/assets/locale/en/common.json
+++ b/src/assets/locale/en/common.json
@@ -116,5 +116,6 @@
         "older": "Older"
     },
     "pin": "Pin",
-    "unpin": "Unpin"
+    "unpin": "Unpin",
+    "generationInfo": "Generation Info"
 }
diff --git a/src/assets/locale/es/common.json b/src/assets/locale/es/common.json
index 65590aa..cf5ccf8 100644
--- a/src/assets/locale/es/common.json
+++ b/src/assets/locale/es/common.json
@@ -111,5 +111,6 @@
         "older": "Más antiguo"
     },
     "pin": "Fijar",
-    "unpin": "Desfijar"
+    "unpin": "Desfijar",
+    "generationInfo": "Información de Generación"
 }
\ No newline at end of file
diff --git a/src/assets/locale/fa/common.json b/src/assets/locale/fa/common.json
index 4adb307..6abb70a 100644
--- a/src/assets/locale/fa/common.json
+++ b/src/assets/locale/fa/common.json
@@ -105,5 +105,6 @@
         "older": "قدیمی‌تر"
     },
     "pin": "پین کردن",
-    "unpin": "حذف پین"
+    "unpin": "حذف پین",
+    "generationInfo": "اطلاعات تولید"
 }
\ No newline at end of file
diff --git a/src/assets/locale/fr/common.json b/src/assets/locale/fr/common.json
index 8dbe609..0775303 100644
--- a/src/assets/locale/fr/common.json
+++ b/src/assets/locale/fr/common.json
@@ -111,5 +111,6 @@
         "older": "Plus ancien"
     },
     "pin": "Épingler",
-    "unpin": "Désépingler"
+    "unpin": "Désépingler",
+    "generationInfo": "Informations de génération"
 }
\ No newline at end of file
diff --git a/src/assets/locale/it/common.json b/src/assets/locale/it/common.json
index 80ede68..102b6c8 100644
--- a/src/assets/locale/it/common.json
+++ b/src/assets/locale/it/common.json
@@ -111,5 +111,6 @@
         "older": "Più Vecchi"
     },
     "pin": "Fissa",
-    "unpin": "Rimuovi"
+    "unpin": "Rimuovi",
+    "generationInfo": "Informazioni sulla Generazione"
 }
\ No newline at end of file
diff --git a/src/assets/locale/ja-JP/common.json b/src/assets/locale/ja-JP/common.json
index df20028..6f38578 100644
--- a/src/assets/locale/ja-JP/common.json
+++ b/src/assets/locale/ja-JP/common.json
@@ -111,5 +111,6 @@
     "older": "それ以前"
   },
   "pin": "固定",
-  "unpin": "固定解除"
+  "unpin": "固定解除",
+  "generationInfo": "生成情報"
 }
\ No newline at end of file
diff --git a/src/assets/locale/ko/common.json b/src/assets/locale/ko/common.json
index 908bb0d..e59e785 100644
--- a/src/assets/locale/ko/common.json
+++ b/src/assets/locale/ko/common.json
@@ -111,5 +111,6 @@
     "older": "그 이전"
   },
   "pin": "고정",
-  "unpin": "고정 해제"
+  "unpin": "고정 해제",
+  "generationInfo": "생성 정보"
 }
diff --git a/src/assets/locale/ml/common.json b/src/assets/locale/ml/common.json
index bb1149a..ed6988a 100644
--- a/src/assets/locale/ml/common.json
+++ b/src/assets/locale/ml/common.json
@@ -110,5 +110,7 @@
         "older": "പഴയത്"
     },
     "pin": "പിൻ ചെയ്യുക",
-    "unpin": "അൺപിൻ ചെയ്യുക"
+    "unpin": "അൺപിൻ ചെയ്യുക",
+    "generationInfo": "ജനറേഷൻ വിവരങ്ങൾ"
+
 }
\ No newline at end of file
diff --git a/src/assets/locale/no/common.json b/src/assets/locale/no/common.json
index 6665280..b95ffce 100644
--- a/src/assets/locale/no/common.json
+++ b/src/assets/locale/no/common.json
@@ -112,5 +112,6 @@
         "older": "Eldre"
     },
     "pin": "Fest",
-    "unpin": "Løsne"
+    "unpin": "Løsne",
+    "generationInfo": "Generasjonsinformasjon"
 }
\ No newline at end of file
diff --git a/src/assets/locale/pt-BR/common.json b/src/assets/locale/pt-BR/common.json
index ba4f252..a2f1a41 100644
--- a/src/assets/locale/pt-BR/common.json
+++ b/src/assets/locale/pt-BR/common.json
@@ -111,5 +111,6 @@
     "older": "Mais Antigos"
   },
   "pin": "Fixar",
-  "unpin": "Desafixar"
+  "unpin": "Desafixar",
+  "generationInfo": "Informações de Geração"
 }
\ No newline at end of file
diff --git a/src/assets/locale/ru/common.json b/src/assets/locale/ru/common.json
index b619def..04b54de 100644
--- a/src/assets/locale/ru/common.json
+++ b/src/assets/locale/ru/common.json
@@ -111,5 +111,6 @@
         "older": "Ранее"
     },
     "pin": "Закрепить",
-    "unpin": "Открепить"
+    "unpin": "Открепить",
+    "generationInfo": "Информация о генерации"
 }
\ No newline at end of file
diff --git a/src/assets/locale/sv/common.json b/src/assets/locale/sv/common.json
index 3127267..b0df981 100644
--- a/src/assets/locale/sv/common.json
+++ b/src/assets/locale/sv/common.json
@@ -116,5 +116,6 @@
         "older": "Äldre"
     },
     "pin": "Fäst",
-    "unpin": "Ta bort fäst"
+    "unpin": "Ta bort fäst",
+    "generationInfo": "Generationsinformation"
 }
diff --git a/src/assets/locale/zh/common.json b/src/assets/locale/zh/common.json
index cd58e05..7fb9f9d 100644
--- a/src/assets/locale/zh/common.json
+++ b/src/assets/locale/zh/common.json
@@ -111,5 +111,6 @@
         "older": "更早"
     },
     "pin": "置顶",
-    "unpin": "取消置顶"
+    "unpin": "取消置顶",
+    "generationInfo": "生成信息"
 }
\ No newline at end of file
diff --git a/src/components/Common/Playground/GenerationInfo.tsx b/src/components/Common/Playground/GenerationInfo.tsx
new file mode 100644
index 0000000..4e34710
--- /dev/null
+++ b/src/components/Common/Playground/GenerationInfo.tsx
@@ -0,0 +1,65 @@
+type GenerationMetrics = {
+  total_duration?: number
+  load_duration?: number
+  prompt_eval_count?: number
+  prompt_eval_duration?: number
+  eval_count?: number
+  eval_duration?: number
+  context?: string
+  response?: string
+}
+
+type Props = {
+  generationInfo: GenerationMetrics
+}
+
+export const GenerationInfo = ({ generationInfo }: Props) => {
+  if (!generationInfo) return null
+
+  const calculateTokensPerSecond = (
+    evalCount?: number,
+    evalDuration?: number
+  ) => {
+    if (!evalCount || !evalDuration) return 0
+    return (evalCount / evalDuration) * 1e9
+  }
+
+  const formatDuration = (nanoseconds?: number) => {
+    if (!nanoseconds) return "0ms"
+    const ms = nanoseconds / 1e6
+    if (ms < 1) return `${ms.toFixed(3)}ms`
+    if (ms < 1000) return `${Math.round(ms)}ms`
+    return `${(ms / 1000).toFixed(2)}s`
+  }
+
+  const metricsToDisplay = {
+    ...generationInfo,
+    ...(generationInfo?.eval_count && generationInfo?.eval_duration
+      ? {
+          tokens_per_second: calculateTokensPerSecond(
+            generationInfo.eval_count,
+            generationInfo.eval_duration
+          ).toFixed(2)
+        }
+      : {})
+  }
+
+  return (
+    <div className="p-2 w-full">
+      <div className="flex flex-col gap-2">
+        {Object.entries(metricsToDisplay)
+          .filter(([key]) => key !== "model")
+          .map(([key, value]) => (
+            <div key={key} className="flex flex-wrap justify-between">
+              <div className="font-medium text-xs">{key}</div>
+              <div className="font-medium text-xs break-all">
+                {key.includes("duration")
+                  ? formatDuration(value as number)
+                  : String(value)}
+              </div>
+            </div>
+          ))}
+      </div>
+    </div>
+  )
+}
diff --git a/src/components/Common/Playground/Message.tsx b/src/components/Common/Playground/Message.tsx
index 0c6299c..925fe4c 100644
--- a/src/components/Common/Playground/Message.tsx
+++ b/src/components/Common/Playground/Message.tsx
@@ -1,10 +1,11 @@
 import Markdown from "../../Common/Markdown"
 import React from "react"
-import { Tag, Image, Tooltip, Collapse } from "antd"
+import { Tag, Image, Tooltip, Collapse, Popover } from "antd"
 import { WebSearch } from "./WebSearch"
 import {
   CheckIcon,
   ClipboardIcon,
+  InfoIcon,
   Pen,
   PlayIcon,
   RotateCcw,
@@ -16,6 +17,7 @@ import { MessageSource } from "./MessageSource"
 import { useTTS } from "@/hooks/useTTS"
 import { tagColors } from "@/utils/color"
 import { removeModelSuffix } from "@/db/models"
+import { GenerationInfo } from "./GenerationInfo"
 
 type Props = {
   message: string
@@ -37,6 +39,7 @@ type Props = {
   hideEditAndRegenerate?: boolean
   onSourceClick?: (source: any) => void
   isTTSEnabled?: boolean
+  generationInfo?: any
 }
 
 export const PlaygroundMessage = (props: Props) => {
@@ -206,6 +209,18 @@ export const PlaygroundMessage = (props: Props) => {
                       </Tooltip>
                     )}
 
+                    {props.generationInfo && (
+                      <Popover
+                        content={
+                         <GenerationInfo generationInfo={props.generationInfo} />
+                        }
+                        title={t("generationInfo")}>
+                        <button className="flex items-center justify-center w-6 h-6 rounded-full bg-gray-100 dark:bg-gray-800 hover:bg-gray-200 dark:hover:bg-gray-700 transition-colors duration-200 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-gray-500">
+                          <InfoIcon className="w-3 h-3 text-gray-400 group-hover:text-gray-500" />
+                        </button>
+                      </Popover>
+                    )}
+
                     {!props.hideEditAndRegenerate &&
                       props.currentMessageIndex === props.totalMessages - 1 && (
                         <Tooltip title={t("regenerate")}>
diff --git a/src/components/Option/Playground/PlaygroundChat.tsx b/src/components/Option/Playground/PlaygroundChat.tsx
index 5f2adfe..aaf3462 100644
--- a/src/components/Option/Playground/PlaygroundChat.tsx
+++ b/src/components/Option/Playground/PlaygroundChat.tsx
@@ -54,6 +54,7 @@ export const PlaygroundChat = () => {
               setIsSourceOpen(true)
             }}
             isTTSEnabled={ttsEnabled}
+            generationInfo={message?.generationInfo}
           />
         ))}
         {messages.length > 0 && (
diff --git a/src/components/Sidepanel/Chat/body.tsx b/src/components/Sidepanel/Chat/body.tsx
index 61d9071..f9f1407 100644
--- a/src/components/Sidepanel/Chat/body.tsx
+++ b/src/components/Sidepanel/Chat/body.tsx
@@ -47,6 +47,7 @@ export const SidePanelBody = () => {
             setIsSourceOpen(true)
           }}
           isTTSEnabled={ttsEnabled}
+          generationInfo={message?.generationInfo}
         />
       ))}
       <div className="w-full h-48 flex-shrink-0"></div>
diff --git a/src/db/index.ts b/src/db/index.ts
index 4a77553..68b3b02 100644
--- a/src/db/index.ts
+++ b/src/db/index.ts
@@ -33,6 +33,7 @@ type Message = {
   search?: WebSearch
   createdAt: number
   messageType?: string
+  generationInfo?: any
 }
 
 type Webshare = {
@@ -254,7 +255,8 @@ export const saveMessage = async (
   images: string[],
   source?: any[],
   time?: number,
-  message_type?: string
+  message_type?: string,
+  generationInfo?: any
 ) => {
   const id = generateID()
   let createdAt = Date.now()
@@ -270,7 +272,8 @@ export const saveMessage = async (
     images,
     createdAt,
     sources: source,
-    messageType: message_type
+    messageType: message_type,
+    generationInfo: generationInfo
   }
   const db = new PageAssitDatabase()
   await db.addMessage(message)
diff --git a/src/hooks/chat-helper/index.ts b/src/hooks/chat-helper/index.ts
index 18f89be..3f725c0 100644
--- a/src/hooks/chat-helper/index.ts
+++ b/src/hooks/chat-helper/index.ts
@@ -118,7 +118,7 @@ export const saveMessageOnSuccess = async ({
   fullText,
   source,
   message_source = "web-ui",
-  message_type
+  message_type, generationInfo
 }: {
   historyId: string | null
   setHistoryId: (historyId: string) => void
@@ -130,6 +130,7 @@ export const saveMessageOnSuccess = async ({
   source: any[]
   message_source?: "copilot" | "web-ui",
   message_type?: string
+  generationInfo?: any
 }) => {
   if (historyId) {
     if (!isRegenerate) {
@@ -141,7 +142,8 @@ export const saveMessageOnSuccess = async ({
         [image],
         [],
         1,
-        message_type
+        message_type,
+        generationInfo
       )
     }
     await saveMessage(
@@ -152,7 +154,8 @@ export const saveMessageOnSuccess = async ({
       [],
       source,
       2,
-      message_type
+      message_type,
+      generationInfo
     )
     await setLastUsedChatModel(historyId, selectedModel!)
   } else {
@@ -166,7 +169,8 @@ export const saveMessageOnSuccess = async ({
       [image],
       [],
       1,
-      message_type
+      message_type,
+      generationInfo
     )
     await saveMessage(
       newHistoryId.id,
@@ -176,7 +180,8 @@ export const saveMessageOnSuccess = async ({
       [],
       source,
       2,
-      message_type
+      message_type,
+      generationInfo
     )
     setHistoryId(newHistoryId.id)
     await setLastUsedChatModel(newHistoryId.id, selectedModel!)
diff --git a/src/hooks/useMessage.tsx b/src/hooks/useMessage.tsx
index c85324a..d9a17ac 100644
--- a/src/hooks/useMessage.tsx
+++ b/src/hooks/useMessage.tsx
@@ -328,10 +328,25 @@ export const useMessage = () => {
 
       const applicationChatHistory = generateHistory(history, selectedModel)
 
+      let generationInfo: any | undefined = undefined
+
       const chunks = await ollama.stream(
         [...applicationChatHistory, humanMessage],
         {
-          signal: signal
+          signal: signal,
+          callbacks: [
+            {
+              handleLLMEnd(
+                output: any,
+              ): any {
+                try {
+                  generationInfo = output?.generations?.[0][0]?.generationInfo
+                } catch (e) {
+                  console.log("handleLLMEnd error", e)
+                }
+              }
+            }
+          ]
         }
       )
       let count = 0
@@ -361,7 +376,8 @@ export const useMessage = () => {
             return {
               ...message,
               message: fullText,
-              sources: source
+              sources: source,
+              generationInfo
             }
           }
           return message
@@ -390,7 +406,8 @@ export const useMessage = () => {
         image,
         fullText,
         source,
-        message_source: "copilot"
+        message_source: "copilot",
+        generationInfo
       })
 
       setIsProcessing(false)
@@ -544,10 +561,25 @@ export const useMessage = () => {
         )
       }
 
+      let generationInfo: any | undefined = undefined
+
       const chunks = await ollama.stream(
         [...applicationChatHistory, humanMessage],
         {
-          signal: signal
+          signal: signal,
+          callbacks: [
+            {
+              handleLLMEnd(
+                output: any,
+              ): any {
+                try {
+                  generationInfo = output?.generations?.[0][0]?.generationInfo
+                } catch (e) {
+                  console.log("handleLLMEnd error", e)
+                }
+              }
+            }
+          ]
         }
       )
       let count = 0
@@ -576,7 +608,8 @@ export const useMessage = () => {
           if (message.id === generateMessageId) {
             return {
               ...message,
-              message: fullText
+              message: fullText,
+              generationInfo
             }
           }
           return message
@@ -605,7 +638,8 @@ export const useMessage = () => {
         image,
         fullText,
         source: [],
-        message_source: "copilot"
+        message_source: "copilot",
+        generationInfo
       })
 
       setIsProcessing(false)
@@ -789,10 +823,24 @@ export const useMessage = () => {
         )
       }
 
+      let generationInfo: any | undefined = undefined
       const chunks = await ollama.stream(
         [...applicationChatHistory, humanMessage],
         {
-          signal: signal
+          signal: signal,
+          callbacks: [
+            {
+              handleLLMEnd(
+                output: any,
+              ): any {
+                try {
+                  generationInfo = output?.generations?.[0][0]?.generationInfo
+                } catch (e) {
+                  console.log("handleLLMEnd error", e)
+                }
+              }
+            }
+          ]
         }
       )
       let count = 0
@@ -822,7 +870,8 @@ export const useMessage = () => {
             return {
               ...message,
               message: fullText,
-              sources: source
+              sources: source,
+              generationInfo
             }
           }
           return message
@@ -850,7 +899,8 @@ export const useMessage = () => {
         message,
         image,
         fullText,
-        source
+        source,
+        generationInfo
       })
 
       setIsProcessing(false)
@@ -982,8 +1032,23 @@ export const useMessage = () => {
         })
       }
 
+      let generationInfo: any | undefined = undefined
+
       const chunks = await ollama.stream([humanMessage], {
-        signal: signal
+        signal: signal,
+        callbacks: [
+          {
+            handleLLMEnd(
+              output: any,
+            ): any {
+              try {
+                generationInfo = output?.generations?.[0][0]?.generationInfo
+              } catch (e) {
+                console.log("handleLLMEnd error", e)
+              }
+            }
+          }
+        ]
       })
       let count = 0
       for await (const chunk of chunks) {
@@ -1011,7 +1076,8 @@ export const useMessage = () => {
           if (message.id === generateMessageId) {
             return {
               ...message,
-              message: fullText
+              message: fullText,
+              generationInfo
             }
           }
           return message
@@ -1042,7 +1108,8 @@ export const useMessage = () => {
         fullText,
         source: [],
         message_source: "copilot",
-        message_type: messageType
+        message_type: messageType,
+        generationInfo
       })
 
       setIsProcessing(false)
diff --git a/src/hooks/useMessageOption.tsx b/src/hooks/useMessageOption.tsx
index 58d56bf..1cccb62 100644
--- a/src/hooks/useMessageOption.tsx
+++ b/src/hooks/useMessageOption.tsx
@@ -243,10 +243,23 @@ export const useMessageOption = () => {
         )
       }
 
+      let generationInfo: any | undefined = undefined
+
       const chunks = await ollama.stream(
         [...applicationChatHistory, humanMessage],
         {
-          signal: signal
+          signal: signal,
+          callbacks: [
+            {
+              handleLLMEnd(output: any): any {
+                try {
+                  generationInfo = output?.generations?.[0][0]?.generationInfo
+                } catch (e) {
+                  console.log("handleLLMEnd error", e)
+                }
+              }
+            }
+          ]
         }
       )
       let count = 0
@@ -276,7 +289,8 @@ export const useMessageOption = () => {
             return {
               ...message,
               message: fullText,
-              sources: source
+              sources: source,
+              generationInfo
             }
           }
           return message
@@ -304,7 +318,8 @@ export const useMessageOption = () => {
         message,
         image,
         fullText,
-        source
+        source,
+        generationInfo
       })
 
       setIsProcessing(false)
@@ -465,10 +480,23 @@ export const useMessageOption = () => {
         )
       }
 
+      let generationInfo: any | undefined = undefined
+
       const chunks = await ollama.stream(
         [...applicationChatHistory, humanMessage],
         {
-          signal: signal
+          signal: signal,
+          callbacks: [
+            {
+              handleLLMEnd(output: any): any {
+                try {
+                  generationInfo = output?.generations?.[0][0]?.generationInfo
+                } catch (e) {
+                  console.log("handleLLMEnd error", e)
+                }
+              }
+            }
+          ],
         }
       )
 
@@ -498,7 +526,8 @@ export const useMessageOption = () => {
           if (message.id === generateMessageId) {
             return {
               ...message,
-              message: fullText
+              message: fullText,
+              generationInfo
             }
           }
           return message
@@ -526,7 +555,8 @@ export const useMessageOption = () => {
         message,
         image,
         fullText,
-        source: []
+        source: [],
+        generationInfo
       })
 
       setIsProcessing(false)
@@ -711,10 +741,23 @@ export const useMessageOption = () => {
 
       const applicationChatHistory = generateHistory(history, selectedModel)
 
+      let generationInfo: any | undefined = undefined
+
       const chunks = await ollama.stream(
         [...applicationChatHistory, humanMessage],
         {
-          signal: signal
+          signal: signal,
+          callbacks: [
+            {
+              handleLLMEnd(output: any): any {
+                try {
+                  generationInfo = output?.generations?.[0][0]?.generationInfo
+                } catch (e) {
+                  console.log("handleLLMEnd error", e)
+                }
+              }
+            }
+          ]
         }
       )
       let count = 0
@@ -744,7 +787,8 @@ export const useMessageOption = () => {
             return {
               ...message,
               message: fullText,
-              sources: source
+              sources: source,
+              generationInfo
             }
           }
           return message
@@ -772,7 +816,8 @@ export const useMessageOption = () => {
         message,
         image,
         fullText,
-        source
+        source,
+        generationInfo
       })
 
       setIsProcessing(false)
diff --git a/src/models/index.ts b/src/models/index.ts
index d459e66..02f2ce8 100644
--- a/src/models/index.ts
+++ b/src/models/index.ts
@@ -49,7 +49,7 @@ export const pageAssistModel = async ({
       configuration: {
         apiKey: providerInfo.apiKey || "temp",
         baseURL: providerInfo.baseUrl || "",
-      }
+      },
     }) as any
   }
 
diff --git a/src/types/message.ts b/src/types/message.ts
index 3be7cdc..1ec1d5a 100644
--- a/src/types/message.ts
+++ b/src/types/message.ts
@@ -1,19 +1,20 @@
 type WebSearch = {
-    search_engine: string
-    search_url: string
-    search_query: string
-    search_results: {
-      title: string
-      link: string
-    }[]
-  }
-  export type Message = {
-    isBot: boolean
-    name: string
-    message: string
-    sources: any[]
-    images?: string[]
-    search?: WebSearch
-    messageType?: string
-    id?: string
-  }
\ No newline at end of file
+  search_engine: string
+  search_url: string
+  search_query: string
+  search_results: {
+    title: string
+    link: string
+  }[]
+}
+export type Message = {
+  isBot: boolean
+  name: string
+  message: string
+  sources: any[]
+  images?: string[]
+  search?: WebSearch
+  messageType?: string
+  id?: string
+  generationInfo?: any
+}
\ No newline at end of file
diff --git a/wxt.config.ts b/wxt.config.ts
index c870e8e..288a555 100644
--- a/wxt.config.ts
+++ b/wxt.config.ts
@@ -50,7 +50,7 @@ export default defineConfig({
   outDir: "build",
 
   manifest: {
-    version: "1.3.3",
+    version: "1.3.4",
     name:
       process.env.TARGET === "firefox"
         ? "Page Assist - A Web UI for Local AI Models"

From 7c805cfe22661f7dd0673a75297d896b0a596086 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 9 Nov 2024 15:20:16 +0530
Subject: [PATCH 03/27] feat: Include generation info in message history

Adds the `generationInfo` field to the message history output to provide more context about the message's origin. This will be helpful for debugging and understanding the provenance of messages.
---
 src/db/index.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/db/index.ts b/src/db/index.ts
index 68b3b02..e4c136c 100644
--- a/src/db/index.ts
+++ b/src/db/index.ts
@@ -301,7 +301,8 @@ export const formatToMessage = (messages: MessageHistory): MessageType[] => {
       message: message.content,
       name: message.name,
       sources: message?.sources || [],
-      images: message.images || []
+      images: message.images || [],
+      generationInfo: message?.generationInfo,
     }
   })
 }

From fd654cafdb8e31dd91d19b5c17e3141483e7f9ec Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 9 Nov 2024 16:56:47 +0530
Subject: [PATCH 04/27] feat: Add max tokens setting for model generations

Adds a new setting to control the maximum number of tokens generated by the model. This provides more control over the length of responses and can be useful for limiting the amount of text generated in certain situations.
---
 src/assets/locale/da/common.json              |  4 ++++
 src/assets/locale/de/common.json              |  4 ++++
 src/assets/locale/en/common.json              |  4 ++++
 src/assets/locale/es/common.json              |  4 ++++
 src/assets/locale/fa/common.json              |  4 ++++
 src/assets/locale/fr/common.json              |  4 ++++
 src/assets/locale/it/common.json              |  4 ++++
 src/assets/locale/ja-JP/common.json           |  4 ++++
 src/assets/locale/ko/common.json              |  5 +++-
 src/assets/locale/ml/common.json              |  4 ++++
 src/assets/locale/no/common.json              |  4 ++++
 src/assets/locale/pt-BR/common.json           |  4 ++++
 src/assets/locale/ru/common.json              |  4 ++++
 src/assets/locale/sv/common.json              |  4 ++++
 src/assets/locale/zh/common.json              |  4 ++++
 .../Settings/CurrentChatModelSettings.tsx     | 13 +++++++++-
 .../Option/Settings/model-settings.tsx        |  9 ++++++-
 src/hooks/useMessage.tsx                      | 24 ++++++++++++++-----
 src/hooks/useMessageOption.tsx                | 19 +++++++++++----
 src/models/index.ts                           | 10 +++++---
 20 files changed, 119 insertions(+), 17 deletions(-)

diff --git a/src/assets/locale/da/common.json b/src/assets/locale/da/common.json
index 881dab7..de82642 100644
--- a/src/assets/locale/da/common.json
+++ b/src/assets/locale/da/common.json
@@ -70,6 +70,10 @@
                 "label": "Længden af Kontekst",
                 "placeholder": "Instast Længden af Kontekst værdien (standard: 2048)"
             },
+            "numPredict": {
+                "label": "Maks Tokens (num_predict)",
+                "placeholder": "Indtast Maks Tokens værdi (fx. 2048, 4096)"
+            },
             "seed": {
                 "label": "Seed",
                 "placeholder": "Indtast Seed værdi (fx. 1234)",
diff --git a/src/assets/locale/de/common.json b/src/assets/locale/de/common.json
index 9a69b22..65b72ab 100644
--- a/src/assets/locale/de/common.json
+++ b/src/assets/locale/de/common.json
@@ -70,6 +70,10 @@
                 "label": "Anzahl der Kontexte",
                 "placeholder": "Geben Sie die Anzahl der Kontexte ein (Standard: 2048)"
             },
+            "numPredict": {
+                "label": "Max Tokens (num_predict)",
+                "placeholder": "Geben Sie den Max-Tokens-Wert ein (z.B. 2048, 4096)"
+            },
             "seed": {
                 "label": "Seed",
                 "placeholder": "Geben Sie den Seed-Wert ein (z.B. 1234)",
diff --git a/src/assets/locale/en/common.json b/src/assets/locale/en/common.json
index a7a3ecc..d033144 100644
--- a/src/assets/locale/en/common.json
+++ b/src/assets/locale/en/common.json
@@ -70,6 +70,10 @@
                 "label": "Number of Contexts",
                 "placeholder": "Enter Number of Contexts value (default: 2048)"
             },
+            "numPredict": {
+                "label": "Max Tokens (num_predict)",
+                "placeholder": "Enter Max Tokens value (e.g. 2048, 4096)"
+            },
             "seed": {
                 "label": "Seed",
                 "placeholder": "Enter Seed value (e.g. 1234)",
diff --git a/src/assets/locale/es/common.json b/src/assets/locale/es/common.json
index cf5ccf8..210e043 100644
--- a/src/assets/locale/es/common.json
+++ b/src/assets/locale/es/common.json
@@ -70,6 +70,10 @@
                 "label": "Cantidad de contextos",
                 "placeholder": "Ingresar el valor de tamaño de la ventana de contexto (por defecto: 2048)"
             },
+            "numPredict": {
+                "label": "Máximo de Tokens (num_predict)",
+                "placeholder": "Ingrese el valor máximo de Tokens (ej: 2048, 4096)"
+            },
             "seed": {
                 "label": "Semilla",
                 "placeholder": "Ingresar el valor de la semilla (ej: 1234)",
diff --git a/src/assets/locale/fa/common.json b/src/assets/locale/fa/common.json
index 6abb70a..72301c7 100644
--- a/src/assets/locale/fa/common.json
+++ b/src/assets/locale/fa/common.json
@@ -70,6 +70,10 @@
                 "label": "Number of Contexts",
                 "placeholder": "مقدار Number of Contexts را وارد کنید (پیش فرض: 2048)"
             },
+            "numPredict": {
+                "label": "حداکثر توکن‌ها (num_predict)",
+                "placeholder": "مقدار حداکثر توکن‌ها را وارد کنید (مثلا 2048، 4096)"
+            },
             "seed": {
                 "label": "Seed",
                 "placeholder": "مقدار Seed را وارد کنید (e.g. 1234)",
diff --git a/src/assets/locale/fr/common.json b/src/assets/locale/fr/common.json
index 0775303..d11ef8c 100644
--- a/src/assets/locale/fr/common.json
+++ b/src/assets/locale/fr/common.json
@@ -70,6 +70,10 @@
                 "label": "Nombre de contextes",
                 "placeholder": "Entrez la valeur du nombre de contextes (par défaut: 2048)"
             },
+            "numPredict": {
+                "label": "Tokens maximum (num_predict)",
+                "placeholder": "Entrez la valeur des tokens maximum (par exemple 2048, 4096)"
+            },
             "seed": {
                 "label": "Graine",
                 "placeholder": "Entrez la valeur des semences (par exemple 1234)",
diff --git a/src/assets/locale/it/common.json b/src/assets/locale/it/common.json
index 102b6c8..6b08509 100644
--- a/src/assets/locale/it/common.json
+++ b/src/assets/locale/it/common.json
@@ -70,6 +70,10 @@
                 "label": "Dimensione del Contesto",
                 "placeholder": "Inserisci la Dimensione del Contesto (default: 2048)"
             },
+            "numPredict": {
+                "label": "Token Massimi (num_predict)",
+                "placeholder": "Inserisci il valore dei Token Massimi (es. 2048, 4096)"
+            },
             "seed": {
                 "label": "Seed",
                 "placeholder": "Inserisci il Valore Seed (e.g. 1234)",
diff --git a/src/assets/locale/ja-JP/common.json b/src/assets/locale/ja-JP/common.json
index 6f38578..92ba8e0 100644
--- a/src/assets/locale/ja-JP/common.json
+++ b/src/assets/locale/ja-JP/common.json
@@ -70,6 +70,10 @@
         "label": "コンテキストの数",
         "placeholder": "コンテキスト数を入力してください（デフォルト：2048）"
       },
+      "numPredict": {
+        "label": "最大トークン数 (num_predict)",
+        "placeholder": "最大トークン数を入力してください（例：2048、4096）"
+      },
       "seed": {
         "label": "シード",
         "placeholder": "シード値を入力してください（例：1234）",
diff --git a/src/assets/locale/ko/common.json b/src/assets/locale/ko/common.json
index e59e785..ed9e0b9 100644
--- a/src/assets/locale/ko/common.json
+++ b/src/assets/locale/ko/common.json
@@ -70,7 +70,10 @@
         "label": "컨텍스트 수",
         "placeholder": "컨텍스트 수를 입력하세요 (기본값: 2048)"
       },
-      "seed": {
+      "numPredict": {
+        "label": "최대 토큰 수 (num_predict)",
+        "placeholder": "최대 토큰 수를 입력하세요 (예: 2048, 4096)"
+      },      "seed": {
         "label": "시드",
         "placeholder": "시드 값을 입력하세요 (예: 1234)",
         "help": "모델 출력의 재현성"
diff --git a/src/assets/locale/ml/common.json b/src/assets/locale/ml/common.json
index ed6988a..cd6868d 100644
--- a/src/assets/locale/ml/common.json
+++ b/src/assets/locale/ml/common.json
@@ -69,6 +69,10 @@
                 "label": "സന്ദർഭങ്ങളുടെ എണ്ണം",
                 "placeholder": "സന്ദർഭങ്ങളുടെ സംഖ്യ നൽകുക (സ്ഥിരം: 2048)"
             },
+            "numPredict": {
+                "label": "പരമാവധി ടോക്കണുകൾ (num_predict)",
+                "placeholder": "പരമാവധി ടോക്കൺ മൂല്യം നൽകുക (ഉദാ: 2048, 4096)"
+            },
             "seed": {
                 "label": "സീഡ്",
                 "placeholder": "സീഡ് വില്യമ നൽകുക (ഉദാ: 1234)",
diff --git a/src/assets/locale/no/common.json b/src/assets/locale/no/common.json
index b95ffce..1720d1f 100644
--- a/src/assets/locale/no/common.json
+++ b/src/assets/locale/no/common.json
@@ -70,6 +70,10 @@
                 "label": "Kontekstlengde",
                 "placeholder": "Skriv inn kontekstlengdeverdi (standard: 2048)"
             },
+            "numPredict": {
+                "label": "Maks Tokens (num_predict)",
+                "placeholder": "Skriv inn Maks Tokens-verdi (f.eks. 2048, 4096)"
+            },
             "seed": {
                 "label": "Seed",
                 "placeholder": "Skriv inn seedverdi (f.eks. 1234)",
diff --git a/src/assets/locale/pt-BR/common.json b/src/assets/locale/pt-BR/common.json
index a2f1a41..660514d 100644
--- a/src/assets/locale/pt-BR/common.json
+++ b/src/assets/locale/pt-BR/common.json
@@ -70,6 +70,10 @@
         "label": "Número de Contextos",
         "placeholder": "Digite o valor do Número de Contextos (padrão: 2048)"
       },
+      "numPredict": {
+        "label": "Máximo de Tokens (num_predict)",
+        "placeholder": "Digite o valor do Máximo de Tokens (ex: 2048, 4096)"
+      },
       "seed": {
         "label": "Semente",
         "placeholder": "Digite o valor da Semente (ex: 1234)",
diff --git a/src/assets/locale/ru/common.json b/src/assets/locale/ru/common.json
index 04b54de..31291a0 100644
--- a/src/assets/locale/ru/common.json
+++ b/src/assets/locale/ru/common.json
@@ -70,6 +70,10 @@
                 "label": "Количество контекстов",
                 "placeholder": "Введите значение количества контекстов (по умолчанию: 2048)"
             },
+            "numPredict": {
+                "label": "Максимальное количество токенов (num_predict)",
+                "placeholder": "Введите значение максимального количества токенов (например, 2048, 4096)"
+            },
             "seed": {
                 "label": "Сид",
                 "placeholder": "Введите значение сида (например, 1234)",
diff --git a/src/assets/locale/sv/common.json b/src/assets/locale/sv/common.json
index b0df981..4698dc7 100644
--- a/src/assets/locale/sv/common.json
+++ b/src/assets/locale/sv/common.json
@@ -70,6 +70,10 @@
                 "label": "Antal kontexter",
                 "placeholder": "Ange antal kontextvärden (standard: 2048)"
             },
+            "numPredict": {
+                "label": "Max antal tokens (num_predict)",
+                "placeholder": "Ange Max antal tokens värde (t.ex. 2048, 4096)"
+            },
             "seed": {
                 "label": "Frö",
                 "placeholder": "Ange frövärde (t.ex. 1234)",
diff --git a/src/assets/locale/zh/common.json b/src/assets/locale/zh/common.json
index 7fb9f9d..080776a 100644
--- a/src/assets/locale/zh/common.json
+++ b/src/assets/locale/zh/common.json
@@ -70,6 +70,10 @@
                 "label": "上下文数量",
                 "placeholder": "输入上下文数量（默认：2048）"
             },
+            "numPredict": {
+                "label": "最大令牌数 (num_predict)",
+                "placeholder": "输入最大令牌数（例如：2048、4096）"
+            },
             "seed": {
                 "label": "随机种子",
                 "placeholder": "输入随机种子值（例如：1234）",
diff --git a/src/components/Common/Settings/CurrentChatModelSettings.tsx b/src/components/Common/Settings/CurrentChatModelSettings.tsx
index 3e5c74f..cb37286 100644
--- a/src/components/Common/Settings/CurrentChatModelSettings.tsx
+++ b/src/components/Common/Settings/CurrentChatModelSettings.tsx
@@ -39,12 +39,14 @@ export const CurrentChatModelSettings = ({
         numCtx: cUserSettings.numCtx ?? data.numCtx,
         seed: cUserSettings.seed,
         numGpu: cUserSettings.numGpu ?? data.numGpu,
+        numPredict: cUserSettings.numPredict ?? data.numPredict,
         systemPrompt: cUserSettings.systemPrompt ?? ""
       })
       return data
     },
     enabled: open,
-    refetchOnMount: true
+    refetchOnMount: false,
+    refetchOnWindowFocus: false
   })
 
   const renderBody = () => {
@@ -115,6 +117,15 @@ export const CurrentChatModelSettings = ({
               />
             </Form.Item>
 
+            <Form.Item
+              name="numPredict"
+              label={t("modelSettings.form.numPredict.label")}>
+              <InputNumber
+                style={{ width: "100%" }}
+                placeholder={t("modelSettings.form.numPredict.placeholder")}
+              />
+            </Form.Item>
+
             <Collapse
               ghost
               className="border-none bg-transparent"
diff --git a/src/components/Option/Settings/model-settings.tsx b/src/components/Option/Settings/model-settings.tsx
index b91dda5..40523f7 100644
--- a/src/components/Option/Settings/model-settings.tsx
+++ b/src/components/Option/Settings/model-settings.tsx
@@ -77,7 +77,14 @@ export const ModelSettings = () => {
               size="large"
             />
           </Form.Item>
-
+          <Form.Item
+              name="numPredict"
+              label={t("modelSettings.form.numPredict.label")}>
+              <InputNumber
+                style={{ width: "100%" }}
+                placeholder={t("modelSettings.form.numPredict.placeholder")}
+              />
+            </Form.Item>
           <Collapse
             ghost
             className="border-none bg-transparent"
diff --git a/src/hooks/useMessage.tsx b/src/hooks/useMessage.tsx
index d9a17ac..cc38949 100644
--- a/src/hooks/useMessage.tsx
+++ b/src/hooks/useMessage.tsx
@@ -133,7 +133,9 @@ export const useMessage = () => {
         currentChatModelSettings?.numCtx ?? userDefaultModelSettings?.numCtx,
       seed: currentChatModelSettings?.seed,
       numGpu:
-        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+
     })
 
     let newMessage: Message[] = []
@@ -261,7 +263,9 @@ export const useMessage = () => {
             userDefaultModelSettings?.numCtx,
           seed: currentChatModelSettings?.seed,
           numGpu:
-            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+
         })
         const response = await questionOllama.invoke(promptForQuestion)
         query = response.content.toString()
@@ -475,7 +479,9 @@ export const useMessage = () => {
         currentChatModelSettings?.numCtx ?? userDefaultModelSettings?.numCtx,
       seed: currentChatModelSettings?.seed,
       numGpu:
-        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+
     })
 
     let newMessage: Message[] = []
@@ -702,7 +708,9 @@ export const useMessage = () => {
         currentChatModelSettings?.numCtx ?? userDefaultModelSettings?.numCtx,
       seed: currentChatModelSettings?.seed,
       numGpu:
-        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+
     })
 
     let newMessage: Message[] = []
@@ -777,7 +785,9 @@ export const useMessage = () => {
             userDefaultModelSettings?.numCtx,
           seed: currentChatModelSettings?.seed,
           numGpu:
-            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+
         })
         const response = await questionOllama.invoke(promptForQuestion)
         query = response.content.toString()
@@ -964,7 +974,9 @@ export const useMessage = () => {
         currentChatModelSettings?.numCtx ?? userDefaultModelSettings?.numCtx,
       seed: currentChatModelSettings?.seed,
       numGpu:
-        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+
     })
 
     let newMessage: Message[] = []
diff --git a/src/hooks/useMessageOption.tsx b/src/hooks/useMessageOption.tsx
index 1cccb62..959cb44 100644
--- a/src/hooks/useMessageOption.tsx
+++ b/src/hooks/useMessageOption.tsx
@@ -122,7 +122,9 @@ export const useMessageOption = () => {
         currentChatModelSettings?.numCtx ?? userDefaultModelSettings?.numCtx,
       seed: currentChatModelSettings?.seed,
       numGpu:
-        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+
     })
 
     let newMessage: Message[] = []
@@ -197,7 +199,9 @@ export const useMessageOption = () => {
             userDefaultModelSettings?.numCtx,
           seed: currentChatModelSettings?.seed,
           numGpu:
-            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+
         })
         const response = await questionOllama.invoke(promptForQuestion)
         query = response.content.toString()
@@ -381,7 +385,8 @@ export const useMessageOption = () => {
         currentChatModelSettings?.numCtx ?? userDefaultModelSettings?.numCtx,
       seed: currentChatModelSettings?.seed,
       numGpu:
-        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
     })
 
     let newMessage: Message[] = []
@@ -616,7 +621,9 @@ export const useMessageOption = () => {
         currentChatModelSettings?.numCtx ?? userDefaultModelSettings?.numCtx,
       seed: currentChatModelSettings?.seed,
       numGpu:
-        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+
     })
 
     let newMessage: Message[] = []
@@ -707,7 +714,9 @@ export const useMessageOption = () => {
             userDefaultModelSettings?.numCtx,
           seed: currentChatModelSettings?.seed,
           numGpu:
-            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu
+            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+
         })
         const response = await questionOllama.invoke(promptForQuestion)
         query = response.content.toString()
diff --git a/src/models/index.ts b/src/models/index.ts
index 02f2ce8..4798f78 100644
--- a/src/models/index.ts
+++ b/src/models/index.ts
@@ -13,7 +13,8 @@ export const pageAssistModel = async ({
   topP,
   numCtx,
   seed,
-  numGpu
+  numGpu,
+  numPredict,
 }: {
   model: string
   baseUrl: string
@@ -24,12 +25,13 @@ export const pageAssistModel = async ({
   numCtx?: number
   seed?: number
   numGpu?: number
+  numPredict?: number
 }) => {
 
   if (model === "chrome::gemini-nano::page-assist") {
     return new ChatChromeAI({
       temperature,
-      topK
+      topK,
     })
   }
 
@@ -46,6 +48,7 @@ export const pageAssistModel = async ({
       openAIApiKey: providerInfo.apiKey || "temp",
       temperature,
       topP,
+      maxTokens: numPredict,
       configuration: {
         apiKey: providerInfo.apiKey || "temp",
         baseURL: providerInfo.baseUrl || "",
@@ -64,7 +67,8 @@ export const pageAssistModel = async ({
     numCtx,
     seed,
     model,
-    numGpu
+    numGpu,
+    numPredict
   })
 
 

From 977723f71f7cf8ea005f2a777b582b5909fad6a3 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 9 Nov 2024 17:13:23 +0530
Subject: [PATCH 05/27] feat: Ability to send image without text

---
 src/components/Option/Playground/PlaygroundForm.tsx | 8 +++++---
 src/components/Sidepanel/Chat/form.tsx              | 8 +++++---
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/components/Option/Playground/PlaygroundForm.tsx b/src/components/Option/Playground/PlaygroundForm.tsx
index be312e2..4e0d637 100644
--- a/src/components/Option/Playground/PlaygroundForm.tsx
+++ b/src/components/Option/Playground/PlaygroundForm.tsx
@@ -159,7 +159,7 @@ export const PlaygroundForm = ({ dropedFile }: Props) => {
       e.preventDefault()
       stopListening()
       form.onSubmit(async (value) => {
-        if (value.message.trim().length === 0) {
+        if (value.message.trim().length === 0 && value.image.length === 0) {
           return
         }
         if (!selectedModel || selectedModel.length === 0) {
@@ -228,6 +228,9 @@ export const PlaygroundForm = ({ dropedFile }: Props) => {
                   return
                 }
               }
+              if (value.message.trim().length === 0 && value.image.length === 0) {
+                return
+              }
               form.reset()
               textAreaFocus()
               await sendMessage({
@@ -261,7 +264,6 @@ export const PlaygroundForm = ({ dropedFile }: Props) => {
                 onKeyDown={(e) => handleKeyDown(e)}
                 ref={textareaRef}
                 className="px-2 py-2 w-full resize-none bg-transparent focus-within:outline-none focus:ring-0 focus-visible:ring-0 ring-0 dark:ring-0 border-0 dark:text-gray-100"
-                required
                 onPaste={handlePaste}
                 rows={1}
                 style={{ minHeight: "40px" }}
@@ -409,4 +411,4 @@ export const PlaygroundForm = ({ dropedFile }: Props) => {
       </div>
     </div>
   )
-}
+}
\ No newline at end of file
diff --git a/src/components/Sidepanel/Chat/form.tsx b/src/components/Sidepanel/Chat/form.tsx
index e1cc3f0..618431d 100644
--- a/src/components/Sidepanel/Chat/form.tsx
+++ b/src/components/Sidepanel/Chat/form.tsx
@@ -75,10 +75,10 @@ export const SidepanelForm = ({ dropedFile }: Props) => {
     ) {
       e.preventDefault()
       form.onSubmit(async (value) => {
-        await stopListening()
-        if (value.message.trim().length === 0) {
+        if (value.message.trim().length === 0 && value.image.length === 0) {
           return
         }
+        await stopListening()
         if (!selectedModel || selectedModel.length === 0) {
           form.setFieldError("message", t("formError.noModel"))
           return
@@ -237,6 +237,9 @@ export const SidepanelForm = ({ dropedFile }: Props) => {
                 }
               }
               await stopListening()
+              if (value.message.trim().length === 0 && value.image.length === 0) {
+                return
+              }
               form.reset()
               textAreaFocus()
               await sendMessage({
@@ -260,7 +263,6 @@ export const SidepanelForm = ({ dropedFile }: Props) => {
                 onKeyDown={(e) => handleKeyDown(e)}
                 ref={textareaRef}
                 className="px-2 py-2 w-full resize-none bg-transparent focus-within:outline-none focus:ring-0 focus-visible:ring-0 ring-0 dark:ring-0 border-0 dark:text-gray-100"
-                required
                 onPaste={handlePaste}
                 rows={1}
                 style={{ minHeight: "60px" }}

From 88d0cb68ae6834c4bd4d68b6ed822359de3ea86d Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 9 Nov 2024 17:58:23 +0530
Subject: [PATCH 06/27] feat: Support for new AI capabilities

Adds support for the new AI capabilities in Chrome. This change includes updated logic for checking availability and creating text sessions.
---
 src/models/utils/chrome.ts | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/models/utils/chrome.ts b/src/models/utils/chrome.ts
index 16e0b8b..3d42686 100644
--- a/src/models/utils/chrome.ts
+++ b/src/models/utils/chrome.ts
@@ -3,13 +3,19 @@ export const checkChromeAIAvailability = async (): Promise<"readily" | "no" | "a
     try {
         const ai = (window as any).ai;
 
-        // upcoming version change 
+        // latest i guess
+        if (ai?.languageModel?.capabilities) {
+            const capabilities = await ai.languageModel.capabilities();
+            return capabilities?.available ?? "no";
+        }
+
+        // old version change 
         if (ai?.assistant?.capabilities) {
             const capabilities = await ai.assistant.capabilities();
             return capabilities?.available ?? "no";
         }
 
-        // old version 
+        // too old version 
         if (ai?.canCreateTextSession) {
             const available = await ai.canCreateTextSession();
             return available ?? "no";
@@ -33,7 +39,15 @@ export interface AITextSession {
 export const createAITextSession = async (data: any): Promise<AITextSession> => {
     const ai = (window as any).ai;
 
-    // upcoming version change 
+    // new version i guess
+    if (ai?.languageModel?.create) {
+        const session = await ai.languageModel.create({
+            ...data
+        })
+        return session
+    }
+
+    // old version change 
     if (ai?.assistant?.create) {
         const session = await ai.assistant.create({
             ...data
@@ -41,7 +55,7 @@ export const createAITextSession = async (data: any): Promise<AITextSession> =>
         return session
     }
 
-    // old version
+    // too old version
     if (ai.createTextSession) {
         const session = await ai.createTextSession({
             ...data

From 8fbdfc35d3a45c3b0d63d8f94b225bafc58add3f Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 9 Nov 2024 18:09:47 +0530
Subject: [PATCH 07/27] feat(settings): Use selected system prompt in current
 chat model temp system fallback

Adds support for using the currently selected system prompt in the current model settings. This allows users to fine-tune their chat experience based on their preferred prompt.
---
 .../Common/Settings/CurrentChatModelSettings.tsx  | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/components/Common/Settings/CurrentChatModelSettings.tsx b/src/components/Common/Settings/CurrentChatModelSettings.tsx
index cb37286..e950bd9 100644
--- a/src/components/Common/Settings/CurrentChatModelSettings.tsx
+++ b/src/components/Common/Settings/CurrentChatModelSettings.tsx
@@ -1,3 +1,5 @@
+import { getPromptById } from "@/db"
+import { useMessageOption } from "@/hooks/useMessageOption"
 import { getAllModelSettings } from "@/services/model-settings"
 import { useStoreChatModelSettings } from "@/store/model"
 import { useQuery } from "@tanstack/react-query"
@@ -27,10 +29,20 @@ export const CurrentChatModelSettings = ({
   const { t } = useTranslation("common")
   const [form] = Form.useForm()
   const cUserSettings = useStoreChatModelSettings()
+  const { selectedSystemPrompt } = useMessageOption()
   const { isPending: isLoading } = useQuery({
     queryKey: ["fetchModelConfig2", open],
     queryFn: async () => {
       const data = await getAllModelSettings()
+
+      let tempSystemPrompt = "";
+
+      // i hate this method but i need this feature so badly that i need to do this
+      if (selectedSystemPrompt) {
+        const prompt = await getPromptById(selectedSystemPrompt)
+        tempSystemPrompt = prompt?.content ?? ""
+      }
+
       form.setFieldsValue({
         temperature: cUserSettings.temperature ?? data.temperature,
         topK: cUserSettings.topK ?? data.topK,
@@ -40,7 +52,7 @@ export const CurrentChatModelSettings = ({
         seed: cUserSettings.seed,
         numGpu: cUserSettings.numGpu ?? data.numGpu,
         numPredict: cUserSettings.numPredict ?? data.numPredict,
-        systemPrompt: cUserSettings.systemPrompt ?? ""
+        systemPrompt: cUserSettings.systemPrompt ?? tempSystemPrompt
       })
       return data
     },
@@ -49,6 +61,7 @@ export const CurrentChatModelSettings = ({
     refetchOnWindowFocus: false
   })
 
+
   const renderBody = () => {
     return (
       <>

From f8791a0707814a0ca2ee3e7babcb1c4ab6435515 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 9 Nov 2024 19:10:34 +0530
Subject: [PATCH 08/27] feat: Introduce temporary chat mode

Adds a new "Temporary Chat" mode for quick, non-persistent conversations. The new mode is available in the header bar and will trigger a visually distinct chat experience with a temporary background color. Temporary chats do not save to the chat history and are meant for short, one-off interactions. This feature enhances flexibility and provides a more convenient option for users who need to quickly interact with the AI without committing the conversation to their history.
---
 src/assets/locale/da/option.json              |  3 +-
 src/assets/locale/de/option.json              |  3 +-
 src/assets/locale/en/option.json              |  3 +-
 src/assets/locale/es/option.json              |  3 +-
 src/assets/locale/fa/option.json              |  3 +-
 src/assets/locale/fr/option.json              |  3 +-
 src/assets/locale/it/option.json              |  3 +-
 src/assets/locale/ja-JP/option.json           |  3 +-
 src/assets/locale/ko/option.json              |  3 +-
 src/assets/locale/ml/option.json              |  3 +-
 src/assets/locale/no/option.json              |  3 +-
 src/assets/locale/pt-BR/option.json           |  3 +-
 src/assets/locale/ru/option.json              |  3 +-
 src/assets/locale/sv/option.json              |  3 +-
 src/assets/locale/zh/option.json              |  3 +-
 src/components/Layouts/Header.tsx             | 20 +++---
 src/components/Layouts/NewChat.tsx            | 57 +++++++++++++++++
 .../Option/Playground/PlaygroundForm.tsx      | 21 ++++---
 src/components/Option/Sidebar.tsx             |  7 ++-
 src/hooks/useMessageOption.tsx                | 61 ++++++++++++++-----
 src/store/option.tsx                          |  8 ++-
 21 files changed, 167 insertions(+), 52 deletions(-)
 create mode 100644 src/components/Layouts/NewChat.tsx

diff --git a/src/assets/locale/da/option.json b/src/assets/locale/da/option.json
index 4548680..5fc0281 100644
--- a/src/assets/locale/da/option.json
+++ b/src/assets/locale/da/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "Noget gik galt",
     "validationSelectModel": "Venligst vælg en model for at forsæætte",
     "deleteHistoryConfirmation": "Er du sikker på at du vil slette denne historik?",
-    "editHistoryTitle": "Indtast en ny titel"
+    "editHistoryTitle": "Indtast en ny titel",
+    "temporaryChat": "Midlertidig Chat"
 }
\ No newline at end of file
diff --git a/src/assets/locale/de/option.json b/src/assets/locale/de/option.json
index f6ec7f9..4303931 100644
--- a/src/assets/locale/de/option.json
+++ b/src/assets/locale/de/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "Etwas ist schiefgelaufen",
     "validationSelectModel": "Bitte wähle ein Modell aus, um fortzufahren",
     "deleteHistoryConfirmation": "Bist du sicher, dass du diesen Verlauf löschen möchtest?",
-    "editHistoryTitle": "Gib einen neuen Titel ein"
+    "editHistoryTitle": "Gib einen neuen Titel ein",
+    "temporaryChat": "Temporärer Chat"
 }
\ No newline at end of file
diff --git a/src/assets/locale/en/option.json b/src/assets/locale/en/option.json
index 5739b40..7d11938 100644
--- a/src/assets/locale/en/option.json
+++ b/src/assets/locale/en/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "Something went wrong",
     "validationSelectModel": "Please select a model to continue",
     "deleteHistoryConfirmation": "Are you sure you want to delete this history?",
-    "editHistoryTitle": "Enter a new title"
+    "editHistoryTitle": "Enter a new title",
+    "temporaryChat": "Temporary Chat"
 }
\ No newline at end of file
diff --git a/src/assets/locale/es/option.json b/src/assets/locale/es/option.json
index 3c48761..d9ab8cf 100644
--- a/src/assets/locale/es/option.json
+++ b/src/assets/locale/es/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "Hubo un error",
     "validationSelectModel": "Selecione un modelo para continuar",
     "deleteHistoryConfirmation": "¿Esta seguro que quiere borrar éste histórico?",
-    "editHistoryTitle": "Ingrese un nuevo título"
+    "editHistoryTitle": "Ingrese un nuevo título",
+    "temporaryChat": "Chat Temporal"
 }
diff --git a/src/assets/locale/fa/option.json b/src/assets/locale/fa/option.json
index 9188710..5278cd0 100644
--- a/src/assets/locale/fa/option.json
+++ b/src/assets/locale/fa/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "مشکلی پیش آمد",
     "validationSelectModel": "لطفا یک مدل را برای ادامه انتخاب کنید",
     "deleteHistoryConfirmation": "آیا مطمئن هستید که می خواهید این تاریخچه را حذف کنید؟",
-    "editHistoryTitle": "یک عنوان جدید وارد کنید"
+    "editHistoryTitle": "یک عنوان جدید وارد کنید",
+    "temporaryChat": "گپ موقت"
 }
diff --git a/src/assets/locale/fr/option.json b/src/assets/locale/fr/option.json
index ec40c0e..ad374aa 100644
--- a/src/assets/locale/fr/option.json
+++ b/src/assets/locale/fr/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "Quelque chose s'est mal passé",
     "validationSelectModel": "Veuillez sélectionner un modèle pour continuer",
     "deleteHistoryConfirmation": "Êtes-vous sûr de vouloir supprimer cette historique ?",
-    "editHistoryTitle": "Entrez un nouveau titre"
+    "editHistoryTitle": "Entrez un nouveau titre",
+    "temporaryChat": "Chat temporaire"
 }
\ No newline at end of file
diff --git a/src/assets/locale/it/option.json b/src/assets/locale/it/option.json
index 6fcd098..fb222c3 100644
--- a/src/assets/locale/it/option.json
+++ b/src/assets/locale/it/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "Qualcosa è andato storto",
     "validationSelectModel": "Scegliere un modello per continuare",
     "deleteHistoryConfirmation": "Sei sicuro che vuoi eliminare la cronologia?",
-    "editHistoryTitle": "Inserisci un nuovo titolo"
+    "editHistoryTitle": "Inserisci un nuovo titolo",
+    "temporaryChat": "Chat Temporanea"
 }
\ No newline at end of file
diff --git a/src/assets/locale/ja-JP/option.json b/src/assets/locale/ja-JP/option.json
index 57b023b..76dc343 100644
--- a/src/assets/locale/ja-JP/option.json
+++ b/src/assets/locale/ja-JP/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "何かが間違っています",
     "validationSelectModel": "続行するにはモデルを選択してください",
     "deleteHistoryConfirmation": "この履歴を削除しますか?",
-    "editHistoryTitle": "新しいタイトルを入力"
+    "editHistoryTitle": "新しいタイトルを入力",
+    "temporaryChat": "一時的なチャット"
   }
\ No newline at end of file
diff --git a/src/assets/locale/ko/option.json b/src/assets/locale/ko/option.json
index 21a5048..2db30cc 100644
--- a/src/assets/locale/ko/option.json
+++ b/src/assets/locale/ko/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "문제가 발생했습니다",
     "validationSelectModel": "계속하려면 모델을 선택하세요",
     "deleteHistoryConfirmation": "이 기록을 삭제하시겠습니까?",
-    "editHistoryTitle": "새 제목 입력"
+    "editHistoryTitle": "새 제목 입력",
+    "temporaryChat": "임시 채팅"
 }
diff --git a/src/assets/locale/ml/option.json b/src/assets/locale/ml/option.json
index f8fcf70..1004b13 100644
--- a/src/assets/locale/ml/option.json
+++ b/src/assets/locale/ml/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "എന്തോ തെറ്റായി",
     "deleteHistoryConfirmation": "നിങ്ങളുടെ ചാറ്റ് ചരിത്രം ഇല്ലാതാക്കണമെന്ന് തീർച്ചയാണോ?",
     "editHistoryTitle": "ചാറ്റ് title എഡിറ്റുചെയ്യുക",
-    "validationSelectModel": "തുടരുന്നതിന് ഒരു മോഡല്‍ തിരഞ്ഞെടുക്കുക"
+    "validationSelectModel": "തുടരുന്നതിന് ഒരു മോഡല്‍ തിരഞ്ഞെടുക്കുക",
+    "temporaryChat": "താൽക്കാലിക ചാറ്റ്"
 }
\ No newline at end of file
diff --git a/src/assets/locale/no/option.json b/src/assets/locale/no/option.json
index 76d335e..64adb7b 100644
--- a/src/assets/locale/no/option.json
+++ b/src/assets/locale/no/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "Noe gikk galt",
     "validationSelectModel": "Vennligst velg en modell for å fortsette",
     "deleteHistoryConfirmation": "Er du sikker på at du vil slette denne historikken?",
-    "editHistoryTitle": "Skriv inn en ny tittel"
+    "editHistoryTitle": "Skriv inn en ny tittel",
+    "temporaryChat": "Midlertidig Chat"
 }
diff --git a/src/assets/locale/pt-BR/option.json b/src/assets/locale/pt-BR/option.json
index de29ff1..eeffe1c 100644
--- a/src/assets/locale/pt-BR/option.json
+++ b/src/assets/locale/pt-BR/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "Algo deu errado",
     "validationSelectModel": "Por favor, selecione um modelo para continuar",
     "deleteHistoryConfirmation": "Tem certeza de que deseja excluir este histórico?",
-    "editHistoryTitle": "Digite um novo título"
+    "editHistoryTitle": "Digite um novo título",
+    "temporaryChat": "Chat Temporário"
 }
\ No newline at end of file
diff --git a/src/assets/locale/ru/option.json b/src/assets/locale/ru/option.json
index f15c106..bd8f987 100644
--- a/src/assets/locale/ru/option.json
+++ b/src/assets/locale/ru/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "Что-то пошло не так",
     "validationSelectModel": "Пожалуйста, выберите модель, чтобы продолжить",
     "deleteHistoryConfirmation": "Вы уверены, что хотите удалить эту историю?",
-    "editHistoryTitle": "Введите новое название"
+    "editHistoryTitle": "Введите новое название",
+    "temporaryChat": "Временный чат"
 }
diff --git a/src/assets/locale/sv/option.json b/src/assets/locale/sv/option.json
index d7017ee..98e1012 100644
--- a/src/assets/locale/sv/option.json
+++ b/src/assets/locale/sv/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "Något gick fel",
     "validationSelectModel": "Vänligen välj en modell för att fortsätta",
     "deleteHistoryConfirmation": "Är du säker på att du vill radera denna historik?",
-    "editHistoryTitle": "Ange en ny titel"
+    "editHistoryTitle": "Ange en ny titel",
+    "temporaryChat": "Tillfällig chatt"
 }
diff --git a/src/assets/locale/zh/option.json b/src/assets/locale/zh/option.json
index 095daca..cba6731 100644
--- a/src/assets/locale/zh/option.json
+++ b/src/assets/locale/zh/option.json
@@ -8,5 +8,6 @@
     "somethingWentWrong": "出现了错误",
     "validationSelectModel": "请选择一个模型以继续",
     "deleteHistoryConfirmation": "你确定要删除这个历史记录吗？",
-    "editHistoryTitle": "输入一个新的标题"
+    "editHistoryTitle": "输入一个新的标题",
+    "temporaryChat": "临时聊天"
 }
\ No newline at end of file
diff --git a/src/components/Layouts/Header.tsx b/src/components/Layouts/Header.tsx
index 3d3ba89..7f3e153 100644
--- a/src/components/Layouts/Header.tsx
+++ b/src/components/Layouts/Header.tsx
@@ -21,6 +21,7 @@ import { Select, Tooltip } from "antd"
 import { getAllPrompts } from "@/db"
 import { ShareBtn } from "~/components/Common/ShareBtn"
 import { ProviderIcons } from "../Common/ProviderIcon"
+import { NewChat } from "./NewChat"
 type Props = {
   setSidebarOpen: (open: boolean) => void
   setOpenModelSettings: (open: boolean) => void
@@ -45,12 +46,12 @@ export const Header: React.FC<Props> = ({
     setSelectedSystemPrompt,
     messages,
     streaming,
-    historyId
+    historyId,
+    temporaryChat
   } = useMessageOption()
   const {
     data: models,
     isLoading: isModelsLoading,
-    isFetching: isModelsFetching
   } = useQuery({
     queryKey: ["fetchModel"],
     queryFn: () => fetchChatModels({ returnEmpty: true }),
@@ -86,7 +87,9 @@ export const Header: React.FC<Props> = ({
   }
 
   return (
-    <div className="sticky top-0 z-[999] flex h-16 p-3  bg-gray-50 border-b  dark:bg-[#171717] dark:border-gray-600">
+    <div className={`sticky top-0 z-[999] flex h-16 p-3  bg-gray-50 border-b  dark:bg-[#171717] dark:border-gray-600 ${
+      temporaryChat && "!bg-gray-200 dark:!bg-black"
+    }`}>
       <div className="flex gap-2 items-center">
         {pathname !== "/" && (
           <div>
@@ -104,14 +107,9 @@ export const Header: React.FC<Props> = ({
             <PanelLeftIcon className="w-6 h-6" />
           </button>
         </div>
-        <div>
-          <button
-            onClick={clearChat}
-            className="inline-flex  dark:bg-transparent bg-white items-center rounded-lg border  dark:border-gray-700 bg-transparent px-3 py-2.5 text-xs lg:text-sm font-medium leading-4 text-gray-800  dark:text-white disabled:opacity-50 ease-in-out transition-colors duration-200 hover:bg-gray-100 dark:hover:bg-gray-800 dark:hover:text-white">
-            <SquarePen className="h-5 w-5 " />
-            <span className=" truncate ml-3">{t("newChat")}</span>
-          </button>
-        </div>
+        <NewChat 
+          clearChat={clearChat}
+        />
         <span className="text-lg font-thin text-zinc-300 dark:text-zinc-600">
           {"/"}
         </span>
diff --git a/src/components/Layouts/NewChat.tsx b/src/components/Layouts/NewChat.tsx
new file mode 100644
index 0000000..e20681a
--- /dev/null
+++ b/src/components/Layouts/NewChat.tsx
@@ -0,0 +1,57 @@
+import { SquarePen, MoreHorizontal, TimerReset } from "lucide-react"
+import { useTranslation } from "react-i18next"
+import { Dropdown, Switch } from "antd"
+import type { MenuProps } from "antd"
+import { useMessageOption } from "@/hooks/useMessageOption"
+
+type Props = {
+  clearChat: () => void
+}
+
+export const NewChat: React.FC<Props> = ({ clearChat }) => {
+  const { t } = useTranslation(["option", "common"])
+
+  const { temporaryChat, setTemporaryChat, messages } = useMessageOption()
+
+  const items: MenuProps["items"] = [
+    {
+      key: "1",
+      label: (
+        <label className="flex items-center gap-6 justify-between px-1 py-0.5 cursor-pointer w-full">
+          <div className="flex items-center gap-2">
+            <TimerReset className="h-4 w-4 text-gray-600" />
+            <span>
+                {t("temporaryChat")}
+            </span>
+          </div>
+          <Switch
+            checked={temporaryChat}
+            onChange={(checked) => {
+              setTemporaryChat(checked)
+              // just like chatgpt
+              if (messages.length > 0) {
+                clearChat()
+              }
+            }}
+            size="small"
+          />
+        </label>
+      )
+    }
+  ]
+  return (
+    <div className="flex items-center justify-between">
+      <button
+        onClick={clearChat}
+        className="inline-flex dark:bg-transparent  bg-white items-center rounded-r-none rounded-lg border dark:border-gray-700 bg-transparent px-3 py-2.5 pr-6 text-xs lg:text-sm font-medium leading-4 text-gray-800 dark:text-white disabled:opacity-50 ease-in-out transition-colors duration-200 hover:bg-gray-100 dark:hover:bg-gray-800 dark:hover:text-white">
+        <SquarePen className="h-5 w-5" />
+        <span className="truncate ml-3">{t("newChat")}</span>
+      </button>
+      <Dropdown menu={{ items }} trigger={["click"]}>
+        <button className="inline-flex dark:bg-transparent bg-white items-center rounded-lg border-l-0 rounded-l-none border dark:border-gray-700 bg-transparent px-3 py-2.5 text-xs lg:text-sm font-medium leading-4 text-gray-800 dark:text-white disabled:opacity-50 ease-in-out transition-colors duration-200 hover:bg-gray-100 dark:hover:bg-gray-800 dark:hover:text-white">
+          <MoreHorizontal className="h-5 w-5 text-gray-600 dark:text-gray-400" />
+        </button>
+      </Dropdown>
+    </div>
+  )
+}
diff --git a/src/components/Option/Playground/PlaygroundForm.tsx b/src/components/Option/Playground/PlaygroundForm.tsx
index 4e0d637..5f19861 100644
--- a/src/components/Option/Playground/PlaygroundForm.tsx
+++ b/src/components/Option/Playground/PlaygroundForm.tsx
@@ -36,7 +36,8 @@ export const PlaygroundForm = ({ dropedFile }: Props) => {
     selectedQuickPrompt,
     textareaRef,
     setSelectedQuickPrompt,
-    selectedKnowledge
+    selectedKnowledge,
+    temporaryChat
   } = useMessageOption()
 
   const isMobile = () => {
@@ -190,7 +191,10 @@ export const PlaygroundForm = ({ dropedFile }: Props) => {
   }
 
   return (
-    <div className="px-3 pt-3 md:px-4 md:pt-4 bg-gray-100 dark:bg-[#262626] border rounded-t-xl  dark:border-gray-600">
+    <div
+      className={`px-3 pt-3 md:px-4 md:pt-4 bg-gray-100 dark:bg-[#262626] border rounded-t-xl  dark:border-gray-600
+    ${temporaryChat && "!bg-gray-300 dark:!bg-black "}
+    `}>
       <div
         className={`h-full rounded-md shadow relative ${
           form.values.image.length === 0 ? "hidden" : "block"
@@ -213,7 +217,9 @@ export const PlaygroundForm = ({ dropedFile }: Props) => {
         </div>
       </div>
       <div>
-        <div className="flex rounded-t-xl bg-white dark:bg-transparent">
+        <div className={`flex rounded-t-xl bg-white dark:bg-transparent ${
+          temporaryChat && "!bg-gray-300 dark:!bg-black"
+        }`}>
           <form
             onSubmit={form.onSubmit(async (value) => {
               stopListening()
@@ -228,7 +234,10 @@ export const PlaygroundForm = ({ dropedFile }: Props) => {
                   return
                 }
               }
-              if (value.message.trim().length === 0 && value.image.length === 0) {
+              if (
+                value.message.trim().length === 0 &&
+                value.image.length === 0
+              ) {
                 return
               }
               form.reset()
@@ -288,8 +297,6 @@ export const PlaygroundForm = ({ dropedFile }: Props) => {
                   )}
                 </div>
                 <div className="flex !justify-end gap-3">
-                  
-
                   {!selectedKnowledge && (
                     <Tooltip title={t("tooltip.uploadImage")}>
                       <button
@@ -411,4 +418,4 @@ export const PlaygroundForm = ({ dropedFile }: Props) => {
       </div>
     </div>
   )
-}
\ No newline at end of file
+}
diff --git a/src/components/Option/Sidebar.tsx b/src/components/Option/Sidebar.tsx
index 1c9fd54..6e6881d 100644
--- a/src/components/Option/Sidebar.tsx
+++ b/src/components/Option/Sidebar.tsx
@@ -34,7 +34,8 @@ export const Sidebar = ({ onClose }: Props) => {
     setHistoryId,
     historyId,
     clearChat,
-    setSelectedModel
+    setSelectedModel,
+    temporaryChat
   } = useMessageOption()
   const { t } = useTranslation(["option", "common"])
   const client = useQueryClient()
@@ -126,7 +127,7 @@ export const Sidebar = ({ onClose }: Props) => {
   })
 
   return (
-    <div className="overflow-y-auto z-99">
+    <div className={`overflow-y-auto z-99 ${temporaryChat ? 'pointer-events-none opacity-50' : ''}`}>
       {status === "success" && chatHistories.length === 0 && (
         <div className="flex justify-center items-center mt-20 overflow-hidden">
           <Empty description={t("common:noHistory")} />
@@ -244,4 +245,4 @@ export const Sidebar = ({ onClose }: Props) => {
       )}
     </div>
   )
-}
+}
\ No newline at end of file
diff --git a/src/hooks/useMessageOption.tsx b/src/hooks/useMessageOption.tsx
index 959cb44..b35e2e1 100644
--- a/src/hooks/useMessageOption.tsx
+++ b/src/hooks/useMessageOption.tsx
@@ -22,7 +22,10 @@ import { notification } from "antd"
 import { getSystemPromptForWeb } from "~/web/web"
 import { generateHistory } from "@/utils/generate-history"
 import { useTranslation } from "react-i18next"
-import { saveMessageOnError, saveMessageOnSuccess } from "./chat-helper"
+import {
+  saveMessageOnError as saveError,
+  saveMessageOnSuccess as saveSuccess
+} from "./chat-helper"
 import { usePageAssist } from "@/context"
 import { PageAssistVectorStore } from "@/libs/PageAssistVectorStore"
 import { formatDocs } from "@/chain/chat-with-x"
@@ -65,7 +68,9 @@ export const useMessageOption = () => {
     selectedSystemPrompt,
     setSelectedSystemPrompt,
     selectedKnowledge,
-    setSelectedKnowledge
+    setSelectedKnowledge,
+    temporaryChat,
+    setTemporaryChat
   } = useStoreMessageOption()
   const currentChatModelSettings = useStoreChatModelSettings()
   const [selectedModel, setSelectedModel] = useStorage("selectedModel")
@@ -123,8 +128,9 @@ export const useMessageOption = () => {
       seed: currentChatModelSettings?.seed,
       numGpu:
         currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
-
+      numPredict:
+        currentChatModelSettings?.numPredict ??
+        userDefaultModelSettings?.numPredict
     })
 
     let newMessage: Message[] = []
@@ -199,9 +205,11 @@ export const useMessageOption = () => {
             userDefaultModelSettings?.numCtx,
           seed: currentChatModelSettings?.seed,
           numGpu:
-            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
-
+            currentChatModelSettings?.numGpu ??
+            userDefaultModelSettings?.numGpu,
+          numPredict:
+            currentChatModelSettings?.numPredict ??
+            userDefaultModelSettings?.numPredict
         })
         const response = await questionOllama.invoke(promptForQuestion)
         query = response.content.toString()
@@ -355,6 +363,22 @@ export const useMessageOption = () => {
     }
   }
 
+  const saveMessageOnSuccess = async (e: any) => {
+    if (!temporaryChat) {
+      return await saveSuccess(e)
+    }
+
+    return true
+  }
+
+  const saveMessageOnError = async (e: any) => {
+    if (!temporaryChat) {
+      return await saveError(e)
+    }
+
+    return true
+  }
+
   const normalChatMode = async (
     message: string,
     image: string,
@@ -386,7 +410,9 @@ export const useMessageOption = () => {
       seed: currentChatModelSettings?.seed,
       numGpu:
         currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
+      numPredict:
+        currentChatModelSettings?.numPredict ??
+        userDefaultModelSettings?.numPredict
     })
 
     let newMessage: Message[] = []
@@ -501,7 +527,7 @@ export const useMessageOption = () => {
                 }
               }
             }
-          ],
+          ]
         }
       )
 
@@ -622,8 +648,9 @@ export const useMessageOption = () => {
       seed: currentChatModelSettings?.seed,
       numGpu:
         currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
-
+      numPredict:
+        currentChatModelSettings?.numPredict ??
+        userDefaultModelSettings?.numPredict
     })
 
     let newMessage: Message[] = []
@@ -714,9 +741,11 @@ export const useMessageOption = () => {
             userDefaultModelSettings?.numCtx,
           seed: currentChatModelSettings?.seed,
           numGpu:
-            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
-
+            currentChatModelSettings?.numGpu ??
+            userDefaultModelSettings?.numGpu,
+          numPredict:
+            currentChatModelSettings?.numPredict ??
+            userDefaultModelSettings?.numPredict
         })
         const response = await questionOllama.invoke(promptForQuestion)
         query = response.content.toString()
@@ -1038,6 +1067,8 @@ export const useMessageOption = () => {
     textareaRef,
     selectedKnowledge,
     setSelectedKnowledge,
-    ttsEnabled
+    ttsEnabled,
+    temporaryChat,
+    setTemporaryChat,
   }
 }
diff --git a/src/store/option.tsx b/src/store/option.tsx
index 8a55501..49911cd 100644
--- a/src/store/option.tsx
+++ b/src/store/option.tsx
@@ -65,6 +65,9 @@ type State = {
 
   setSpeechToTextLanguage: (language: string) => void
   speechToTextLanguage: string
+
+  temporaryChat: boolean
+  setTemporaryChat: (temporaryChat: boolean) => void
 }
 
 export const useStoreMessageOption = create<State>((set) => ({
@@ -102,5 +105,8 @@ export const useStoreMessageOption = create<State>((set) => ({
   setSelectedQuickPrompt: (selectedQuickPrompt) => set({ selectedQuickPrompt }),
 
   selectedKnowledge: null,
-  setSelectedKnowledge: (selectedKnowledge) => set({ selectedKnowledge })
+  setSelectedKnowledge: (selectedKnowledge) => set({ selectedKnowledge }),
+
+  temporaryChat: false,
+  setTemporaryChat: (temporaryChat) => set({ temporaryChat }),
 }))

From 9c7a3f5ddc10079ded309d75850b74489e495bc2 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 10 Nov 2024 12:37:48 +0530
Subject: [PATCH 09/27] fix: Prevent errors from optional fields in chat
 history and chunk content

The code was relying on optional fields like `content` in chat history and chunk objects, leading to potential errors if these fields were missing. This commit ensures proper handling of these fields by adding optional chaining (`?.`) for safer access. This prevents crashes and ensures the application handles the missing fields gracefully.
---
 src/chain/chat-with-website.ts |  7 ++++---
 src/chain/chat-with-x.ts       |  1 +
 src/hooks/useMessage.tsx       | 16 ++++++++--------
 src/hooks/useMessageOption.tsx | 12 ++++++------
 4 files changed, 19 insertions(+), 17 deletions(-)

diff --git a/src/chain/chat-with-website.ts b/src/chain/chat-with-website.ts
index 3b909d4..1c6b5d3 100644
--- a/src/chain/chat-with-website.ts
+++ b/src/chain/chat-with-website.ts
@@ -1,3 +1,4 @@
+//@ts-nocheck
 import { BaseLanguageModel } from "langchain/base_language";
 import { Document } from "@langchain/core/documents";
 import {
@@ -28,8 +29,8 @@ export function groupMessagesByConversation(messages: ChatHistory) {
   const groupedMessages = [];
   for (let i = 0; i < messages.length; i += 2) {
     groupedMessages.push({
-      human: messages[i].content,
-      ai: messages[i + 1].content,
+      human: messages[i]?.content,
+      ai: messages[i + 1]?.content,
     });
   }
 
@@ -38,7 +39,7 @@ export function groupMessagesByConversation(messages: ChatHistory) {
 
 const formatChatHistoryAsString = (history: BaseMessage[]) => {
   return history
-    .map((message) => `${message._getType()}: ${message.content}`)
+    .map((message) => `${message._getType()}: ${message?.content}`)
     .join("\n");
 };
 
diff --git a/src/chain/chat-with-x.ts b/src/chain/chat-with-x.ts
index 4a63829..296277f 100644
--- a/src/chain/chat-with-x.ts
+++ b/src/chain/chat-with-x.ts
@@ -1,3 +1,4 @@
+//@ts-nocheck
 import { BaseLanguageModel } from "@langchain/core/language_models/base"
 import { Document } from "@langchain/core/documents"
 import {
diff --git a/src/hooks/useMessage.tsx b/src/hooks/useMessage.tsx
index cc38949..d06102d 100644
--- a/src/hooks/useMessage.tsx
+++ b/src/hooks/useMessage.tsx
@@ -355,8 +355,8 @@ export const useMessage = () => {
       )
       let count = 0
       for await (const chunk of chunks) {
-        contentToSave += chunk.content
-        fullText += chunk.content
+        contentToSave += chunk?.content
+        fullText += chunk?.content
         if (count === 0) {
           setIsProcessing(true)
         }
@@ -590,8 +590,8 @@ export const useMessage = () => {
       )
       let count = 0
       for await (const chunk of chunks) {
-        contentToSave += chunk.content
-        fullText += chunk.content
+        contentToSave += chunk?.content
+        fullText += chunk?.content
         if (count === 0) {
           setIsProcessing(true)
         }
@@ -855,8 +855,8 @@ export const useMessage = () => {
       )
       let count = 0
       for await (const chunk of chunks) {
-        contentToSave += chunk.content
-        fullText += chunk.content
+        contentToSave += chunk?.content
+        fullText += chunk?.content
         if (count === 0) {
           setIsProcessing(true)
         }
@@ -1064,8 +1064,8 @@ export const useMessage = () => {
       })
       let count = 0
       for await (const chunk of chunks) {
-        contentToSave += chunk.content
-        fullText += chunk.content
+        contentToSave += chunk?.content
+        fullText += chunk?.content
         if (count === 0) {
           setIsProcessing(true)
         }
diff --git a/src/hooks/useMessageOption.tsx b/src/hooks/useMessageOption.tsx
index b35e2e1..90fc06a 100644
--- a/src/hooks/useMessageOption.tsx
+++ b/src/hooks/useMessageOption.tsx
@@ -276,8 +276,8 @@ export const useMessageOption = () => {
       )
       let count = 0
       for await (const chunk of chunks) {
-        contentToSave += chunk.content
-        fullText += chunk.content
+        contentToSave += chunk?.content
+        fullText += chunk?.content
         if (count === 0) {
           setIsProcessing(true)
         }
@@ -533,8 +533,8 @@ export const useMessageOption = () => {
 
       let count = 0
       for await (const chunk of chunks) {
-        contentToSave += chunk.content
-        fullText += chunk.content
+        contentToSave += chunk?.content
+        fullText += chunk?.content
         if (count === 0) {
           setIsProcessing(true)
         }
@@ -800,8 +800,8 @@ export const useMessageOption = () => {
       )
       let count = 0
       for await (const chunk of chunks) {
-        contentToSave += chunk.content
-        fullText += chunk.content
+        contentToSave += chunk?.content
+        fullText += chunk?.content
         if (count === 0) {
           setIsProcessing(true)
         }

From f52e3d564acdb0cb83a9fe0de43876fa7752e518 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 10 Nov 2024 13:29:41 +0530
Subject: [PATCH 10/27] Fix: Handle image URLs in custom model responses

Improves the formatting of image URLs within responses from custom models.  This ensures that image URLs are correctly presented in the user interface.
---
 src/utils/human-message.tsx | 69 +++++++++++++++++++------------------
 1 file changed, 36 insertions(+), 33 deletions(-)

diff --git a/src/utils/human-message.tsx b/src/utils/human-message.tsx
index 6712339..9c1200f 100644
--- a/src/utils/human-message.tsx
+++ b/src/utils/human-message.tsx
@@ -1,43 +1,46 @@
 import { isCustomModel } from "@/db/models"
 import { HumanMessage, type MessageContent } from "@langchain/core/messages"
 
-
 type HumanMessageType = {
-    content: MessageContent,
-    model: string
+  content: MessageContent
+  model: string
 }
 
 export const humanMessageFormatter = ({ content, model }: HumanMessageType) => {
-    
-    const isCustom = isCustomModel(model)
+  const isCustom = isCustomModel(model)
 
-    if(isCustom) {
-        if(typeof content !== 'string') {
-            if(content.length > 1) {
-                // this means that we need to reformat the image_url 
-                const newContent: MessageContent = [
-                    {
-                        type: "text",
-                        //@ts-ignore
-                        text: content[0].text
-                    },
-                    {
-                        type: "image_url",
-                        image_url: {
-                            //@ts-ignore
-                            url: content[1].image_url
-                        }
-                    }
-                ]
-
-                return new HumanMessage({
-                    content: newContent
-                })
+  if (isCustom) {
+    if (typeof content !== "string") {
+      if (content.length > 1) {
+        // this means that we need to reformat the image_url
+        const newContent: MessageContent = [
+          {
+            type: "text",
+            //@ts-ignore
+            text: content[0].text
+          },
+          {
+            type: "image_url",
+            image_url: {
+              //@ts-ignore
+              url: content[1].image_url
             }
-         }
+          }
+        ]
+
+        return new HumanMessage({
+          content: newContent
+        })
+      } else {
+        return new HumanMessage({
+          //@ts-ignore
+          content: content[0].text
+        })
+      }
     }
-    
-    return new HumanMessage({
-        content,
-    })
-}
\ No newline at end of file
+  }
+
+  return new HumanMessage({
+    content
+  })
+}

From c6a62126dd2c14917dc4a25de90617e747f7ebbe Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 10 Nov 2024 14:02:44 +0530
Subject: [PATCH 11/27] feat: Add LlamaFile support

Add support for LlamaFile, a new model provider that allows users to interact with models stored in LlamaFile format. This includes:

- Adding an icon for LlamaFile in the provider selection menu.
- Updating the model provider selection to include LlamaFile.
- Updating the model handling logic to properly identify and process LlamaFile models.
- Updating the API providers list to include LlamaFile.

This enables users to leverage the capabilities of LlamaFile models within the application.
---
 src/components/Common/ProviderIcon.tsx    |  3 +
 src/components/Icons/Llamafile.tsx        | 24 +++++++
 src/components/Option/Settings/openai.tsx |  3 +-
 src/db/models.ts                          | 84 ++++++++++++++++++++++-
 src/utils/oai-api-providers.ts            |  5 ++
 5 files changed, 117 insertions(+), 2 deletions(-)
 create mode 100644 src/components/Icons/Llamafile.tsx

diff --git a/src/components/Common/ProviderIcon.tsx b/src/components/Common/ProviderIcon.tsx
index 38ba504..8142adc 100644
--- a/src/components/Common/ProviderIcon.tsx
+++ b/src/components/Common/ProviderIcon.tsx
@@ -6,6 +6,7 @@ import { LMStudioIcon } from "../Icons/LMStudio"
 import { OpenAiIcon } from "../Icons/OpenAI"
 import { TogtherMonoIcon } from "../Icons/Togther"
 import { OpenRouterIcon } from "../Icons/OpenRouter"
+import { LLamaFile } from "../Icons/Llamafile"
 
 export const ProviderIcons = ({
   provider,
@@ -31,6 +32,8 @@ export const ProviderIcons = ({
       return <TogtherMonoIcon className={className} />
     case "openrouter":
       return <OpenRouterIcon className={className} />
+    case "llamafile":
+      return <LLamaFile className={className} />
     default:
       return <OllamaIcon className={className} />
   }
diff --git a/src/components/Icons/Llamafile.tsx b/src/components/Icons/Llamafile.tsx
new file mode 100644
index 0000000..734cdd9
--- /dev/null
+++ b/src/components/Icons/Llamafile.tsx
@@ -0,0 +1,24 @@
+// copied logo from Hugging Face webiste 
+import React from "react"
+
+export const LLamaFile = React.forwardRef<
+  SVGSVGElement,
+  React.SVGProps<SVGSVGElement>
+>((props, ref) => {
+  return (
+    <svg
+      xmlns="http://www.w3.org/2000/svg"
+      width="1em"
+      height="1em"
+      className="text-black inline-block text-sm"
+      viewBox="0 0 16 16"
+      ref={ref}
+      {...props}>
+      <path
+        fill="currentColor"
+        fillRule="evenodd"
+        d="M7.66 5.82H4.72a.31.31 0 0 1-.32-.32c0-.64-.52-1.16-1.16-1.16H1.6a.7.7 0 0 0-.7.7v8.3c0 .52.43.95.96.95h9.4c.4 0 .75-.3.82-.7l.65-3.84.74-.07a2.08 2.08 0 0 0 .28-4.1l-.94-.22-.11-.2a2.3 2.3 0 0 0-.54-.64v-.05a5.6 5.6 0 0 1 .1-1.02l.07-.45c.01-.1.02-.21.01-.32a.6.6 0 0 0-.1-.27.5.5 0 0 0-.54-.21c-.12.03-.2.1-.24.14a1 1 0 0 0-.12.13A4.8 4.8 0 0 0 10.76 4h-.33l.05-.49.04-.28.06-.53c.01-.1.02-.26 0-.42a1 1 0 0 0-.15-.43.87.87 0 0 0-.93-.35.96.96 0 0 0-.4.22c-.08.06-.14.13-.18.19a5.5 5.5 0 0 0-.55 1.25c-.15.52-.28 1.2-.24 1.85-.2.24-.36.51-.47.8Zm.66.8c.06-.52.3-.98.67-1.3l.02-.01c-.15-.72.05-1.65.28-2.28.15-.41.31-.7.41-.72.05-.01.06.07.06.22l-.07.56v.06a11 11 0 0 0-.1 1.28c.01.21.05.39.14.49a2 2 0 0 1 .57-.08h.42c.52 0 1 .28 1.25.73l.2.36c.06.1.16.18.28.21l1.11.26a1.24 1.24 0 0 1-.17 2.45l-1.38.12-.76 4.48h-4.2L8.33 6.6Z"
+        clipRule="evenodd"></path>
+    </svg>
+  )
+})
diff --git a/src/components/Option/Settings/openai.tsx b/src/components/Option/Settings/openai.tsx
index 273b1ce..fe4dcdb 100644
--- a/src/components/Option/Settings/openai.tsx
+++ b/src/components/Option/Settings/openai.tsx
@@ -47,7 +47,8 @@ export const OpenAIApp = () => {
       })
       setOpen(false)
       message.success(t("addSuccess"))
-      if (provider !== "lmstudio") {
+      const noPopupProvider = ["lmstudio", "llamafile"]
+      if (!noPopupProvider.includes(provider)) {
         setOpenaiId(data)
         setOpenModelModal(true)
       }
diff --git a/src/db/models.ts b/src/db/models.ts
index 4575c67..eaa7650 100644
--- a/src/db/models.ts
+++ b/src/db/models.ts
@@ -24,6 +24,7 @@ export const removeModelSuffix = (id: string) => {
   return id
     .replace(/_model-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{3,4}-[a-f0-9]{4}/, "")
     .replace(/_lmstudio_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/, "")
+    .replace(/_llamafile_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/, "")
 }
 export const isLMStudioModel = (model: string) => {
   const lmstudioModelRegex =
@@ -31,6 +32,12 @@ export const isLMStudioModel = (model: string) => {
   return lmstudioModelRegex.test(model)
 }
 
+export const isLlamafileModel = (model: string) => {
+  const llamafileModelRegex =
+    /_llamafile_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/
+  return llamafileModelRegex.test(model)
+}
+
 export const getLMStudioModelId = (
   model: string
 ): { model_id: string; provider_id: string } => {
@@ -44,10 +51,29 @@ export const getLMStudioModelId = (
   }
   return null
 }
+
+export const getLlamafileModelId = (
+  model: string
+): { model_id: string; provider_id: string } => {
+  const llamafileModelRegex =
+    /_llamafile_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/
+  const match = model.match(llamafileModelRegex)
+  if (match) {
+    const modelId = match[0]
+    const providerId = match[0].replace("_llamafile_openai-", "")
+    return { model_id: modelId, provider_id: providerId }
+  }
+  return null
+}
 export const isCustomModel = (model: string) => {
   if (isLMStudioModel(model)) {
     return true
   }
+
+  if (isLlamafileModel(model)) {
+    return true
+  }
+
   const customModelRegex =
     /_model-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{3,4}-[a-f0-9]{4}/
   return customModelRegex.test(model)
@@ -201,6 +227,25 @@ export const getModelInfo = async (id: string) => {
     }
   }
 
+
+  if (isLlamafileModel(id)) {
+    const llamafileId = getLlamafileModelId(id)
+    if (!llamafileId) {
+      throw new Error("Invalid LMStudio model ID")
+    }
+    return {
+      model_id: id.replace(
+        /_llamafile_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/,
+        ""
+      ),
+      provider_id: `openai-${llamafileId.provider_id}`,
+      name: id.replace(
+        /_llamafile_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/,
+        ""
+      )
+    }
+  }
+
   const model = await db.getById(id)
   return model
 }
@@ -264,6 +309,27 @@ export const dynamicFetchLMStudio = async ({
   return lmstudioModels
 }
 
+export const dynamicFetchLlamafile = async ({
+  baseUrl,
+  providerId
+}: {
+  baseUrl: string
+  providerId: string
+}) => {
+  const models = await getAllOpenAIModels(baseUrl)
+  const llamafileModels = models.map((e) => {
+    return {
+      name: e?.name || e?.id,
+      id: `${e?.id}_llamafile_${providerId}`,
+      provider: providerId,
+      lookup: `${e?.id}_${providerId}`,
+      provider_id: providerId
+    }
+  })
+
+  return llamafileModels
+}
+
 export const ollamaFormatAllCustomModels = async (
   modelType: "all" | "chat" | "embedding" = "all"
 ) => {
@@ -276,6 +342,10 @@ export const ollamaFormatAllCustomModels = async (
     (provider) => provider.provider === "lmstudio"
   )
 
+  const llamafileProviders = allProviders.filter(
+    (provider) => provider.provider === "llamafile"
+  )
+
   const lmModelsPromises = lmstudioProviders.map((provider) =>
     dynamicFetchLMStudio({
       baseUrl: provider.baseUrl,
@@ -283,16 +353,28 @@ export const ollamaFormatAllCustomModels = async (
     })
   )
 
+  const llamafileModelsPromises = llamafileProviders.map((provider) =>
+    dynamicFetchLlamafile({
+      baseUrl: provider.baseUrl,
+      providerId: provider.id
+    })
+  )
+
   const lmModelsFetch = await Promise.all(lmModelsPromises)
 
+  const llamafileModelsFetch = await Promise.all(llamafileModelsPromises)
+
   const lmModels = lmModelsFetch.flat()
 
+  const llamafileModels = llamafileModelsFetch.flat()
+
   // merge allModels and lmModels
   const allModlesWithLMStudio = [
     ...(modelType !== "all"
       ? allModles.filter((model) => model.model_type === modelType)
       : allModles),
-    ...lmModels
+    ...lmModels,
+    ...llamafileModels
   ]
 
   const ollamaModels = allModlesWithLMStudio.map((model) => {
diff --git a/src/utils/oai-api-providers.ts b/src/utils/oai-api-providers.ts
index 5d65105..52bfcf9 100644
--- a/src/utils/oai-api-providers.ts
+++ b/src/utils/oai-api-providers.ts
@@ -4,6 +4,11 @@ export const OAI_API_PROVIDERS = [
         value: "lmstudio",
         baseUrl: "http://localhost:1234/v1"
     },
+    {
+        label: "LlamaFile",
+        value: "llamafile",
+        baseUrl: "http://127.0.0.1:8080/v1"
+    },
     {
         label: "OpenAI",
         value: "openai",

From a7f461da0b1d4af9025283f5dd82aa927fc3cc82 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 10 Nov 2024 15:31:28 +0530
Subject: [PATCH 12/27] feat: Add multiple Ollama  support

Adds support for using Ollama 2 as a model provider. This includes:

- Adding Ollama 2 to the list of supported providers in the UI
- Updating the model identification logic to properly handle Ollama 2 models
- Modifying the model loading and runtime configuration to work with Ollama 2
- Implementing Ollama 2 specific functionality in the embedding and chat models

This change allows users to leverage the capabilities of Ollama 2 for both embeddings and conversational AI tasks.
---
 src/components/Option/Settings/openai.tsx |  2 +-
 src/db/models.ts                          | 79 ++++++++++++++++++++++-
 src/models/embedding.ts                   |  2 +-
 src/models/index.ts                       |  7 +-
 src/utils/oai-api-providers.ts            | 18 ++++--
 5 files changed, 96 insertions(+), 12 deletions(-)

diff --git a/src/components/Option/Settings/openai.tsx b/src/components/Option/Settings/openai.tsx
index fe4dcdb..a0807ed 100644
--- a/src/components/Option/Settings/openai.tsx
+++ b/src/components/Option/Settings/openai.tsx
@@ -47,7 +47,7 @@ export const OpenAIApp = () => {
       })
       setOpen(false)
       message.success(t("addSuccess"))
-      const noPopupProvider = ["lmstudio", "llamafile"]
+      const noPopupProvider = ["lmstudio", "llamafile", "ollama2"]
       if (!noPopupProvider.includes(provider)) {
         setOpenaiId(data)
         setOpenModelModal(true)
diff --git a/src/db/models.ts b/src/db/models.ts
index eaa7650..012379c 100644
--- a/src/db/models.ts
+++ b/src/db/models.ts
@@ -25,6 +25,7 @@ export const removeModelSuffix = (id: string) => {
     .replace(/_model-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{3,4}-[a-f0-9]{4}/, "")
     .replace(/_lmstudio_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/, "")
     .replace(/_llamafile_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/, "")
+    .replace(/_ollama2_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/, "")
 }
 export const isLMStudioModel = (model: string) => {
   const lmstudioModelRegex =
@@ -37,7 +38,11 @@ export const isLlamafileModel = (model: string) => {
     /_llamafile_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/
   return llamafileModelRegex.test(model)
 }
-
+export const isOllamaModel = (model: string) => {
+  const ollamaModelRegex =
+    /_ollama2_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/
+  return ollamaModelRegex.test(model)
+}
 export const getLMStudioModelId = (
   model: string
 ): { model_id: string; provider_id: string } => {
@@ -51,7 +56,19 @@ export const getLMStudioModelId = (
   }
   return null
 }
-
+export const getOllamaModelId = (
+  model: string
+): { model_id: string; provider_id: string } => {
+  const ollamaModelRegex =
+    /_ollama2_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/
+  const match = model.match(ollamaModelRegex)
+  if (match) {
+    const modelId = match[0]
+    const providerId = match[0].replace("_ollama2_openai-", "")
+    return { model_id: modelId, provider_id: providerId }
+  }
+  return null
+}
 export const getLlamafileModelId = (
   model: string
 ): { model_id: string; provider_id: string } => {
@@ -74,6 +91,10 @@ export const isCustomModel = (model: string) => {
     return true
   }
 
+  if (isOllamaModel(model)) {
+    return true
+  }
+
   const customModelRegex =
     /_model-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{3,4}-[a-f0-9]{4}/
   return customModelRegex.test(model)
@@ -246,6 +267,25 @@ export const getModelInfo = async (id: string) => {
     }
   }
 
+
+  if (isOllamaModel(id)) {
+    const ollamaId = getOllamaModelId(id)
+    if (!ollamaId) {
+      throw new Error("Invalid LMStudio model ID")
+    }
+    return {
+      model_id: id.replace(
+        /_ollama2_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/,
+        ""
+      ),
+      provider_id: `openai-${ollamaId.provider_id}`,
+      name: id.replace(
+        /_ollama2_openai-[a-f0-9]{4}-[a-f0-9]{3}-[a-f0-9]{4}/,
+        ""
+      )
+    }
+  }
+
   const model = await db.getById(id)
   return model
 }
@@ -309,6 +349,27 @@ export const dynamicFetchLMStudio = async ({
   return lmstudioModels
 }
 
+export const dynamicFetchOllama2 = async ({
+  baseUrl,
+  providerId
+}: {
+  baseUrl: string
+  providerId: string
+}) => {
+  const models = await getAllOpenAIModels(baseUrl)
+  const ollama2Models = models.map((e) => {
+    return {
+      name: e?.name || e?.id,
+      id: `${e?.id}_ollama2_${providerId}`,
+      provider: providerId,
+      lookup: `${e?.id}_${providerId}`,
+      provider_id: providerId
+    }
+  })
+
+  return ollama2Models
+}
+
 export const dynamicFetchLlamafile = async ({
   baseUrl,
   providerId
@@ -360,21 +421,33 @@ export const ollamaFormatAllCustomModels = async (
     })
   )
 
+  const ollamaModelsPromises = allProviders.map((provider) => (
+    dynamicFetchOllama2({
+      baseUrl: provider.baseUrl,
+      providerId: provider.id
+    })
+  ))
+
   const lmModelsFetch = await Promise.all(lmModelsPromises)
 
   const llamafileModelsFetch = await Promise.all(llamafileModelsPromises)
 
+  const ollamaModelsFetch = await Promise.all(ollamaModelsPromises)
+
   const lmModels = lmModelsFetch.flat()
 
   const llamafileModels = llamafileModelsFetch.flat()
 
+  const ollama2Models = ollamaModelsFetch.flat()
+
   // merge allModels and lmModels
   const allModlesWithLMStudio = [
     ...(modelType !== "all"
       ? allModles.filter((model) => model.model_type === modelType)
       : allModles),
     ...lmModels,
-    ...llamafileModels
+    ...llamafileModels,
+    ...ollama2Models
   ]
 
   const ollamaModels = allModlesWithLMStudio.map((model) => {
diff --git a/src/models/embedding.ts b/src/models/embedding.ts
index 03eb663..0c1026b 100644
--- a/src/models/embedding.ts
+++ b/src/models/embedding.ts
@@ -1,4 +1,4 @@
-import { getModelInfo, isCustomModel } from "@/db/models"
+import { getModelInfo, isCustomModel, isOllamaModel } from "@/db/models"
 import { OllamaEmbeddingsPageAssist } from "./OllamaEmbedding"
 import { OAIEmbedding } from "./OAIEmbedding"
 import { getOpenAIConfigById } from "@/db/openai"
diff --git a/src/models/index.ts b/src/models/index.ts
index 4798f78..135025f 100644
--- a/src/models/index.ts
+++ b/src/models/index.ts
@@ -1,8 +1,9 @@
-import { getModelInfo, isCustomModel } from "@/db/models"
+import { getModelInfo, isCustomModel, isOllamaModel } from "@/db/models"
 import { ChatChromeAI } from "./ChatChromeAi"
 import { ChatOllama } from "./ChatOllama"
 import { getOpenAIConfigById } from "@/db/openai"
 import { ChatOpenAI } from "@langchain/openai"
+import { urlRewriteRuntime } from "@/libs/runtime"
 
 export const pageAssistModel = async ({
   model,
@@ -43,6 +44,10 @@ export const pageAssistModel = async ({
     const modelInfo = await getModelInfo(model)
     const providerInfo = await getOpenAIConfigById(modelInfo.provider_id)
 
+    if (isOllamaModel(model)) {
+      await urlRewriteRuntime(providerInfo.baseUrl || "")
+    }
+
     return new ChatOpenAI({
       modelName: modelInfo.model_id,
       openAIApiKey: providerInfo.apiKey || "temp",
diff --git a/src/utils/oai-api-providers.ts b/src/utils/oai-api-providers.ts
index 52bfcf9..40c1e21 100644
--- a/src/utils/oai-api-providers.ts
+++ b/src/utils/oai-api-providers.ts
@@ -1,14 +1,24 @@
 export const OAI_API_PROVIDERS = [
+    {
+        label: "Custom",
+        value: "custom",
+        baseUrl: ""
+    },
     {
         label: "LM Studio",
         value: "lmstudio",
         baseUrl: "http://localhost:1234/v1"
     },
     {
-        label: "LlamaFile",
+        label: "Llamafile",
         value: "llamafile",
         baseUrl: "http://127.0.0.1:8080/v1"
     },
+    {
+        label: "Ollama",
+        value: "ollama2",
+        baseUrl: "http://localhost:11434/v1"
+    },
     {
         label: "OpenAI",
         value: "openai",
@@ -34,9 +44,5 @@ export const OAI_API_PROVIDERS = [
         value: "openrouter",
         baseUrl: "https://openrouter.ai/api/v1"
     },
-    {
-        label: "Custom",
-        value: "custom",
-        baseUrl: ""
-    }
+   
 ]
\ No newline at end of file

From 2409ebc75deb37ecd943b1aa63b8b859dcf669be Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 10 Nov 2024 15:38:03 +0530
Subject: [PATCH 13/27] feat: Add Ollama and Llamafile to dynamic model
 fetching

Expanded the list of providers for which models are fetched dynamically to include Ollama and Llamafile, removing the need for manual model addition in the user interface for these providers. This simplifies the user experience and ensures users always have access to the latest models without manual intervention.
---
 src/assets/locale/da/openai.json          | 2 +-
 src/assets/locale/de/openai.json          | 2 +-
 src/assets/locale/en/openai.json          | 2 +-
 src/assets/locale/es/openai.json          | 2 +-
 src/assets/locale/fa/openai.json          | 2 +-
 src/assets/locale/fr/openai.json          | 2 +-
 src/assets/locale/it/openai.json          | 2 +-
 src/assets/locale/ja-JP/openai.json       | 2 +-
 src/assets/locale/ko/openai.json          | 2 +-
 src/assets/locale/ml/openai.json          | 2 +-
 src/assets/locale/no/openai.json          | 2 +-
 src/assets/locale/ru/openai.json          | 2 +-
 src/assets/locale/zh/openai.json          | 2 +-
 src/components/Option/Settings/openai.tsx | 8 +++++---
 14 files changed, 18 insertions(+), 16 deletions(-)

diff --git a/src/assets/locale/da/openai.json b/src/assets/locale/da/openai.json
index 3410ae6..8ef3c07 100644
--- a/src/assets/locale/da/openai.json
+++ b/src/assets/locale/da/openai.json
@@ -42,7 +42,7 @@
     "delete": "Slet",
     "edit": "Rediger",
     "newModel": "Tilføj Modeller til Udbyder",
-    "noNewModel": "For LMStudio henter vi dynamisk. Ingen manuel tilføjelse nødvendig.",
+    "noNewModel": "For LMStudio, Ollama, Llamafile, henter vi dynamisk. Ingen manuel tilføjelse nødvendig.",
     "searchModel": "Søg Model",
     "selectAll": "Vælg Alle",
     "save": "Gem",
diff --git a/src/assets/locale/de/openai.json b/src/assets/locale/de/openai.json
index 23a684e..98e7519 100644
--- a/src/assets/locale/de/openai.json
+++ b/src/assets/locale/de/openai.json
@@ -42,7 +42,7 @@
     "delete": "Löschen",
     "edit": "Bearbeiten",
     "newModel": "Modelle zum Anbieter hinzufügen",
-    "noNewModel": "Für LMStudio holen wir die Daten dynamisch. Keine manuelle Hinzufügung erforderlich.",
+    "noNewModel": "Für LMStudio, Ollama, Llamafile, holen wir die Daten dynamisch. Keine manuelle Hinzufügung erforderlich.",
     "searchModel": "Modell suchen",
     "selectAll": "Alle auswählen",
     "save": "Speichern",
diff --git a/src/assets/locale/en/openai.json b/src/assets/locale/en/openai.json
index c0838cb..f8ad298 100644
--- a/src/assets/locale/en/openai.json
+++ b/src/assets/locale/en/openai.json
@@ -42,7 +42,7 @@
     "delete": "Delete",
     "edit": "Edit",
     "newModel": "Add Models to Provider",
-    "noNewModel": "For LMStudio, we fetch dynamically. No manual addition needed.",
+    "noNewModel": "For LMStudio, Ollama, Llamafile, we fetch dynamically. No manual addition needed.",
     "searchModel": "Search Model",
     "selectAll": "Select All",
     "save": "Save",
diff --git a/src/assets/locale/es/openai.json b/src/assets/locale/es/openai.json
index 2e84a19..b4bd9b2 100644
--- a/src/assets/locale/es/openai.json
+++ b/src/assets/locale/es/openai.json
@@ -42,7 +42,7 @@
     "delete": "Eliminar",
     "edit": "Editar",
     "newModel": "Añadir Modelos al Proveedor",
-    "noNewModel": "Para LMStudio, obtenemos dinámicamente. No se necesita adición manual.",
+    "noNewModel": "Para LMStudio, Ollama, Llamafile, obtenemos dinámicamente. No se necesita adición manual.",
     "searchModel": "Buscar Modelo",
     "selectAll": "Seleccionar Todo",
     "save": "Guardar",
diff --git a/src/assets/locale/fa/openai.json b/src/assets/locale/fa/openai.json
index 479e509..7258e83 100644
--- a/src/assets/locale/fa/openai.json
+++ b/src/assets/locale/fa/openai.json
@@ -42,7 +42,7 @@
     "delete": "حذف",
     "edit": "ویرایش",
     "newModel": "افزودن مدل‌ها به ارائه‌دهنده",
-    "noNewModel": "برای LMStudio، ما به صورت پویا دریافت می‌کنیم. نیازی به افزودن دستی نیست.",
+    "noNewModel": "برای LMStudio, Ollama, Llamafile, ما به صورت پویا دریافت می‌کنیم. نیازی به افزودن دستی نیست.",
     "searchModel": "جستجوی مدل",
     "selectAll": "انتخاب همه",
     "save": "ذخیره",
diff --git a/src/assets/locale/fr/openai.json b/src/assets/locale/fr/openai.json
index a71256c..2d9a0c2 100644
--- a/src/assets/locale/fr/openai.json
+++ b/src/assets/locale/fr/openai.json
@@ -42,7 +42,7 @@
     "delete": "Supprimer",
     "edit": "Modifier",
     "newModel": "Ajouter des modèles au fournisseur",
-    "noNewModel": "Pour LMStudio, nous récupérons dynamiquement. Pas besoin d'ajout manuel.",
+    "noNewModel": "Pour LMStudio, Ollama, Llamafile, nous récupérons dynamiquement. Pas besoin d'ajout manuel.",
     "searchModel": "Rechercher un modèle",
     "selectAll": "Tout sélectionner",
     "save": "Enregistrer",
diff --git a/src/assets/locale/it/openai.json b/src/assets/locale/it/openai.json
index c9d27ef..589da51 100644
--- a/src/assets/locale/it/openai.json
+++ b/src/assets/locale/it/openai.json
@@ -42,7 +42,7 @@
     "delete": "Elimina",
     "edit": "Modifica",
     "newModel": "Aggiungi Modelli al Provider",
-    "noNewModel": "Per LMStudio, recuperiamo dinamicamente. Non è necessaria l'aggiunta manuale.",
+    "noNewModel": "Per LMStudio, Ollama, Llamafile, recuperiamo dinamicamente. Non è necessaria l'aggiunta manuale.",
     "searchModel": "Cerca Modello",
     "selectAll": "Seleziona Tutto",
     "save": "Salva",
diff --git a/src/assets/locale/ja-JP/openai.json b/src/assets/locale/ja-JP/openai.json
index 829e886..2e755e8 100644
--- a/src/assets/locale/ja-JP/openai.json
+++ b/src/assets/locale/ja-JP/openai.json
@@ -42,7 +42,7 @@
     "delete": "削除",
     "edit": "編集",
     "newModel": "プロバイダーにモデルを追加",
-    "noNewModel": "LMStudioの場合、動的に取得します。手動での追加は不要です。",
+    "noNewModel": "LMStudio, Ollama, Llamafile,の場合、動的に取得します。手動での追加は不要です。",
     "searchModel": "モデルを検索",
     "selectAll": "すべて選択",
     "save": "保存",
diff --git a/src/assets/locale/ko/openai.json b/src/assets/locale/ko/openai.json
index 13bc431..fae6aca 100644
--- a/src/assets/locale/ko/openai.json
+++ b/src/assets/locale/ko/openai.json
@@ -42,7 +42,7 @@
     "delete": "삭제",
     "edit": "편집",
     "newModel": "공급자에 모델 추가",
-    "noNewModel": "LMStudio의 경우 동적으로 가져옵니다. 수동 추가는 필요하지 않습니다.",
+    "noNewModel": "LMStudio, Ollama, Llamafile,의 경우 동적으로 가져옵니다. 수동 추가는 필요하지 않습니다.",
     "searchModel": "모델 검색",
     "selectAll": "모두 선택",
     "save": "저장",
diff --git a/src/assets/locale/ml/openai.json b/src/assets/locale/ml/openai.json
index 6fe486e..5bbaaea 100644
--- a/src/assets/locale/ml/openai.json
+++ b/src/assets/locale/ml/openai.json
@@ -42,7 +42,7 @@
     "delete": "ഇല്ലാതാക്കുക",
     "edit": "തിരുത്തുക",
     "newModel": "ദാതാവിലേക്ക് മോഡലുകൾ ചേർക്കുക",
-    "noNewModel": "LMStudio-യ്ക്കായി, ഞങ്ങൾ ഡൈനാമിക്കായി ലഭ്യമാക്കുന്നു. മാനുവലായി ചേർക്കേണ്ടതില്ല.",
+    "noNewModel": "LMStudio, Ollama, Llamafile-യ്ക്കായി, ഞങ്ങൾ ഡൈനാമിക്കായി ലഭ്യമാക്കുന്നു. മാനുവലായി ചേർക്കേണ്ടതില്ല.",
     "searchModel": "മോഡൽ തിരയുക",
     "selectAll": "എല്ലാം തിരഞ്ഞെടുക്കുക",
     "save": "സംരക്ഷിക്കുക",
diff --git a/src/assets/locale/no/openai.json b/src/assets/locale/no/openai.json
index 91ddc3d..aa748bb 100644
--- a/src/assets/locale/no/openai.json
+++ b/src/assets/locale/no/openai.json
@@ -42,7 +42,7 @@
     "delete": "Slett",
     "edit": "Rediger",
     "newModel": "Legg til modeller for leverandør",
-    "noNewModel": "For LMStudio henter vi dynamisk. Ingen manuell tillegging nødvendig.",
+    "noNewModel": "For LMStudio, Ollama, Llamafile, henter vi dynamisk. Ingen manuell tillegging nødvendig.",
     "searchModel": "Søk etter modell",
     "selectAll": "Velg alle",
     "save": "Lagre",
diff --git a/src/assets/locale/ru/openai.json b/src/assets/locale/ru/openai.json
index 436c676..2b63545 100644
--- a/src/assets/locale/ru/openai.json
+++ b/src/assets/locale/ru/openai.json
@@ -42,7 +42,7 @@
     "delete": "Удалить",
     "edit": "Редактировать",
     "newModel": "Добавить модели к провайдеру",
-    "noNewModel": "Для LMStudio мы загружаем динамически. Ручное добавление не требуется.",
+    "noNewModel": "Для LMStudio, Ollama, Llamafile, мы загружаем динамически. Ручное добавление не требуется.",
     "searchModel": "Поиск модели",
     "selectAll": "Выбрать все",
     "save": "Сохранить",
diff --git a/src/assets/locale/zh/openai.json b/src/assets/locale/zh/openai.json
index 1b02184..00583f3 100644
--- a/src/assets/locale/zh/openai.json
+++ b/src/assets/locale/zh/openai.json
@@ -42,7 +42,7 @@
     "delete": "删除",
     "edit": "编辑",
     "newModel": "向提供商添加模型",
-    "noNewModel": "对于 LMStudio，我们动态获取。无需手动添加。",
+    "noNewModel": "对于 LMStudio, Ollama, Llamafile，我们动态获取。无需手动添加。",
     "searchModel": "搜索模型",
     "selectAll": "全选",
     "save": "保存",
diff --git a/src/components/Option/Settings/openai.tsx b/src/components/Option/Settings/openai.tsx
index a0807ed..b44694f 100644
--- a/src/components/Option/Settings/openai.tsx
+++ b/src/components/Option/Settings/openai.tsx
@@ -23,6 +23,7 @@ import {
 } from "lucide-react"
 import { OpenAIFetchModel } from "./openai-fetch-model"
 import { OAI_API_PROVIDERS } from "@/utils/oai-api-providers"
+const noPopupProvider = ["lmstudio", "llamafile", "ollama2"]
 
 export const OpenAIApp = () => {
   const { t } = useTranslation("openai")
@@ -47,7 +48,6 @@ export const OpenAIApp = () => {
       })
       setOpen(false)
       message.success(t("addSuccess"))
-      const noPopupProvider = ["lmstudio", "llamafile", "ollama2"]
       if (!noPopupProvider.includes(provider)) {
         setOpenaiId(data)
         setOpenModelModal(true)
@@ -158,7 +158,7 @@ export const OpenAIApp = () => {
 
                   <Tooltip
                     title={
-                      record.provider !== "lmstudio"
+                      !noPopupProvider.includes(record.provider)
                         ? t("newModel")
                         : t("noNewModel")
                     }>
@@ -168,7 +168,9 @@ export const OpenAIApp = () => {
                         setOpenModelModal(true)
                         setOpenaiId(record.id)
                       }}
-                      disabled={!record.id || record.provider === "lmstudio"}>
+                      disabled={
+                        !record.id || noPopupProvider.includes(record.provider)
+                      }>
                       <DownloadIcon className="size-4" />
                     </button>
                   </Tooltip>

From 55f3838b6de58aab5b20cc6fe6adf63723ea01af Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 10 Nov 2024 17:30:33 +0530
Subject: [PATCH 14/27] feat: Improve ollama2 model fetching

This commit introduces a more efficient approach to fetching ollama2 models, ensuring proper filtering and handling of providers. This enhances the robustness and reliability of the model loading process, streamlining the overall user experience.
---
 src/db/models.ts | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/db/models.ts b/src/db/models.ts
index 012379c..928beb9 100644
--- a/src/db/models.ts
+++ b/src/db/models.ts
@@ -407,6 +407,10 @@ export const ollamaFormatAllCustomModels = async (
     (provider) => provider.provider === "llamafile"
   )
 
+  const ollamaProviders = allProviders.filter(
+    (provider) => provider.provider === "ollama2"
+  )
+
   const lmModelsPromises = lmstudioProviders.map((provider) =>
     dynamicFetchLMStudio({
       baseUrl: provider.baseUrl,
@@ -421,12 +425,11 @@ export const ollamaFormatAllCustomModels = async (
     })
   )
 
-  const ollamaModelsPromises = allProviders.map((provider) => (
+  const ollamaModelsPromises = ollamaProviders.map((provider) =>
     dynamicFetchOllama2({
       baseUrl: provider.baseUrl,
       providerId: provider.id
-    })
-  ))
+    }))
 
   const lmModelsFetch = await Promise.all(lmModelsPromises)
 

From 0f75de02cbadbb5623176276528894e71aa01344 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 10 Nov 2024 18:08:05 +0530
Subject: [PATCH 15/27] Fix: Update temporary chat history

This commit addresses an issue where temporary chat history was not being updated correctly when using voice input.

The `setHistoryId` function is now called within the `saveMessageOnError` function to ensure that the history ID is set correctly when a message is saved.
---
 src/hooks/useMessageOption.tsx | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/hooks/useMessageOption.tsx b/src/hooks/useMessageOption.tsx
index 90fc06a..322287c 100644
--- a/src/hooks/useMessageOption.tsx
+++ b/src/hooks/useMessageOption.tsx
@@ -366,6 +366,8 @@ export const useMessageOption = () => {
   const saveMessageOnSuccess = async (e: any) => {
     if (!temporaryChat) {
       return await saveSuccess(e)
+    } else {
+      setHistoryId("temp")
     }
 
     return true
@@ -374,6 +376,21 @@ export const useMessageOption = () => {
   const saveMessageOnError = async (e: any) => {
     if (!temporaryChat) {
       return await saveError(e)
+    } else {
+      setHistory([
+        ...history,
+        {
+          role: "user",
+          content: e.userMessage,
+          image: e.image
+        },
+        {
+          role: "assistant",
+          content: e.botMessage
+        }
+      ])
+
+      setHistoryId("temp")
     }
 
     return true
@@ -1069,6 +1086,6 @@ export const useMessageOption = () => {
     setSelectedKnowledge,
     ttsEnabled,
     temporaryChat,
-    setTemporaryChat,
+    setTemporaryChat
   }
 }

From a96193bbf804e8bb8dadfbfa9372cccbddc1b5a4 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 10 Nov 2024 18:14:44 +0530
Subject: [PATCH 16/27] feat: Add error handling for updating message by index

Adds error handling to the `updateMessageByIndex` function to prevent the temporary chat from breaking when an error occurs during the update process. This ensures a more robust and reliable experience for users.
---
 src/db/index.ts | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/db/index.ts b/src/db/index.ts
index e4c136c..da837b2 100644
--- a/src/db/index.ts
+++ b/src/db/index.ts
@@ -355,10 +355,14 @@ export const updateMessageByIndex = async (
   index: number,
   message: string
 ) => {
+  try {
   const db = new PageAssitDatabase()
   const chatHistory = (await db.getChatHistory(history_id)).reverse()
   chatHistory[index].content = message
   await db.db.set({ [history_id]: chatHistory.reverse() })
+  } catch(e) {
+    // temp chat will break
+  }
 }
 
 export const deleteChatForEdit = async (history_id: string, index: number) => {

From 5678a0f8b288568ec782a4c09bfa6ea71c33dde3 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 10 Nov 2024 19:34:38 +0530
Subject: [PATCH 17/27] feat: Add Korean language support

Add Korean language support to the application. This includes translating the necessary UI elements and adding Korean to the supported language list.
---
 src/i18n/index.ts            |  4 +++-
 src/i18n/lang/ko.ts          | 19 +++++++++++++++++++
 src/i18n/support-language.ts |  4 ++++
 3 files changed, 26 insertions(+), 1 deletion(-)
 create mode 100644 src/i18n/lang/ko.ts

diff --git a/src/i18n/index.ts b/src/i18n/index.ts
index 97bbf6f..effddb4 100644
--- a/src/i18n/index.ts
+++ b/src/i18n/index.ts
@@ -14,6 +14,7 @@ import { de } from "./lang/de";
 import { da } from "./lang/da";
 import { no } from "./lang/no";
 import { sv } from "./lang/sv";
+import { ko } from "./lang/ko";
 
 
 i18n
@@ -37,7 +38,8 @@ i18n
             da: da,
             no: no,
             de: de,
-            sv: sv
+            sv: sv,
+            ko: ko,
         },
         fallbackLng: "en",
         lng: localStorage.getItem("i18nextLng") || "en",
diff --git a/src/i18n/lang/ko.ts b/src/i18n/lang/ko.ts
new file mode 100644
index 0000000..037ce8f
--- /dev/null
+++ b/src/i18n/lang/ko.ts
@@ -0,0 +1,19 @@
+import option from "@/assets/locale/ko/option.json";
+import playground from "@/assets/locale/ko/playground.json";
+import common from "@/assets/locale/ko/common.json";
+import sidepanel from "@/assets/locale/ko/sidepanel.json";
+import settings from "@/assets/locale/ko/settings.json";
+import knowledge from "@/assets/locale/ko/knowledge.json";
+import chrome from "@/assets/locale/ko/chrome.json";
+import openai from "@/assets/locale/ko/openai.json";
+
+export const ko = {
+    option,
+    playground,
+    common,
+    sidepanel,
+    settings,
+    knowledge,
+    chrome,
+    openai
+}
\ No newline at end of file
diff --git a/src/i18n/support-language.ts b/src/i18n/support-language.ts
index 2c62ad4..273964d 100644
--- a/src/i18n/support-language.ts
+++ b/src/i18n/support-language.ts
@@ -55,5 +55,9 @@ export const supportLanguage = [
     {
         value: "sv",
         label: "Svenska"
+    },
+    {
+        value: "ko",
+        label: "한국어"
     }
 ]

From 4292dc45ea3c180dc11e4a345b3efaae4891f3ad Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 16 Nov 2024 15:42:31 +0530
Subject: [PATCH 18/27] feat: Bump application version to 1.3.5

Updates the application version in the wxt.config.ts file from 1.3.4 to 1.3.5.
---
 wxt.config.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wxt.config.ts b/wxt.config.ts
index 288a555..18b2a44 100644
--- a/wxt.config.ts
+++ b/wxt.config.ts
@@ -50,7 +50,7 @@ export default defineConfig({
   outDir: "build",
 
   manifest: {
-    version: "1.3.4",
+    version: "1.3.5",
     name:
       process.env.TARGET === "firefox"
         ? "Page Assist - A Web UI for Local AI Models"

From c4d9e3aeed5b39a07e555c415ca22addc52cffd4 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 16 Nov 2024 15:50:11 +0530
Subject: [PATCH 19/27] feat: Add page assist select component to header

This commit introduces a new `PageAssistSelect` component to the header, which replaces the previous `Select` component for selecting the active chat model. The new component provides improved functionality, including:

- Ability to display provider icons alongside the model name
- Truncation of long model names to ensure they fit within the available space
- Improved loading state handling
- Ability to refresh the model list on demand

These changes enhance the user experience and make it easier for users to quickly select the desired chat model.
---
 src/components/Layouts/Header.tsx          |  45 ++-
 src/components/Select/LoadingIndicator.tsx |  27 ++
 src/components/Select/index.tsx            | 303 +++++++++++++++++++++
 3 files changed, 350 insertions(+), 25 deletions(-)
 create mode 100644 src/components/Select/LoadingIndicator.tsx
 create mode 100644 src/components/Select/index.tsx

diff --git a/src/components/Layouts/Header.tsx b/src/components/Layouts/Header.tsx
index 7f3e153..3e4a1f6 100644
--- a/src/components/Layouts/Header.tsx
+++ b/src/components/Layouts/Header.tsx
@@ -22,6 +22,7 @@ import { getAllPrompts } from "@/db"
 import { ShareBtn } from "~/components/Common/ShareBtn"
 import { ProviderIcons } from "../Common/ProviderIcon"
 import { NewChat } from "./NewChat"
+import { PageAssistSelect } from "../Select"
 type Props = {
   setSidebarOpen: (open: boolean) => void
   setOpenModelSettings: (open: boolean) => void
@@ -49,14 +50,10 @@ export const Header: React.FC<Props> = ({
     historyId,
     temporaryChat
   } = useMessageOption()
-  const {
-    data: models,
-    isLoading: isModelsLoading,
-  } = useQuery({
+  const { data: models, isLoading: isModelsLoading, refetch } = useQuery({
     queryKey: ["fetchModel"],
     queryFn: () => fetchChatModels({ returnEmpty: true }),
-    refetchInterval: 15_000,
-    refetchIntervalInBackground: true,
+    refetchIntervalInBackground: false,
     placeholderData: (prev) => prev
   })
 
@@ -87,9 +84,10 @@ export const Header: React.FC<Props> = ({
   }
 
   return (
-    <div className={`sticky top-0 z-[999] flex h-16 p-3  bg-gray-50 border-b  dark:bg-[#171717] dark:border-gray-600 ${
-      temporaryChat && "!bg-gray-200 dark:!bg-black"
-    }`}>
+    <div
+      className={`sticky top-0 z-[999] flex h-16 p-3  bg-gray-50 border-b  dark:bg-[#171717] dark:border-gray-600 ${
+        temporaryChat && "!bg-gray-200 dark:!bg-black"
+      }`}>
       <div className="flex gap-2 items-center">
         {pathname !== "/" && (
           <div>
@@ -107,41 +105,38 @@ export const Header: React.FC<Props> = ({
             <PanelLeftIcon className="w-6 h-6" />
           </button>
         </div>
-        <NewChat 
-          clearChat={clearChat}
-        />
+        <NewChat clearChat={clearChat} />
         <span className="text-lg font-thin text-zinc-300 dark:text-zinc-600">
           {"/"}
         </span>
         <div className="hidden lg:block">
-          <Select
+          <PageAssistSelect
+            className="w-80"
+            placeholder={t("common:selectAModel")}
             value={selectedModel}
             onChange={(e) => {
-              setSelectedModel(e)
-              localStorage.setItem("selectedModel", e)
+              setSelectedModel(e.value)
+              localStorage.setItem("selectedModel", e.value)
             }}
-            size="large"
-            loading={isModelsLoading}
-            filterOption={(input, option) =>
-              option.label.key.toLowerCase().indexOf(input.toLowerCase()) >= 0
-            }
-            showSearch
-            placeholder={t("common:selectAModel")}
-            className="w-72"
+            isLoading={isModelsLoading}
             options={models?.map((model) => ({
               label: (
                 <span
                   key={model.model}
-                  className="flex flex-row gap-3 items-center truncate">
+                  className="flex flex-row gap-3 items-center ">
                   <ProviderIcons
                     provider={model?.provider}
                     className="w-5 h-5"
                   />
-                  <span className="truncate">{model.name}</span>
+                  <span className="line-clamp-2">{model.name}</span>
                 </span>
               ),
               value: model.model
             }))}
+
+            onRefresh={() => {
+              refetch()
+            }}
           />
         </div>
         <div className="lg:hidden">
diff --git a/src/components/Select/LoadingIndicator.tsx b/src/components/Select/LoadingIndicator.tsx
new file mode 100644
index 0000000..dc06722
--- /dev/null
+++ b/src/components/Select/LoadingIndicator.tsx
@@ -0,0 +1,27 @@
+import React from "react"
+
+export const LoadingIndicator: React.FC<{ className?: string }> = ({
+  className = ""
+}) => (
+  <div className={`animate-spin ${className}`}>
+    <svg
+      className="w-4 h-4"
+      viewBox="0 0 24 24"
+      fill="none"
+      xmlns="http://www.w3.org/2000/svg">
+      <circle
+        className="opacity-25"
+        cx="12"
+        cy="12"
+        r="10"
+        stroke="currentColor"
+        strokeWidth="4"
+      />
+      <path
+        className="opacity-75"
+        fill="currentColor"
+        d="M4 12a8 8 0 018-8V0C5.373 0 0 5.373 0 12h4zm2 5.291A7.962 7.962 0 014 12H0c0 3.042 1.135 5.824 3 7.938l3-2.647z"
+      />
+    </svg>
+  </div>
+)
diff --git a/src/components/Select/index.tsx b/src/components/Select/index.tsx
new file mode 100644
index 0000000..6ccfeda
--- /dev/null
+++ b/src/components/Select/index.tsx
@@ -0,0 +1,303 @@
+import React, { useState, useRef, useEffect, useMemo } from "react"
+import { Search, RotateCw, ChevronDown } from "lucide-react"
+import { LoadingIndicator } from "./LoadingIndicator"
+import { Empty } from "antd"
+
+export interface SelectOption {
+  label: string | JSX.Element
+  value: string
+}
+
+interface SelectProps {
+  options: SelectOption[]
+  value?: string
+  onChange: (option: SelectOption) => void
+  placeholder?: string
+  onRefresh?: () => void
+  className?: string
+  dropdownClassName?: string
+  optionClassName?: string
+  searchClassName?: string
+  disabled?: boolean
+  isLoading?: boolean
+  loadingText?: string
+  filterOption?: (input: string, option: SelectOption) => boolean
+}
+
+export const PageAssistSelect: React.FC<SelectProps> = ({
+  options = [],
+  value,
+  onChange,
+  placeholder = "Select an option",
+  onRefresh,
+  className = "",
+  dropdownClassName = "",
+  optionClassName = "",
+  searchClassName = "",
+  disabled = false,
+  isLoading = false,
+  loadingText = "Loading...",
+  filterOption
+}) => {
+  const [isOpen, setIsOpen] = useState(false)
+  const [searchTerm, setSearchTerm] = useState("")
+  const [filteredOptions, setFilteredOptions] = useState<SelectOption[]>([])
+  const containerRef = useRef<HTMLDivElement>(null)
+  const [activeIndex, setActiveIndex] = useState(-1)
+
+  useEffect(() => {
+    const handleClickOutside = (event: MouseEvent) => {
+      if (
+        containerRef.current &&
+        !containerRef.current.contains(event.target as Node)
+      ) {
+        setIsOpen(false)
+      }
+    }
+
+    document.addEventListener("mousedown", handleClickOutside)
+    return () => document.removeEventListener("mousedown", handleClickOutside)
+  }, [])
+
+  useEffect(() => {
+    if (!options) return
+
+    const filtered = options.filter((option) => {
+      if (!searchTerm) return true
+
+      if (filterOption) {
+        return filterOption(searchTerm, option)
+      }
+
+      if (typeof option.label === "string") {
+        return option.label.toLowerCase().includes(searchTerm.toLowerCase())
+      }
+
+      if (React.isValidElement(option.label)) {
+        const textContent = extractTextFromJSX(option.label)
+        return textContent.toLowerCase().includes(searchTerm.toLowerCase())
+      }
+
+      return false
+    })
+    setFilteredOptions(filtered)
+    setActiveIndex(-1)
+  }, [searchTerm, options, filterOption])
+
+  const extractTextFromJSX = (element: React.ReactElement): string => {
+    if (typeof element.props.children === "string") {
+      return element.props.children
+    }
+
+    if (Array.isArray(element.props.children)) {
+      return element.props.children
+        .map((child) => {
+          if (typeof child === "string") return child
+          if (React.isValidElement(child)) return extractTextFromJSX(child)
+          return ""
+        })
+        .join(" ")
+    }
+
+    if (React.isValidElement(element.props.children)) {
+      return extractTextFromJSX(element.props.children)
+    }
+
+    return ""
+  }
+
+  const handleRefresh = (e: React.MouseEvent) => {
+    e.stopPropagation()
+    onRefresh?.()
+  }
+
+  const handleKeyDown = (e: React.KeyboardEvent) => {
+    if (disabled || isLoading) return
+
+    switch (e.key) {
+      case "Enter":
+        if (isOpen && activeIndex >= 0) {
+          e.preventDefault()
+          const selectedOption = filteredOptions[activeIndex]
+          if (selectedOption) {
+            onChange(selectedOption)
+            setIsOpen(false)
+            setSearchTerm("")
+          }
+        } else {
+          setIsOpen(!isOpen)
+        }
+        break
+      case " ":
+        if (!isOpen) {
+          e.preventDefault()
+          setIsOpen(true)
+        }
+        break
+      case "Escape":
+        setIsOpen(false)
+        break
+      case "ArrowDown":
+        e.preventDefault()
+        if (!isOpen) {
+          setIsOpen(true)
+        } else {
+          setActiveIndex((prev) =>
+            prev < filteredOptions.length - 1 ? prev + 1 : prev
+          )
+        }
+        break
+      case "ArrowUp":
+        e.preventDefault()
+        setActiveIndex((prev) => (prev > 0 ? prev - 1 : prev))
+        break
+    }
+  }
+
+  const defaultSelectClass = `
+    flex items-center justify-between p-2.5 rounded-lg border
+    ${disabled || isLoading ? "cursor-not-allowed opacity-50" : "cursor-pointer"}
+    ${isOpen ? "ring-2 ring-blue-500" : ""}
+    bg-transparent border-gray-200 text-gray-900
+    transition-all duration-200
+    dark:text-white
+    dark:border-[#353534]
+  `
+
+  const defaultDropdownClass = `
+    absolute z-50 w-full mt-1 bg-white dark:bg-[#1e1e1f] dark:text-white rounded-lg shadow-lg 
+    border border-gray-200 dark:border-[#353534]
+  `
+
+  const defaultSearchClass = `
+    w-full pl-8 pr-8 py-1.5 rounded-md
+    bg-gray-50 border border-gray-200
+    focus:outline-none focus:ring-2 focus:ring-gray-100
+    text-gray-900
+    dark:bg-[#1e1e1f] dark:text-white
+    dark:border-[#353534]
+    dark:focus:ring-gray-700
+    dark:focus:border-gray-700
+    dark:placeholder-gray-400
+    dark:bg-opacity-90
+    dark:hover:bg-opacity-100
+    dark:focus:bg-opacity-100
+    dark:hover:border-gray-700
+    dark:hover:bg-[#2a2a2b]
+    dark:focus:bg-[#2a2a2b]
+  `
+
+  const defaultOptionClass = `
+    p-2 cursor-pointer transition-colors duration-150
+  `
+
+  const selectedOption = useMemo(() => {
+    if (!value || !options) return null
+    return options?.find(opt => opt.value === value)
+  }, [value, options])
+
+  if (!options) {
+    return (
+      <div className={`relative w-full ${className}`}>
+        <div className={`${defaultSelectClass} ${className}`}>
+          <LoadingIndicator />
+          <span>{loadingText}</span>
+        </div>
+      </div>
+    )
+  }
+
+  return (
+    <div ref={containerRef} className={`relative w-full ${className}`}>
+      <div
+        role="combobox"
+        aria-expanded={isOpen}
+        aria-haspopup="listbox"
+        aria-controls="select-dropdown"
+        aria-label={placeholder}
+        tabIndex={disabled ? -1 : 0}
+        onClick={() => !disabled && !isLoading && setIsOpen(!isOpen)}
+        onKeyDown={handleKeyDown}
+        className={`${defaultSelectClass} ${className}`}>
+        <span className="!truncate flex items-center gap-2 ">
+          {isLoading && <LoadingIndicator />}
+          {isLoading ? loadingText : selectedOption ? selectedOption.label : <span className="dark:text-gray-400 text-sm">{placeholder}</span>}
+        </span>
+        <ChevronDown
+          aria-hidden="true"
+          className={`w-4 h-4 transition-transform duration-200 ${
+            isOpen ? "transform rotate-180" : ""
+          }`}
+        />
+      </div>
+
+      {isOpen && (
+        <div 
+          id="select-dropdown"
+          role="listbox"
+          className={`${defaultDropdownClass} ${dropdownClassName}`}>
+          <div className="p-2 border-b border-gray-200 dark:border-[#353534]">
+            <div className="relative">
+              <input
+                type="text"
+                value={searchTerm}
+                onChange={(e) => setSearchTerm(e.target.value)}
+                placeholder="Search..."
+                className={`${defaultSearchClass} ${searchClassName}`}
+                disabled={isLoading}
+                aria-label="Search options"
+              />
+              <Search aria-hidden="true" className="absolute left-2 top-1/2 transform -translate-y-1/2 w-4 h-4 text-gray-400" />
+              {onRefresh && (
+                <button
+                  onClick={handleRefresh}
+                  disabled={isLoading}
+                  aria-label="Refresh options"
+                  className={`absolute right-2 top-1/2 transform -translate-y-1/2
+                    hover:text-blue-500 dark:hover:text-blue-400 transition-colors duration-200
+                    ${isLoading ? "cursor-not-allowed opacity-50" : ""}`}>
+                  <RotateCw
+                    aria-hidden="true"
+                    className={`w-4 h-4 dark:text-gray-400 ${isLoading ? "animate-spin" : ""}`}
+                  />
+                </button>
+              )}{" "}
+            </div>
+          </div>
+
+          <div className="max-h-60 overflow-y-auto custom-scrollbar">
+            {isLoading ? (
+              <div className="p-4 text-center text-gray-500 flex items-center justify-center gap-2">
+                <LoadingIndicator />
+                <span>{loadingText}</span>
+              </div>
+            ) : filteredOptions.length === 0 ? (
+              <div className="p-6">
+                <Empty />
+              </div>
+            ) : (
+              filteredOptions.map((option, index) => (
+                <div
+                  key={option.value}
+                  role="option"
+                  aria-selected={value === option.value}
+                  onClick={() => {
+                    onChange(option)
+                    setIsOpen(false)
+                    setSearchTerm("")
+                  }}
+                  className={`
+                    ${defaultOptionClass}
+                    ${value === option.value ? "bg-blue-50 dark:bg-[#262627]" : "hover:bg-gray-100 dark:hover:bg-[#272728]"}
+                    ${activeIndex === index ? "bg-gray-100 dark:bg-[#272728]" : ""}
+                    ${optionClassName}`}>
+                  {option.label}
+                </div>
+              ))
+            )}
+          </div>
+        </div>
+      )}
+    </div>
+  )
+}
\ No newline at end of file

From 726d3e34274834126c3983f5aec1b389e323c823 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 16 Nov 2024 19:33:04 +0530
Subject: [PATCH 20/27] feat: Improve PageAssistSelect component

This commit enhances the `PageAssistSelect` component with the following improvements:

- Adds a `ref` to the options container to automatically scroll to the selected option when the dropdown is opened.
- Fixes an issue where the `selectedOption` was not being correctly determined when the `options` array was updated.
- Improves the rendering of the selected option, ensuring that the loading state, placeholder, and selected option are displayed correctly.
- Adds a `data-value` attribute to the option elements to facilitate scrolling to the selected option.

These changes improve the overall user experience and functionality of the `PageAssistSelect` component.
---
 src/components/Select/index.tsx | 40 ++++++++++++++++++++++++++++-----
 1 file changed, 34 insertions(+), 6 deletions(-)

diff --git a/src/components/Select/index.tsx b/src/components/Select/index.tsx
index 6ccfeda..e3da503 100644
--- a/src/components/Select/index.tsx
+++ b/src/components/Select/index.tsx
@@ -43,6 +43,7 @@ export const PageAssistSelect: React.FC<SelectProps> = ({
   const [searchTerm, setSearchTerm] = useState("")
   const [filteredOptions, setFilteredOptions] = useState<SelectOption[]>([])
   const containerRef = useRef<HTMLDivElement>(null)
+  const optionsContainerRef = useRef<HTMLDivElement>(null)
   const [activeIndex, setActiveIndex] = useState(-1)
 
   useEffect(() => {
@@ -59,6 +60,21 @@ export const PageAssistSelect: React.FC<SelectProps> = ({
     return () => document.removeEventListener("mousedown", handleClickOutside)
   }, [])
 
+  useEffect(() => {
+    try {
+      if (isOpen && optionsContainerRef.current && value) {
+        const selectedOptionElement = optionsContainerRef.current.querySelector(
+          `[data-value="${value}"]`
+        )
+        if (selectedOptionElement) {
+          selectedOptionElement.scrollIntoView({ block: "nearest" })
+        }
+      }
+    } catch (error) {
+      console.error("Error scrolling to selected option:", error)
+    }
+  }, [isOpen, value])
+
   useEffect(() => {
     if (!options) return
 
@@ -193,7 +209,7 @@ export const PageAssistSelect: React.FC<SelectProps> = ({
 
   const selectedOption = useMemo(() => {
     if (!value || !options) return null
-    return options?.find(opt => opt.value === value)
+    return options?.find((opt) => opt.value === value)
   }, [value, options])
 
   if (!options) {
@@ -221,7 +237,13 @@ export const PageAssistSelect: React.FC<SelectProps> = ({
         className={`${defaultSelectClass} ${className}`}>
         <span className="!truncate flex items-center gap-2 ">
           {isLoading && <LoadingIndicator />}
-          {isLoading ? loadingText : selectedOption ? selectedOption.label : <span className="dark:text-gray-400 text-sm">{placeholder}</span>}
+          {isLoading ? (
+            loadingText
+          ) : selectedOption ? (
+            selectedOption.label
+          ) : (
+            <span className="dark:text-gray-400 text-sm">{placeholder}</span>
+          )}
         </span>
         <ChevronDown
           aria-hidden="true"
@@ -232,7 +254,7 @@ export const PageAssistSelect: React.FC<SelectProps> = ({
       </div>
 
       {isOpen && (
-        <div 
+        <div
           id="select-dropdown"
           role="listbox"
           className={`${defaultDropdownClass} ${dropdownClassName}`}>
@@ -247,7 +269,10 @@ export const PageAssistSelect: React.FC<SelectProps> = ({
                 disabled={isLoading}
                 aria-label="Search options"
               />
-              <Search aria-hidden="true" className="absolute left-2 top-1/2 transform -translate-y-1/2 w-4 h-4 text-gray-400" />
+              <Search
+                aria-hidden="true"
+                className="absolute left-2 top-1/2 transform -translate-y-1/2 w-4 h-4 text-gray-400"
+              />
               {onRefresh && (
                 <button
                   onClick={handleRefresh}
@@ -265,7 +290,9 @@ export const PageAssistSelect: React.FC<SelectProps> = ({
             </div>
           </div>
 
-          <div className="max-h-60 overflow-y-auto custom-scrollbar">
+          <div
+            ref={optionsContainerRef}
+            className="max-h-60 overflow-y-auto custom-scrollbar">
             {isLoading ? (
               <div className="p-4 text-center text-gray-500 flex items-center justify-center gap-2">
                 <LoadingIndicator />
@@ -281,6 +308,7 @@ export const PageAssistSelect: React.FC<SelectProps> = ({
                   key={option.value}
                   role="option"
                   aria-selected={value === option.value}
+                  data-value={option.value}
                   onClick={() => {
                     onChange(option)
                     setIsOpen(false)
@@ -300,4 +328,4 @@ export const PageAssistSelect: React.FC<SelectProps> = ({
       )}
     </div>
   )
-}
\ No newline at end of file
+}

From 64e88bd4938db94de6443f73f8cef5e06c56af8a Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 16 Nov 2024 19:33:20 +0530
Subject: [PATCH 21/27] feat: Add support for chat mode in MessageSource
 component

This commit extends the `MessageSource` component to support the `chat` mode in addition to the existing `rag` mode. This allows the component to handle and display messages from both chat and RAG (Rapid Automated Generation) sources.
---
 src/components/Common/Playground/MessageSource.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/components/Common/Playground/MessageSource.tsx b/src/components/Common/Playground/MessageSource.tsx
index d0a006a..b0d3ad9 100644
--- a/src/components/Common/Playground/MessageSource.tsx
+++ b/src/components/Common/Playground/MessageSource.tsx
@@ -13,7 +13,7 @@ type Props = {
 }
 
 export const MessageSource: React.FC<Props> = ({ source, onSourceClick }) => {
-  if (source?.mode === "rag") {
+  if (source?.mode === "rag" || source?.mode === "chat") {
     return (
       <button
         onClick={() => {

From ca26e059eb12ecda65d4da8ae1f15f0b0a12e794 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 16 Nov 2024 19:33:51 +0530
Subject: [PATCH 22/27] feat: Improve memory embedding and vector store
 handling

This commit includes the following improvements:

- Update the `memoryEmbedding` function to use the `PAMemoryVectorStore` instead of the generic `MemoryVectorStore`. This ensures that the vector store is specifically designed for the Page Assist application.
- Modify the `useMessage` hook to use the `PAMemoryVectorStore` type for the `keepTrackOfEmbedding` state.
- Update the `rerankDocs` function to use the `EmbeddingsInterface` type instead of the deprecated `Embeddings` type.
- Add a new `PageAssistVectorStore` class that extends the `VectorStore` interface and provides a custom implementation for the Page Assist application.

These changes improve the handling of memory embeddings and vector stores, ensuring better compatibility and performance within the Page Assist application.
---
 src/hooks/useMessage.tsx          |   8 ++-
 src/libs/PAMemoryVectorStore.ts   | 104 ++++++++++++++++++++++++++++++
 src/libs/PageAssistVectorStore.ts |   2 -
 src/utils/memory-embeddings.ts    |   8 +--
 src/utils/rerank.ts               |   5 +-
 5 files changed, 116 insertions(+), 11 deletions(-)
 create mode 100644 src/libs/PAMemoryVectorStore.ts

diff --git a/src/hooks/useMessage.tsx b/src/hooks/useMessage.tsx
index d06102d..6fb2175 100644
--- a/src/hooks/useMessage.tsx
+++ b/src/hooks/useMessage.tsx
@@ -34,6 +34,8 @@ import { pageAssistModel } from "@/models"
 import { getPrompt } from "@/services/application"
 import { humanMessageFormatter } from "@/utils/human-message"
 import { pageAssistEmbeddingModel } from "@/models/embedding"
+import { PageAssistVectorStore } from "@/libs/PageAssistVectorStore"
+import { PAMemoryVectorStore } from "@/libs/PAMemoryVectorStore"
 
 export const useMessage = () => {
   const {
@@ -90,7 +92,7 @@ export const useMessage = () => {
   )
 
   const [keepTrackOfEmbedding, setKeepTrackOfEmbedding] = React.useState<{
-    [key: string]: MemoryVectorStore
+    [key: string]: PAMemoryVectorStore
   }>({})
 
   const clearChat = () => {
@@ -177,7 +179,7 @@ export const useMessage = () => {
     let embedURL: string, embedHTML: string, embedType: string
     let embedPDF: { content: string; page: number }[] = []
 
-    let isAlreadyExistEmbedding: MemoryVectorStore
+    let isAlreadyExistEmbedding: PAMemoryVectorStore
     const {
       content: html,
       url: websiteUrl,
@@ -212,7 +214,7 @@ export const useMessage = () => {
         currentChatModelSettings?.keepAlive ??
         userDefaultModelSettings?.keepAlive
     })
-    let vectorstore: MemoryVectorStore
+    let vectorstore: PAMemoryVectorStore
 
     try {
       if (isAlreadyExistEmbedding) {
diff --git a/src/libs/PAMemoryVectorStore.ts b/src/libs/PAMemoryVectorStore.ts
new file mode 100644
index 0000000..3c30083
--- /dev/null
+++ b/src/libs/PAMemoryVectorStore.ts
@@ -0,0 +1,104 @@
+
+import { similarity as ml_distance_similarity } from "ml-distance"
+import { VectorStore } from "@langchain/core/vectorstores"
+import type { EmbeddingsInterface } from "@langchain/core/embeddings"
+import { Document, DocumentInterface } from "@langchain/core/documents"
+import { rerankDocs } from "../utils/rerank"
+
+interface MemoryVector {
+    content: string
+    embedding: number[]
+    metadata: Record<string, any>
+}
+
+interface MemoryVectorStoreArgs {
+    similarity?: typeof ml_distance_similarity.cosine
+}
+
+export class PAMemoryVectorStore extends VectorStore {
+
+
+    declare FilterType: (doc: Document) => boolean
+
+    private memoryVectors: MemoryVector[] = []
+    private similarity: typeof ml_distance_similarity.cosine
+
+    constructor(embeddings: EmbeddingsInterface, args?: MemoryVectorStoreArgs) {
+        super(embeddings, args)
+        this.similarity = args?.similarity ?? ml_distance_similarity.cosine
+    }
+
+    _vectorstoreType(): string {
+        return "memory"
+    }
+
+    async addVectors(vectors: number[][], documents: DocumentInterface[], options?: { [x: string]: any }): Promise<void> {
+        const memoryVectors = documents.map((doc, index) => ({
+            content: doc.pageContent,
+            embedding: vectors[index],
+            metadata: doc.metadata
+        }))
+
+        this.memoryVectors.push(...memoryVectors)
+    }
+    similaritySearchVectorWithScore(query: number[], k: number, filter?: this["FilterType"]): Promise<[DocumentInterface, number][]> {
+        throw new Error("Method not implemented.")
+    }
+
+    async addDocuments(documents: Document[]): Promise<void> {
+        const texts = documents.map((doc) => doc.pageContent)
+        const embeddings = await this.embeddings.embedDocuments(texts)
+        await this.addVectors(embeddings, documents)
+    }
+
+    async similaritySearch(query: string, k = 4): Promise<Document[]> {
+        const queryEmbedding = await this.embeddings.embedQuery(query)
+
+        const similarities = this.memoryVectors.map((vector) => ({
+            similarity: this.similarity(queryEmbedding, vector.embedding),
+            document: vector
+        }))
+
+        similarities.sort((a, b) => b.similarity - a.similarity)
+        const topK = similarities.slice(0, k)
+
+        const docs = topK.map(({ document }) =>
+            new Document({
+                pageContent: document.content,
+                metadata: document.metadata
+            })
+        )
+
+        return docs
+    }
+
+    async similaritySearchWithScore(query: string, k = 4): Promise<[Document, number][]> {
+        const queryEmbedding = await this.embeddings.embedQuery(query)
+
+        const similarities = this.memoryVectors.map((vector) => ({
+            similarity: this.similarity(queryEmbedding, vector.embedding),
+            document: vector
+        }))
+
+        similarities.sort((a, b) => b.similarity - a.similarity)
+        const topK = similarities.slice(0, k)
+
+        return topK.map(({ document, similarity }) => [
+            new Document({
+                pageContent: document.content,
+                metadata: document.metadata
+            }),
+            similarity
+        ])
+    }
+
+    static async fromDocuments(
+        docs: Document[],
+        embeddings: EmbeddingsInterface,
+        args?: MemoryVectorStoreArgs
+    ): Promise<PAMemoryVectorStore> {
+        const store = new PAMemoryVectorStore(embeddings, args)
+        await store.addDocuments(docs)
+        return store
+    }
+}
diff --git a/src/libs/PageAssistVectorStore.ts b/src/libs/PageAssistVectorStore.ts
index fd1d4f6..d52425e 100644
--- a/src/libs/PageAssistVectorStore.ts
+++ b/src/libs/PageAssistVectorStore.ts
@@ -3,8 +3,6 @@ import { VectorStore } from "@langchain/core/vectorstores"
 import type { EmbeddingsInterface } from "@langchain/core/embeddings"
 import { Document } from "@langchain/core/documents"
 import { getVector, insertVector } from "@/db/vector"
-import { cp } from "fs"
-
 /**
  * Interface representing a vector in memory. It includes the content
  * (text), the corresponding embedding (vector), and any associated
diff --git a/src/utils/memory-embeddings.ts b/src/utils/memory-embeddings.ts
index 99d2be4..9cb1b82 100644
--- a/src/utils/memory-embeddings.ts
+++ b/src/utils/memory-embeddings.ts
@@ -1,12 +1,12 @@
 import { PageAssistHtmlLoader } from "~/loader/html"
 import { RecursiveCharacterTextSplitter } from "langchain/text_splitter"
-import { MemoryVectorStore } from "langchain/vectorstores/memory"
 
 import {
   defaultEmbeddingChunkOverlap,
   defaultEmbeddingChunkSize
 } from "@/services/ollama"
 import { PageAssistPDFLoader } from "@/loader/pdf"
+import { PAMemoryVectorStore } from "@/libs/PAMemoryVectorStore"
 
 export const getLoader = ({
   html,
@@ -46,10 +46,10 @@ export const memoryEmbedding = async ({
   html: string
   type: string
   pdf: { content: string; page: number }[]
-  keepTrackOfEmbedding: Record<string, MemoryVectorStore>
+  keepTrackOfEmbedding: Record<string, PAMemoryVectorStore>
   ollamaEmbedding: any
   setIsEmbedding: (value: boolean) => void
-  setKeepTrackOfEmbedding: (value: Record<string, MemoryVectorStore>) => void
+  setKeepTrackOfEmbedding: (value: Record<string, PAMemoryVectorStore>) => void
 }) => {
   setIsEmbedding(true)
   const loader = getLoader({ html, pdf, type, url })
@@ -63,7 +63,7 @@ export const memoryEmbedding = async ({
 
   const chunks = await textSplitter.splitDocuments(docs)
 
-  const store = new MemoryVectorStore(ollamaEmbedding)
+  const store = new PAMemoryVectorStore(ollamaEmbedding)
 
   await store.addDocuments(chunks)
   setKeepTrackOfEmbedding({
diff --git a/src/utils/rerank.ts b/src/utils/rerank.ts
index fcf30cf..e24f7c8 100644
--- a/src/utils/rerank.ts
+++ b/src/utils/rerank.ts
@@ -1,4 +1,4 @@
-import type { Embeddings } from "@langchain/core/embeddings"
+import type { EmbeddingsInterface } from "@langchain/core/embeddings"
 import type { Document } from "@langchain/core/documents"
 import * as ml_distance from "ml-distance"
 
@@ -9,7 +9,7 @@ export const rerankDocs = async ({
 }: {
   query: string
   docs: Document[]
-  embedding: Embeddings
+  embedding: EmbeddingsInterface
 }) => {
   if (docs.length === 0) {
     return docs
@@ -34,6 +34,7 @@ export const rerankDocs = async ({
     }
   })
 
+  console.log("similarity", similarity)
   const sortedDocs = similarity
     .sort((a, b) => b.similarity - a.similarity)
     .filter((sim) => sim.similarity > 0.5)

From 92013f3bfcb17bba886c297c2fb6c6b9f48f6d87 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 17 Nov 2024 12:26:14 +0530
Subject: [PATCH 23/27] feat(settings): add Ollama connection status check
 setting - Add new setting to enable/disable Ollama connection status check -
 Update translations for the new setting across all supported languages

---
 src/assets/locale/da/playground.json          |  3 +-
 src/assets/locale/da/settings.json            |  3 ++
 src/assets/locale/de/playground.json          |  3 +-
 src/assets/locale/de/settings.json            |  3 ++
 src/assets/locale/en/playground.json          |  3 +-
 src/assets/locale/en/settings.json            |  3 ++
 src/assets/locale/es/playground.json          |  3 +-
 src/assets/locale/es/settings.json            |  3 ++
 src/assets/locale/fa/playground.json          |  3 +-
 src/assets/locale/fa/settings.json            |  3 ++
 src/assets/locale/fr/playground.json          |  3 +-
 src/assets/locale/fr/settings.json            |  3 ++
 src/assets/locale/it/playground.json          |  3 +-
 src/assets/locale/it/settings.json            |  3 ++
 src/assets/locale/ja-JP/playground.json       |  3 +-
 src/assets/locale/ja-JP/settings.json         |  3 ++
 src/assets/locale/ko/playground.json          |  3 +-
 src/assets/locale/ko/settings.json            |  4 +-
 src/assets/locale/ml/playground.json          |  3 +-
 src/assets/locale/ml/settings.json            |  3 ++
 src/assets/locale/no/playground.json          |  3 +-
 src/assets/locale/no/settings.json            |  3 ++
 src/assets/locale/pt-BR/playground.json       |  3 +-
 src/assets/locale/pt-BR/settings.json         |  3 ++
 src/assets/locale/ru/playground.json          |  3 +-
 src/assets/locale/ru/settings.json            |  3 ++
 src/assets/locale/sv/playground.json          |  3 +-
 src/assets/locale/sv/settings.json            |  3 ++
 src/assets/locale/zh/playground.json          |  3 +-
 src/assets/locale/zh/settings.json            |  3 ++
 src/components/Layouts/Header.tsx             |  1 +
 .../Option/Playground/PlaygroundEmpty.tsx     | 31 +++++++++++----
 .../Option/Settings/general-settings.tsx      | 29 +++++++++++---
 src/components/Select/index.tsx               |  8 ++--
 src/components/Sidepanel/Chat/empty.tsx       |  9 +++--
 src/services/ollama.ts                        | 38 +++++++------------
 36 files changed, 148 insertions(+), 59 deletions(-)

diff --git a/src/assets/locale/da/playground.json b/src/assets/locale/da/playground.json
index f8df2b2..8fdcd54 100644
--- a/src/assets/locale/da/playground.json
+++ b/src/assets/locale/da/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Stop Streaming",
         "knowledge": "Viden"
     },
-    "sendWhenEnter": "Søg, når Indtast trykkes"
+    "sendWhenEnter": "Søg, når Indtast trykkes",
+    "welcome": "Hej! Hvordan kan jeg hjælpe dig i dag?"
 }
\ No newline at end of file
diff --git a/src/assets/locale/da/settings.json b/src/assets/locale/da/settings.json
index 1666d9c..1d088d1 100644
--- a/src/assets/locale/da/settings.json
+++ b/src/assets/locale/da/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Generer titel med AI"
+      },
+      "ollamaStatus": {
+        "label": "Aktivér eller deaktivér Ollama forbindelsesstatus kontrol"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/de/playground.json b/src/assets/locale/de/playground.json
index 119d87f..43eafb9 100644
--- a/src/assets/locale/de/playground.json
+++ b/src/assets/locale/de/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Streaming stoppen",
         "knowledge": "Wissen"
     },
-    "sendWhenEnter": "Senden bei Drücken der Eingabetaste"
+    "sendWhenEnter": "Senden bei Drücken der Eingabetaste",
+    "welcome": "Hallo! Wie kann ich Ihnen heute helfen?"
 }
\ No newline at end of file
diff --git a/src/assets/locale/de/settings.json b/src/assets/locale/de/settings.json
index 30a5da3..c6b134b 100644
--- a/src/assets/locale/de/settings.json
+++ b/src/assets/locale/de/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Titel mit KI generieren"
+      },
+      "ollamaStatus": {
+        "label": "Ollama-Verbindungsstatus-Überprüfung aktivieren oder deaktivieren"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/en/playground.json b/src/assets/locale/en/playground.json
index 05da9ad..2fc5041 100644
--- a/src/assets/locale/en/playground.json
+++ b/src/assets/locale/en/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Stop Streaming",
         "knowledge": "Knowledge"
     },
-    "sendWhenEnter": "Send when Enter pressed"
+    "sendWhenEnter": "Send when Enter pressed",
+    "welcome": "Hello! How can I help you today?"
 }
\ No newline at end of file
diff --git a/src/assets/locale/en/settings.json b/src/assets/locale/en/settings.json
index 8f19b6a..eb98bf4 100644
--- a/src/assets/locale/en/settings.json
+++ b/src/assets/locale/en/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Generate Title using AI"
+      },
+      "ollamaStatus" :{
+        "label":"Enable or disable Ollama connection status check"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/es/playground.json b/src/assets/locale/es/playground.json
index afc3771..2e16639 100644
--- a/src/assets/locale/es/playground.json
+++ b/src/assets/locale/es/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Parar Transmisión",
         "knowledge": "Conocimiento"
     },
-    "sendWhenEnter": "Enviar cuando presione Enter"
+    "sendWhenEnter": "Enviar cuando presione Enter",
+    "welcome": "¡Hola! ¿Cómo puedo ayudarte hoy?"
 }
diff --git a/src/assets/locale/es/settings.json b/src/assets/locale/es/settings.json
index 241df94..4ed17d8 100644
--- a/src/assets/locale/es/settings.json
+++ b/src/assets/locale/es/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Generar título usando IA"
+      },
+      "ollamaStatus": {
+        "label": "Habilitar o deshabilitar la verificación del estado de conexión de Ollama"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/fa/playground.json b/src/assets/locale/fa/playground.json
index e9161a7..3732e7f 100644
--- a/src/assets/locale/fa/playground.json
+++ b/src/assets/locale/fa/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "توقف Streaming",
         "knowledge": "دانش"
     },
-    "sendWhenEnter": "با فشار دادن Enter ارسال شود"
+    "sendWhenEnter": "با فشار دادن Enter ارسال شود",
+    "welcome": "سلام! امروز چطور می‌توانم به شما کمک کنم؟"
 }
diff --git a/src/assets/locale/fa/settings.json b/src/assets/locale/fa/settings.json
index 039f222..0a0cde8 100644
--- a/src/assets/locale/fa/settings.json
+++ b/src/assets/locale/fa/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "تولید عنوان با استفاده از هوش مصنوعی"
+      },
+      "ollamaStatus": {
+        "label": "فعال یا غیرفعال کردن بررسی وضعیت اتصال Ollama"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/fr/playground.json b/src/assets/locale/fr/playground.json
index fddf9c0..e81c542 100644
--- a/src/assets/locale/fr/playground.json
+++ b/src/assets/locale/fr/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Arrêtez la diffusion",
         "knowledge": "Connaissance"
     },
-    "sendWhenEnter": "Envoyer en appuyant sur Entrée"
+    "sendWhenEnter": "Envoyer en appuyant sur Entrée",
+    "welcome": "Bonjour ! Comment puis-je vous aider aujourd'hui ?"
 }
\ No newline at end of file
diff --git a/src/assets/locale/fr/settings.json b/src/assets/locale/fr/settings.json
index 027c1b3..7d3b191 100644
--- a/src/assets/locale/fr/settings.json
+++ b/src/assets/locale/fr/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Générer le titre en utilisant l'IA"
+      },
+      "ollamaStatus": {
+        "label": "Activer ou désactiver la vérification de l'état de la connexion Ollama"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/it/playground.json b/src/assets/locale/it/playground.json
index 766fe20..c43cf87 100644
--- a/src/assets/locale/it/playground.json
+++ b/src/assets/locale/it/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Ferma lo Streaming",
         "knowledge": "Conoscenza"
     },
-    "sendWhenEnter": "Invia subito dopo Enter"
+    "sendWhenEnter": "Invia subito dopo Enter",
+    "welcome": "Ciao! Come posso aiutarti oggi?"
 }
\ No newline at end of file
diff --git a/src/assets/locale/it/settings.json b/src/assets/locale/it/settings.json
index af15a71..e227d22 100644
--- a/src/assets/locale/it/settings.json
+++ b/src/assets/locale/it/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Genera titolo utilizzando l'IA"
+      },
+      "ollamaStatus": {
+        "label": "Abilita o disabilita il controllo dello stato della connessione Ollama"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/ja-JP/playground.json b/src/assets/locale/ja-JP/playground.json
index 96c4d1a..f89911c 100644
--- a/src/assets/locale/ja-JP/playground.json
+++ b/src/assets/locale/ja-JP/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "ストリーミングを停止",
         "knowledge": "知識"
     },
-    "sendWhenEnter": "Enterキーを押すと送信"
+    "sendWhenEnter": "Enterキーを押すと送信",
+    "welcome": "こんにちは！本日はどのようなお手伝いができますか？"
 }
\ No newline at end of file
diff --git a/src/assets/locale/ja-JP/settings.json b/src/assets/locale/ja-JP/settings.json
index af5a1f5..0058062 100644
--- a/src/assets/locale/ja-JP/settings.json
+++ b/src/assets/locale/ja-JP/settings.json
@@ -35,6 +35,9 @@
       },
       "generateTitle": {
         "label": "AIを使用してタイトルを生成"
+      },
+      "ollamaStatus": {
+        "label": "Ollamaの接続状態チェックを有効または無効にする"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/ko/playground.json b/src/assets/locale/ko/playground.json
index 082ad5d..d60b102 100644
--- a/src/assets/locale/ko/playground.json
+++ b/src/assets/locale/ko/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "스트리밍 중지",
         "knowledge": "지식"
     },
-    "sendWhenEnter": "Enter 키를 누르면 전송"
+    "sendWhenEnter": "Enter 키를 누르면 전송",
+    "welcome": "안녕하세요! 오늘 어떻게 도와드릴까요?"
 }
diff --git a/src/assets/locale/ko/settings.json b/src/assets/locale/ko/settings.json
index 072c7f6..f3c7109 100644
--- a/src/assets/locale/ko/settings.json
+++ b/src/assets/locale/ko/settings.json
@@ -35,6 +35,9 @@
       },
       "generateTitle": {
         "label": "AI로 제목 생성"
+      },
+      "ollamaStatus": {
+        "label": "Ollama 연결 상태 확인 활성화 또는 비활성화"
       }
     },
     "sidepanelRag": {
@@ -342,4 +345,3 @@
     "title": "Chrome AI 설정"
   }
 }
-
diff --git a/src/assets/locale/ml/playground.json b/src/assets/locale/ml/playground.json
index 8297aa4..e789350 100644
--- a/src/assets/locale/ml/playground.json
+++ b/src/assets/locale/ml/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "സ്ട്രീമിംഗ് നിർത്തുക",
         "knowledge": "അറിവ്"
     },
-    "sendWhenEnter": "എന്റര്‍ അമര്‍ത്തുമ്പോള്‍ അയയ്ക്കുക"
+    "sendWhenEnter": "എന്റര്‍ അമര്‍ത്തുമ്പോള്‍ അയയ്ക്കുക",
+    "welcome": "നമസ്കാരം! ഇന്ന് എനിക്ക് നിങ്ങളെ എങ്ങനെ സഹായിക്കാൻ കഴിയും?"
 }
\ No newline at end of file
diff --git a/src/assets/locale/ml/settings.json b/src/assets/locale/ml/settings.json
index b6ebd30..03d67f1 100644
--- a/src/assets/locale/ml/settings.json
+++ b/src/assets/locale/ml/settings.json
@@ -35,6 +35,9 @@
       },
       "generateTitle": {
         "label": "എഐ ഉപയോഗിച്ച് ശീർഷകം സൃഷ്ടിക്കുക"
+      },
+      "ollamaStatus": {
+        "label": "ഒല്ലാമ കണക്ഷൻ സ്റ്റാറ്റസ് പരിശോധന പ്രവർത്തനക്ഷമമാക്കുകയോ പ്രവർത്തനരഹിതമാക്കുകയോ ചെയ്യുക"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/no/playground.json b/src/assets/locale/no/playground.json
index b812649..43e2fa3 100644
--- a/src/assets/locale/no/playground.json
+++ b/src/assets/locale/no/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Stopp Streaming",
         "knowledge": "Kunnskap"
     },
-    "sendWhenEnter": "Søk når Enter trykkes"
+    "sendWhenEnter": "Søk når Enter trykkes",
+    "welcome": "Hei! Hvordan kan jeg hjelpe deg i dag?"
 }
diff --git a/src/assets/locale/no/settings.json b/src/assets/locale/no/settings.json
index a09437f..8085e3e 100644
--- a/src/assets/locale/no/settings.json
+++ b/src/assets/locale/no/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Generer tittel med AI"
+      },
+      "ollamaStatus": {
+        "label": "Aktiver eller deaktiver Ollama tilkoblingsstatussjekk"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/pt-BR/playground.json b/src/assets/locale/pt-BR/playground.json
index 6e91e97..ea11ffa 100644
--- a/src/assets/locale/pt-BR/playground.json
+++ b/src/assets/locale/pt-BR/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Parar Streaming",
         "knowledge": "Conhecimento"
     },
-    "sendWhenEnter": "Enviar ao pressionar Enter"
+    "sendWhenEnter": "Enviar ao pressionar Enter",
+    "welcome": "Olá! Como posso ajudar você hoje?"
 }
\ No newline at end of file
diff --git a/src/assets/locale/pt-BR/settings.json b/src/assets/locale/pt-BR/settings.json
index ac7e70b..fad3b11 100644
--- a/src/assets/locale/pt-BR/settings.json
+++ b/src/assets/locale/pt-BR/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Gerar título usando IA"
+      },
+      "ollamaStatus": {
+        "label": "Ativar ou desativar verificação de status da conexão Ollama"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/ru/playground.json b/src/assets/locale/ru/playground.json
index fdc2933..39ba724 100644
--- a/src/assets/locale/ru/playground.json
+++ b/src/assets/locale/ru/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Остановить поток",
         "knowledge": "Знание"
     },
-    "sendWhenEnter": "Отправить при нажатии клавиши Enter"
+    "sendWhenEnter": "Отправить при нажатии клавиши Enter",
+    "welcome": "Здравствуйте! Как я могу помочь вам сегодня?"
 }
diff --git a/src/assets/locale/ru/settings.json b/src/assets/locale/ru/settings.json
index 0c3c183..adf0008 100644
--- a/src/assets/locale/ru/settings.json
+++ b/src/assets/locale/ru/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Сгенерировать заголовок с помощью ИИ"
+      },
+      "ollamaStatus": {
+        "label": "Включить или отключить проверку состояния подключения Ollama"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/sv/playground.json b/src/assets/locale/sv/playground.json
index d7195c5..16c8e80 100644
--- a/src/assets/locale/sv/playground.json
+++ b/src/assets/locale/sv/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Stoppa strömning",
         "knowledge": "Kunskap"
     },
-    "sendWhenEnter": "Skicka när Enter trycks"
+    "sendWhenEnter": "Skicka när Enter trycks",
+    "welcome": "Hej! Hur kan jag hjälpa dig idag?"
 }
diff --git a/src/assets/locale/sv/settings.json b/src/assets/locale/sv/settings.json
index cb28f1e..d1e11d0 100644
--- a/src/assets/locale/sv/settings.json
+++ b/src/assets/locale/sv/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Generera titel med AI"
+      },
+      "ollamaStatus": {
+        "label": "Aktivera eller inaktivera Ollama anslutningsstatuskontroll"
       }
     },
     "sidepanelRag": {
diff --git a/src/assets/locale/zh/playground.json b/src/assets/locale/zh/playground.json
index 8f7da46..2b99e39 100644
--- a/src/assets/locale/zh/playground.json
+++ b/src/assets/locale/zh/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "停止流媒体",
         "knowledge": "知识"
     },
-    "sendWhenEnter": "按Enter发送"
+    "sendWhenEnter": "按Enter发送",
+    "welcome": "你好！今天我能帮你什么？"
 }
\ No newline at end of file
diff --git a/src/assets/locale/zh/settings.json b/src/assets/locale/zh/settings.json
index 4c86040..2f834e9 100644
--- a/src/assets/locale/zh/settings.json
+++ b/src/assets/locale/zh/settings.json
@@ -35,6 +35,9 @@
       },
       "generateTitle": {
         "label": "使用人工智能生成标题"
+      },
+      "ollamaStatus": {
+        "label": "启用或禁用Ollama连接状态检查"
       }
     },
     "sidepanelRag": {
diff --git a/src/components/Layouts/Header.tsx b/src/components/Layouts/Header.tsx
index 3e4a1f6..72f31da 100644
--- a/src/components/Layouts/Header.tsx
+++ b/src/components/Layouts/Header.tsx
@@ -113,6 +113,7 @@ export const Header: React.FC<Props> = ({
           <PageAssistSelect
             className="w-80"
             placeholder={t("common:selectAModel")}
+            loadingText={t("common:selectAModel")}
             value={selectedModel}
             onChange={(e) => {
               setSelectedModel(e.value)
diff --git a/src/components/Option/Playground/PlaygroundEmpty.tsx b/src/components/Option/Playground/PlaygroundEmpty.tsx
index 7fb94ab..4e23d7d 100644
--- a/src/components/Option/Playground/PlaygroundEmpty.tsx
+++ b/src/components/Option/Playground/PlaygroundEmpty.tsx
@@ -1,4 +1,5 @@
 import { cleanUrl } from "@/libs/clean-url"
+import { useStorage } from "@plasmohq/storage/hook"
 import { useQuery } from "@tanstack/react-query"
 import { RotateCcw } from "lucide-react"
 import { useEffect, useState } from "react"
@@ -12,6 +13,9 @@ import {
 export const PlaygroundEmpty = () => {
   const [ollamaURL, setOllamaURL] = useState<string>("")
   const { t } = useTranslation(["playground", "common"])
+
+  const [checkOllamaStatus] = useStorage("checkOllamaStatus", true)
+
   const {
     data: ollamaInfo,
     status: ollamaStatus,
@@ -23,19 +27,32 @@ export const PlaygroundEmpty = () => {
       const ollamaURL = await getOllamaURL()
       const isOk = await isOllamaRunning()
 
+      if (ollamaURL) {
+        saveOllamaURL(ollamaURL)
+      }
+
       return {
         isOk,
         ollamaURL
       }
-    }
+    },
+    enabled: checkOllamaStatus
   })
 
-  useEffect(() => {
-    if (ollamaInfo?.ollamaURL) {
-      setOllamaURL(ollamaInfo.ollamaURL)
-    }
-  }, [ollamaInfo])
-
+  if (!checkOllamaStatus) {
+    return (
+      <div className="mx-auto sm:max-w-xl px-4 mt-10">
+        <div className="rounded-lg justify-center items-center flex flex-col border p-8 bg-gray-50 dark:bg-[#262626] dark:border-gray-600">
+          <h1 className="text-sm  font-medium text-center text-gray-500 dark:text-gray-400 flex gap-3 items-center justify-center">
+            <span >👋</span>
+            <span className="text-gray-700 dark:text-gray-300">
+              {t("welcome")}
+            </span>
+          </h1>
+        </div>
+      </div>
+    )
+  }
   return (
     <div className="mx-auto sm:max-w-xl px-4 mt-10">
       <div className="rounded-lg justify-center items-center flex flex-col border p-8 bg-gray-50 dark:bg-[#262626]  dark:border-gray-600">
diff --git a/src/components/Option/Settings/general-settings.tsx b/src/components/Option/Settings/general-settings.tsx
index 7ceba76..f615236 100644
--- a/src/components/Option/Settings/general-settings.tsx
+++ b/src/components/Option/Settings/general-settings.tsx
@@ -16,13 +16,12 @@ import {
 import { useStorage } from "@plasmohq/storage/hook"
 
 export const GeneralSettings = () => {
-  const { clearChat } =
-    useMessageOption()
+  const { clearChat } = useMessageOption()
 
-  const [ speechToTextLanguage, setSpeechToTextLanguage ] = useStorage(
-      "speechToTextLanguage",
-      "en-US"
-    )
+  const [speechToTextLanguage, setSpeechToTextLanguage] = useStorage(
+    "speechToTextLanguage",
+    "en-US"
+  )
   const [copilotResumeLastChat, setCopilotResumeLastChat] = useStorage(
     "copilotResumeLastChat",
     false
@@ -41,6 +40,11 @@ export const GeneralSettings = () => {
   const [sendNotificationAfterIndexing, setSendNotificationAfterIndexing] =
     useStorage("sendNotificationAfterIndexing", false)
 
+  const [checkOllamaStatus, setCheckOllamaStatus] = useStorage(
+    "checkOllamaStatus",
+    true
+  )
+
   const queryClient = useQueryClient()
 
   const { mode, toggleDarkMode } = useDarkMode()
@@ -160,6 +164,19 @@ export const GeneralSettings = () => {
         />
       </div>
 
+      <div className="flex flex-row justify-between">
+        <div className="inline-flex items-center gap-2">
+          <span className="text-gray-700   dark:text-neutral-50">
+            {t("generalSettings.settings.ollamaStatus.label")}
+          </span>
+        </div>
+
+        <Switch
+          checked={checkOllamaStatus}
+          onChange={(checked) => setCheckOllamaStatus(checked)}
+        />
+      </div>
+
       <div className="flex flex-row justify-between">
         <span className="text-gray-700 dark:text-neutral-50 ">
           {t("generalSettings.settings.darkMode.label")}
diff --git a/src/components/Select/index.tsx b/src/components/Select/index.tsx
index e3da503..34d205f 100644
--- a/src/components/Select/index.tsx
+++ b/src/components/Select/index.tsx
@@ -235,14 +235,16 @@ export const PageAssistSelect: React.FC<SelectProps> = ({
         onClick={() => !disabled && !isLoading && setIsOpen(!isOpen)}
         onKeyDown={handleKeyDown}
         className={`${defaultSelectClass} ${className}`}>
-        <span className="!truncate flex items-center gap-2 ">
-          {isLoading && <LoadingIndicator />}
+        <span className="!truncate flex items-center gap-2 select-none">
+          {isLoading && <LoadingIndicator  />}
           {isLoading ? (
             loadingText
           ) : selectedOption ? (
             selectedOption.label
           ) : (
-            <span className="dark:text-gray-400 text-sm">{placeholder}</span>
+            <span className="dark:text-gray-500 font-semibold text-[14px]">
+              {placeholder}
+            </span>
           )}
         </span>
         <ChevronDown
diff --git a/src/components/Sidepanel/Chat/empty.tsx b/src/components/Sidepanel/Chat/empty.tsx
index f94304b..cf38d1a 100644
--- a/src/components/Sidepanel/Chat/empty.tsx
+++ b/src/components/Sidepanel/Chat/empty.tsx
@@ -1,4 +1,5 @@
 import { cleanUrl } from "@/libs/clean-url"
+import { useStorage } from "@plasmohq/storage/hook"
 import { useQuery, useQueryClient } from "@tanstack/react-query"
 import { Select } from "antd"
 import { RotateCcw } from "lucide-react"
@@ -17,13 +18,15 @@ export const EmptySidePanel = () => {
   const [ollamaURL, setOllamaURL] = useState<string>("")
   const { t } = useTranslation(["playground", "common"])
   const queryClient = useQueryClient()
+  const [checkOllamaStatus] = useStorage("checkOllamaStatus", true)
+
   const {
     data: ollamaInfo,
     status: ollamaStatus,
     refetch,
     isRefetching
   } = useQuery({
-    queryKey: ["ollamaStatus"],
+    queryKey: ["ollamaStatus", checkOllamaStatus],
     queryFn: async () => {
       const ollamaURL = await getOllamaURL()
       const isOk = await isOllamaRunning()
@@ -32,7 +35,7 @@ export const EmptySidePanel = () => {
         queryKey: ["getAllModelsForSelect"]
       })
       return {
-        isOk,
+        isOk: checkOllamaStatus ? isOk : true,
         models,
         ollamaURL
       }
@@ -59,7 +62,7 @@ export const EmptySidePanel = () => {
             </p>
           </div>
         )}
-        {!isRefetching && ollamaStatus === "success" ? (
+        {!isRefetching && ollamaStatus === "success" && checkOllamaStatus ? (
           ollamaInfo.isOk ? (
             <div className="inline-flex  items-center space-x-2">
               <div className="w-3 h-3 bg-green-500 rounded-full"></div>
diff --git a/src/services/ollama.ts b/src/services/ollama.ts
index e8689da..42046ff 100644
--- a/src/services/ollama.ts
+++ b/src/services/ollama.ts
@@ -102,6 +102,7 @@ export const getAllModels = async ({
   returnEmpty?: boolean
 }) => {
   try {
+
     const baseUrl = await getOllamaURL()
     const response = await fetcher(`${cleanUrl(baseUrl)}/api/tags`)
     if (!response.ok) {
@@ -178,30 +179,9 @@ export const fetchChatModels = async ({
   returnEmpty?: boolean
 }) => {
   try {
-    const baseUrl = await getOllamaURL()
-    const response = await fetcher(`${cleanUrl(baseUrl)}/api/tags`)
-    if (!response.ok) {
-      if (returnEmpty) {
-        return []
-      }
-      throw new Error(response.statusText)
-    }
-    const json = await response.json()
-    const models = json.models as {
-      name: string
-      model: string
-      modified_at: string
-      size: number
-      digest: string
-      details?: {
-        parent_model?: string
-        format: string
-        family: string
-        families: string[]
-        parameter_size: string
-        quantization_level: string
-      }
-    }[]
+
+    const models = await getAllModels({ returnEmpty })
+
     const chatModels = models
       ?.filter((model) => {
         return (
@@ -408,3 +388,13 @@ export const getPageShareUrl = async () => {
 export const setPageShareUrl = async (pageShareUrl: string) => {
   await storage.set("pageShareUrl", pageShareUrl)
 }
+
+
+export const isOllamaEnabled = async () => {
+  const ollamaStatus = await storage.get<boolean>("checkOllamaStatus")
+  // if data is empty or null then return true 
+  if (typeof ollamaStatus === "undefined" || ollamaStatus === null) {
+    return true
+  }
+  return ollamaStatus
+}
\ No newline at end of file

From eaf0e5b2414e39c199656a0e1f76cb1bff79c6b6 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sun, 17 Nov 2024 12:29:41 +0530
Subject: [PATCH 24/27] feat(NewChat): improve button styles and layout

- Update the styles and layout of the "New Chat" and dropdown buttons in the NewChat component
- Rounded the buttons to match the design
- Adjusted the spacing and alignment of the button elements
- Ensured consistent styling between light and dark modes
---
 src/components/Layouts/NewChat.tsx | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/components/Layouts/NewChat.tsx b/src/components/Layouts/NewChat.tsx
index e20681a..fa8cab8 100644
--- a/src/components/Layouts/NewChat.tsx
+++ b/src/components/Layouts/NewChat.tsx
@@ -43,15 +43,15 @@ export const NewChat: React.FC<Props> = ({ clearChat }) => {
     <div className="flex items-center justify-between">
       <button
         onClick={clearChat}
-        className="inline-flex dark:bg-transparent  bg-white items-center rounded-r-none rounded-lg border dark:border-gray-700 bg-transparent px-3 py-2.5 pr-6 text-xs lg:text-sm font-medium leading-4 text-gray-800 dark:text-white disabled:opacity-50 ease-in-out transition-colors duration-200 hover:bg-gray-100 dark:hover:bg-gray-800 dark:hover:text-white">
+        className="inline-flex dark:bg-transparent bg-white items-center rounded-s-lg rounded-e-none border dark:border-gray-700 bg-transparent px-3 py-2.5 pe-6 text-xs lg:text-sm font-medium leading-4 text-gray-800 dark:text-white disabled:opacity-50 ease-in-out transition-colors duration-200 hover:bg-gray-100 dark:hover:bg-gray-800 dark:hover:text-white">
         <SquarePen className="h-5 w-5" />
-        <span className="truncate ml-3">{t("newChat")}</span>
+        <span className="truncate ms-3">{t("newChat")}</span>
       </button>
       <Dropdown menu={{ items }} trigger={["click"]}>
-        <button className="inline-flex dark:bg-transparent bg-white items-center rounded-lg border-l-0 rounded-l-none border dark:border-gray-700 bg-transparent px-3 py-2.5 text-xs lg:text-sm font-medium leading-4 text-gray-800 dark:text-white disabled:opacity-50 ease-in-out transition-colors duration-200 hover:bg-gray-100 dark:hover:bg-gray-800 dark:hover:text-white">
+        <button className="inline-flex dark:bg-transparent bg-white items-center rounded-lg border-s-0 rounded-s-none border dark:border-gray-700 bg-transparent px-3 py-2.5 text-xs lg:text-sm font-medium leading-4 text-gray-800 dark:text-white disabled:opacity-50 ease-in-out transition-colors duration-200 hover:bg-gray-100 dark:hover:bg-gray-800 dark:hover:text-white">
           <MoreHorizontal className="h-5 w-5 text-gray-600 dark:text-gray-400" />
         </button>
       </Dropdown>
     </div>
   )
-}
+}
\ No newline at end of file

From 73306d95dbf0f93b509d4d3b8e877e2be3c20963 Mon Sep 17 00:00:00 2001
From: "Volodymyr M. Lisivka" <vlisivka@gmail.com>
Date: Fri, 22 Nov 2024 17:45:39 +0200
Subject: [PATCH 25/27] Ukrainian translation for Page Assist is added.

---
 src/assets/locale/uk/chrome.json     |  13 +
 src/assets/locale/uk/common.json     | 125 ++++++++++
 src/assets/locale/uk/knowledge.json  |  40 ++++
 src/assets/locale/uk/openai.json     |  90 +++++++
 src/assets/locale/uk/option.json     |  13 +
 src/assets/locale/uk/playground.json |  29 +++
 src/assets/locale/uk/settings.json   | 341 +++++++++++++++++++++++++++
 src/assets/locale/uk/sidepanel.json  |   7 +
 src/i18n/index.ts                    |   3 +
 src/i18n/lang/uk.ts                  |  19 ++
 src/i18n/support-language.ts         |   4 +
 11 files changed, 684 insertions(+)
 create mode 100644 src/assets/locale/uk/chrome.json
 create mode 100644 src/assets/locale/uk/common.json
 create mode 100644 src/assets/locale/uk/knowledge.json
 create mode 100644 src/assets/locale/uk/openai.json
 create mode 100644 src/assets/locale/uk/option.json
 create mode 100644 src/assets/locale/uk/playground.json
 create mode 100644 src/assets/locale/uk/settings.json
 create mode 100644 src/assets/locale/uk/sidepanel.json
 create mode 100644 src/i18n/lang/uk.ts

diff --git a/src/assets/locale/uk/chrome.json b/src/assets/locale/uk/chrome.json
new file mode 100644
index 0000000..12fa7ee
--- /dev/null
+++ b/src/assets/locale/uk/chrome.json
@@ -0,0 +1,13 @@
+{
+    "heading": "Налаштування Chrome AI",
+    "status": {
+        "label": "Ввімкнути або вимкнути підтримку Chrome AI на Page Assist"
+    },
+    "error": {
+        "browser_not_supported": "Ця версія Chrome не підтримується моделлю Gemini Nano. Будь ласка, оновіть до версії 127 або новішої",
+        "ai_not_supported": "Налаштування chrome://flags/#prompt-api-for-gemini-nano не увімкнено. Будь ласка, увімкніть його.",
+        "ai_not_ready": "Модель Gemini Nano ще не готова; вам потрібно перевірити налаштування Chrome знову",
+        "internal_error": "Стався внутрішній збій. Будь ласка, спробуйте пізніше."
+    },
+    "errorDescription": "Щоб використовувати Chrome AI, вам потрібна версія переглядача новіша за 127-му, яка наразі доступна в каналах Dev і Canary. Після завантаження підтримуваної версії виконайте наступні кроки:\n\n1. Перейдіть до `chrome://flags/#prompt-api-for-gemini-nano` та виберіть \"Ввімкнути\".\n2. Перейдіть до `chrome://flags/#optimization-guide-on-device-model` та виберіть \"EnabledBypassPrefRequirement\".\n3. Перейдіть до `chrome://components`, знайдіть \"Optimization Guide On Device Model\" і натисніть \"Перевірити оновлення\". Це завантажить модель. Якщо ви не бачите налаштувань, повторіть кроки 1 і 2 та перезапустіть переглядач."
+}
diff --git a/src/assets/locale/uk/common.json b/src/assets/locale/uk/common.json
new file mode 100644
index 0000000..899e12b
--- /dev/null
+++ b/src/assets/locale/uk/common.json
@@ -0,0 +1,125 @@
+{
+    "pageAssist": "Допомога на сторінці",
+    "selectAModel": "Виберіть модель",
+    "save": "Зберегти",
+    "saved": "Збережено",
+    "cancel": "Скасувати",
+    "retry": "Спробувати знову",
+    "share": {
+        "tooltip": {
+            "share": "Поділитися"
+        },
+        "modal": {
+            "title": "Поділитися посиланням на чат"
+        },
+        "form": {
+            "defaultValue": {
+                "name": "Анонімний",
+                "title": "Неназваний чат"
+            },
+            "title": {
+                "label": "Назва чату",
+                "placeholder": "Введіть назву чату",
+                "required": "Назва чату обовʼязкова"
+            },
+            "name": {
+                "label": "Ваше імʼя",
+                "placeholder": "Введіть ваше імʼя",
+                "required": "Ваше імʼя обов'язкове"
+            },
+            "btn": {
+                "save": "Створити посилання",
+                "saving": "Створення посилання..."
+            }
+        },
+        "notification": {
+            "successGenerate": "Посилання скопійовано до буфера обміну",
+            "failGenerate": "Не вдалося створити посилання"
+        }
+    },
+    "copyToClipboard": "Копіювати в буфер обміну",
+    "webSearch": "Пошук у мережі",
+    "regenerate": "Перегенерувати",
+    "edit": "Редагувати",
+    "delete": "Видалити",
+    "saveAndSubmit": "Зберегти та надіслати",
+    "editMessage": {
+        "placeholder": "Введіть повідомлення..."
+    },
+    "submit": "Надіслати",
+    "noData": "Немає даних",
+    "noHistory": "Немає історії чату",
+    "chatWithCurrentPage": "Чат з поточної сторінки",
+    "beta": "Бета",
+    "tts": "Читання вголос",
+    "currentChatModelSettings": "Налаштування поточної моделі чату",
+    "modelSettings": {
+        "label": "Налаштування моделі",
+        "description": "Встановіть глобальні параметри моделі для всіх чатів",
+        "form": {
+            "keepAlive": {
+                "label": "Тривалість збереження в памʼяті",
+                "help": "вказує, як довго модель залишатиметься завантаженою у памʼять після запиту (за замовчуванням: 5 хв)",
+                "placeholder": "Введіть тривалість збереження у памʼяті (наприклад, 5m, 10m, 1h)"
+            },
+            "temperature": {
+                "label": "Температура",
+                "placeholder": "Введіть значення температури (наприклад, 0.7, 1.0)"
+            },
+            "numCtx": {
+                "label": "Кількість контекстів",
+                "placeholder": "Введіть кількість контекстів (типово: 2048)"
+            },
+            "numPredict": {
+                "label": "Максимальна кількість токенів",
+                "placeholder": "Введіть максимальну кількість токенів (наприклад, 2048, 4096)"
+            },
+            "seed": {
+                "label": "Зерно",
+                "placeholder": "Введіть значення зерна (наприклад, 1234)",
+                "help": "Повторюваність виводу моделі"
+            },
+            "topK": {
+                "label": "Top K",
+                "placeholder": "Введіть значення для Верхніх K (наприклад, 40, 100)"
+            },
+            "topP": {
+                "label": "Top P",
+                "placeholder": "Введіть значення для Верхнього P (наприклад, 0.9, 0.95)"
+            },
+            "numGpu": {
+                "label": "Кількість GPU",
+                "placeholder": "Введіть кількість шарів для відправки на GPU"
+            },
+            "systemPrompt": {
+                "label": "Тимчасовий системний запит",
+                "placeholder": "Введіть системний запит",
+                "help": "Швидкий спосіб встановити системний запит для поточного чату, який замінить вибраний системний запит, якщо він існує."
+            }
+        },
+        "advanced": "Додаткові налаштування моделі"
+    },
+    "copilot": {
+        "summary": "Підсумувати",
+        "explain": "Пояснити",
+        "rephrase": "Перефразувати",
+        "translate": "Перекласти",
+        "custom": "Власне"
+    },
+    "citations": "Цитати",
+    "segmented": {
+        "ollama": "Моделі Ollama",
+        "custom": "Власні моделі"
+    },
+    "downloadCode": "Завантажити код",
+    "date": {
+        "pinned": "Прикріплено",
+        "today": "Сьогодні",
+        "yesterday": "Вчора",
+        "last7Days": "Останні 7 днів",
+        "older": "Старіше"
+    },
+    "pin": "Прикріпити",
+    "unpin": "Відкріпити",
+    "generationInfo": "Інформація про генерацію"
+}
diff --git a/src/assets/locale/uk/knowledge.json b/src/assets/locale/uk/knowledge.json
new file mode 100644
index 0000000..dd5dd9f
--- /dev/null
+++ b/src/assets/locale/uk/knowledge.json
@@ -0,0 +1,40 @@
+{
+    "addBtn": "Додати нове знання",
+    "columns": {
+        "title": "Назва",
+        "status": "Статус",
+        "embeddings": "Модель вкладень",
+        "createdAt": "Створено в",
+        "action": "Дії"
+    },
+    "expandedColumns": {
+        "name": "Імʼя"
+    },
+    "confirm": {
+        "delete": "Ви впевнені, що хочете видалити це знання?"
+    },
+    "deleteSuccess": "Знання успішно видалено",
+    "status": {
+        "pending": "Очікування",
+        "finished": "Завершено",
+        "processing": "Обробка",
+        "failed": "Не вдалося"
+    },
+    "addKnowledge": "Додати знання",
+    "form": {
+        "title": {
+            "label": "Назва знань",
+            "placeholder": "Введіть назву знань",
+            "required": "Назва знань обовʼязкова"
+        },
+        "uploadFile": {
+            "label": "Завантажити файл",
+            "uploadText": "Перетягніть файл сюди або натисніть для завантаження",
+            "uploadHint": "Підтримуються типи файлів: .pdf, .csv, .txt, .md, .docx",
+            "required": "Файл обовʼязковий"
+        },
+        "submit": "Надіслати",
+        "success": "Знання успішно додано"
+    },
+    "noEmbeddingModel": "Будь ласка, додайте модель вкладень з сторінки налаштувань RAG спочатку"
+}
diff --git a/src/assets/locale/uk/openai.json b/src/assets/locale/uk/openai.json
new file mode 100644
index 0000000..f714a97
--- /dev/null
+++ b/src/assets/locale/uk/openai.json
@@ -0,0 +1,90 @@
+{
+    "settings": "Сумісний з OpenAI API",
+    "heading": "Сумісний з OpenAI API",
+    "subheading": "Тут керування та налаштування ваших постачальників OpenAI API.",
+    "addBtn": "Додати постачальника",
+    "table": {
+        "name": "Назва постачальника",
+        "baseUrl": "Базовий URL",
+        "actions": "Дії"
+    },
+    "modal": {
+        "titleAdd": "Додати нового постачальника",
+        "name": {
+            "label": "Назва постачальника",
+            "required": "Назва постачальника обовʼязкова.",
+            "placeholder": "Введіть назву постачальника"
+        },
+        "baseUrl": {
+            "label": "Базовий URL",
+            "help": "Базовий URL постачальника OpenAI API. Наприклад, (http://localhost:1234/v1)",
+            "required": "Базовий URL обовʼязковий.",
+            "placeholder": "Введіть базовий URL"
+        },
+        "apiKey": {
+            "label": "Ключ API",
+            "required": "Ключ API обовʼязковий.",
+            "placeholder": "Введіть ключ API"
+        },
+        "submit": "Зберегти",
+        "update": "Оновити",
+        "deleteConfirm": "Ви впевнені, що хочете видалити цього постачальника?",
+        "model": {
+            "title": "Список моделей",
+            "subheading": "Виберіть чат-моделі, які ви хочете використовувати з цим постачальником.",
+            "success": "Моделі успішно додано."
+        },
+        "tipLMStudio": "Page Assist автоматично отримає моделі, завантажені в LM Studio. Вам не потрібно додавати їх вручну."
+    },
+    "addSuccess": "Постачальника додано успішно.",
+    "deleteSuccess": "Постачальника видалено успішно.",
+    "updateSuccess": "Постачальника оновлено успішно.",
+    "delete": "Видалити",
+    "edit": "Редагувати",
+    "newModel": "Додати моделі до постачальника",
+    "noNewModel": "Для LMStudio, Ollama, Llamafile ми отримуємо моделі динамічно. Ручного додавання не потрібно.",
+    "searchModel": "Пошук моделі",
+    "selectAll": "Вибрати все",
+    "save": "Зберегти",
+    "saving": "Збереження...",
+    "manageModels": {
+        "columns": {
+            "name": "Назва моделі",
+            "model_type": "Тип моделі",
+            "model_id": "Ідентифікатор моделі",
+            "provider": "Назва постачальника",
+            "actions": "Дії"
+        },
+        "tooltip": {
+            "delete": "Видалити"
+        },
+        "confirm": {
+            "delete": "Ви впевнені, що хочете видалити цю модель?"
+        },
+        "modal": {
+            "title": "Додати власну модель",
+            "form": {
+                "name": {
+                    "label": "Ідентифікатор моделі",
+                    "placeholder": "llama3.2",
+                    "required": "Ідентифікатор моделі обовʼязковий."
+                },
+                "provider": {
+                    "label": "Постачальник",
+                    "placeholder": "Виберіть постачальника",
+                    "required": "Постачальник обовʼязковий."
+                },
+                "type": {
+                    "label": "Тип моделі"
+                }
+            }
+        }
+    },
+    "noModelFound": "Не знайдено моделей. Переконайтеся, що ви додали правильного постачальника з базовим URL та ключем API.",
+    "radio": {
+        "chat": "Модель чату",
+        "embedding": "Модель вкладень",
+        "chatInfo": "використовується для завершення чату та генерації розмови",
+        "embeddingInfo": "використовується для RAG та інших повʼязаних завдань семантичного пошуку."
+    }
+}
diff --git a/src/assets/locale/uk/option.json b/src/assets/locale/uk/option.json
new file mode 100644
index 0000000..22a76d4
--- /dev/null
+++ b/src/assets/locale/uk/option.json
@@ -0,0 +1,13 @@
+{
+    "newChat": "Новий чат",
+    "selectAPrompt": "Виберіть запит",
+    "githubRepository": "Репозиторій GitHub",
+    "settings": "Налаштування",
+    "sidebarTitle": "Історія чату",
+    "error": "Збій",
+    "somethingWentWrong": "Щось пішло не так",
+    "validationSelectModel": "Будь ласка, виберіть модель для продовження",
+    "deleteHistoryConfirmation": "Ви впевнені, що хочете видалити цю історію?",
+    "editHistoryTitle": "Введіть нову назву",
+    "temporaryChat": "Тимчасовий чат"
+}
diff --git a/src/assets/locale/uk/playground.json b/src/assets/locale/uk/playground.json
new file mode 100644
index 0000000..a481804
--- /dev/null
+++ b/src/assets/locale/uk/playground.json
@@ -0,0 +1,29 @@
+{
+    "ollamaState": {
+        "searching": "Пошук вашого Ollama 🦙",
+        "running": "Ollama працює 🦙",
+        "notRunning": "Неможливо підключитися до Ollama 🦙",
+        "connectionError": "Здається, у вас виникла проблема з підключенням. Будь ласка, зверніться до цієї документації<anchor> для усунення несправностей.</anchor>"
+    },
+    "formError": {
+        "noModel": "Виберіть модель",
+        "noEmbeddingModel": "Налаштуйте модель вкладень на сторінці Налаштування > RAG"
+    },
+    "form": {
+        "textarea": {
+            "placeholder": "Введіть повідомлення..."
+        },
+        "webSearch": {
+            "on": "Увімкнено",
+            "off": "Вимкнено"
+        }
+    },
+    "tooltip": {
+        "searchInternet": "Пошук в Інтернеті",
+        "speechToText": "Голос у текст",
+        "uploadImage": "Завантажити зображення",
+        "stopStreaming": "Зупинити трансляцію",
+        "knowledge": "Знання"
+    },
+    "sendWhenEnter": "Надсилати при натисканні Enter"
+}
diff --git a/src/assets/locale/uk/settings.json b/src/assets/locale/uk/settings.json
new file mode 100644
index 0000000..02dc3e8
--- /dev/null
+++ b/src/assets/locale/uk/settings.json
@@ -0,0 +1,341 @@
+{
+  "generalSettings": {
+    "title": "Загальні налаштування",
+    "settings": {
+      "heading": "Налаштування веб-інтерфейсу",
+      "speechRecognitionLang": {
+        "label": "Мова для розпізнавання голосу",
+        "placeholder": "Виберіть мову"
+      },
+      "language": {
+        "label": "Мова інтерфейсу",
+        "placeholder": "Виберіть мову"
+      },
+      "darkMode": {
+        "label": "Змінити тему",
+        "options": {
+          "light": "Світла",
+          "dark": "Темна"
+        }
+      },
+      "copilotResumeLastChat": {
+        "label": "Поновити останню розмову при відкритті бічної панелі (Copilot)"
+      },
+      "hideCurrentChatModelSettings": {
+        "label": "Приховати налаштування поточної моделі чату"
+      },
+      "restoreLastChatModel": {
+        "label": "Відновити останню використану модель для попередніх чатів"
+      },
+      "sendNotificationAfterIndexing": {
+        "label": "Надсилати сповіщення після завершення обробки бази знань"
+      },
+      "generateTitle": {
+        "label": "Створювати заголовок за допомогою AI"
+      }
+    },
+    "sidepanelRag": {
+      "heading": "Чат з Copilot з налаштуваннями веб-сайту",
+      "ragEnabled": {
+        "label": "Чат з веб-сайтом за допомогою векторних вкладень"
+      },
+      "maxWebsiteContext": {
+        "label": "Розмір вмісту нормального режиму",
+        "placeholder": "Розмір вмісту (типово 4028)"
+      }
+    },
+    "webSearch": {
+      "heading": "Управління веб-пошуком",
+      "searchMode": {
+        "label": "Виконувати простий пошук в Інтернеті"
+      },
+      "provider": {
+        "label": "Пошукова система",
+        "placeholder": "Виберіть пошукову систему"
+      },
+      "totalSearchResults": {
+        "label": "Загальна кількість результатів пошуку",
+        "placeholder": "Введіть загальну кількість результатів пошуку"
+      },
+      "visitSpecificWebsite": {
+        "label": "Відвідати веб-сайт, згаданий у повідомленні"
+      }
+    },
+    "system": {
+      "heading": "Системні налаштування",
+      "deleteChatHistory": {
+        "label": "Видалити історію чату",
+        "button": "Видалити",
+        "confirm": "Ви впевнені, що хочете видалити історію чату? Ця дія не може бути відвернута."
+      },
+      "export": {
+        "label": "Експорт історії чату, бази знань та запитів",
+        "button": "Експортувати дані",
+        "success": "Експорт успішний"
+      },
+      "import": {
+        "label": "Імпорт історії чату, бази знань, та запитів",
+        "button": "Імпортувати дані",
+        "success": "Імпорт успішний",
+        "error": "Помилка імпорту"
+      }
+    },
+    "tts": {
+      "heading": "Налаштування Текст-у-Голос",
+      "ttsEnabled": {
+        "label": "Увімкнути Текст-у-Голос"
+      },
+      "ttsProvider": {
+        "label": "Поставник Текст-у-Голос",
+        "placeholder": "Виберіть постачальника"
+      },
+      "ttsVoice": {
+        "label": "Голос Текст-у-Голос",
+        "placeholder": "Виберіть голос"
+      },
+      "ssmlEnabled": {
+        "label": "Ввімкнути SSML (Мова Розмітки для Синтезу Голосу)"
+      }
+    }
+  },
+  "manageModels": {
+    "title": "Управління моделями",
+    "addBtn": "Додати нову модель",
+    "columns": {
+      "name": "Назва",
+      "digest": "Хеш",
+      "modifiedAt": "Оновлено",
+      "size": "Розмір",
+      "actions": "Дії"
+    },
+    "expandedColumns": {
+      "parentModel": "Батьківська модель",
+      "format": "Формат",
+      "family": "Сімʼя",
+      "parameterSize": "Розмір параметрів",
+      "quantizationLevel": "Рівень квантування"
+    },
+    "tooltip": {
+      "delete": "Видалити модель",
+      "repull": "Повторно завантажити модель"
+    },
+    "confirm": {
+      "delete": "Ви впевнені, що хочете видалити цю модель?",
+      "repull": "Ви впевнені, що хочете повторно завантажити цю модель?"
+    },
+    "modal": {
+      "title": "Додати нову модель",
+      "placeholder": "Введіть назву моделі",
+      "pull": "Завантажити модель"
+    },
+    "notification": {
+      "pullModel": "Завантаження моделі",
+      "pullModelDescription": "Завантажується модель {{modelName}}. Для отримання додаткової інформації перевірте іконку розширення.",
+      "success": "Успіх",
+      "error": "Помилка",
+      "successDescription": "Модель успішно завантажена",
+      "successDeleteDescription": "Модель успішно видалена",
+      "someError": "Щось пішло не так. Будь ласка, спробуйте пізніше."
+    }
+  },
+  "managePrompts": {
+    "title": "Управління запитами",
+    "addBtn": "Додати новий запит",
+    "option1": "Нормальний",
+    "option2": "RAG",
+    "questionPrompt": "Питання-запит",
+    "segmented": {
+      "custom": "Власні запити",
+      "copilot": "Запити Copilot"
+    },
+    "columns": {
+      "title": "Назва",
+      "prompt": "Запит",
+      "type": "Тип запиту",
+      "actions": "Дії"
+    },
+    "systemPrompt": "Системний запит",
+    "quickPrompt": "Швидкий запит",
+    "tooltip": {
+      "delete": "Видалити запит",
+      "edit": "Редагувати запит"
+    },
+    "confirm": {
+      "delete": "Ви впевнені, що хочете видалити цей запит? Його не вдасться відновити."
+    },
+    "modal": {
+      "addTitle": "Додати новий запит",
+      "editTitle": "Редагувати запит"
+    },
+    "form": {
+      "title": {
+        "label": "Назва",
+        "placeholder": "Мій чудовий запит",
+        "required": "Будь ласка, введіть назву"
+      },
+      "prompt": {
+        "label": "Запит",
+        "placeholder": "Введіть запит",
+        "required": "Будь ласка, введіть запит",
+        "help": "Ви можете використовувати {key} як змінну у вашому запиті.",
+        "missingTextPlaceholder": "Змінна {text} відсутня в запиті. Додайте її, будь ласка."
+      },
+      "isSystem": {
+        "label": "Це системний запит"
+      },
+      "btnSave": {
+        "saving": "Додавання запиту...",
+        "save": "Додати запит"
+      },
+      "btnEdit": {
+        "saving": "Оновлення запиту...",
+        "save": "Оновити запит"
+      }
+    },
+    "notification": {
+      "addSuccess": "Запит додано",
+      "addSuccessDesc": "Запит успішно додано",
+      "error": "Збій",
+      "someError": "Щось пішло не так. Спробуйте знову пізніше",
+      "updatedSuccess": "Запит оновлено",
+      "updatedSuccessDesc": "Запит успішно оновлено",
+      "deletedSuccess": "Запит видалено",
+      "deletedSuccessDesc": "Запит успішно видалено"
+    }
+  },
+  "manageShare": {
+    "title": "Управління спільним доступом",
+    "heading": "Налаштування URL спільного доступу до сторінки",
+    "form": {
+      "url": {
+        "label": "URL спільного доступу до сторінки",
+        "placeholder": "Введіть URL спільного доступу до сторінки",
+        "required": "Будь ласка, введіть ваш URL спільного доступу!",
+        "help": "З міркувань конфіденційності, ви можете самостійно обробляти спільний доступ до сторінки через цей URL. <anchor>Дізнатися більше</anchor>."
+      }
+    },
+    "webshare": {
+      "heading": "Спільний доступ через веб",
+      "columns": {
+        "title": "Назва",
+        "url": "URL",
+        "actions": "Дії"
+      },
+      "tooltip": {
+        "delete": "Видалити спільний доступ"
+      },
+      "confirm": {
+        "delete": "Ви впевнені, що хочете видалити цей спільний доступ? Його не вдасться відновити."
+      },
+      "label": "Управління спільним доступом до сторінки",
+      "description": "Включіть або вимкніть функцію спільного доступу до сторінки"
+    },
+    "notification": {
+      "pageShareSuccess": "URL спільного доступу до сторінки оновлено успішно",
+      "someError": "Щось пішло не так. Будь ласка, спробуйте пізніше",
+      "webShareDeleteSuccess": "Спільний доступ видалено успішно"
+    }
+  },
+  "ollamaSettings": {
+    "title": "Налаштування Ollama",
+    "heading": "Налаштуйте Ollama",
+    "settings": {
+      "ollamaUrl": {
+        "label": "URL до Ollama",
+        "placeholder": "Введіть URL до Ollama"
+      },
+      "advanced": {
+        "label": "Розширене налаштування URL Ollama",
+        "urlRewriteEnabled": {
+          "label": "Увімкнути або вимкнути налаштування власного URL походження запиту"
+        },
+        "rewriteUrl": {
+          "label": "Власний URL походження запиту",
+          "placeholder": "Введіть власний URL походження"
+        },
+        "headers": {
+          "label": "Власні заголовки",
+          "add": "Додати заголовок",
+          "key": {
+            "label": "Ключ заголовка",
+            "placeholder": "Authorization"
+          },
+          "value": {
+            "label": "Значення заголовка",
+            "placeholder": "Bearer token"
+          }
+        },
+        "help": "Якщо у вас виникають проблеми з підключенням до Ollama на Page Assist, ви можете налаштувати власний URL походження запиту. Щоб дізнатися більше про це налаштування, <anchor>натисніть тут</anchor>."
+      }
+    }
+  },
+  "manageSearch": {
+    "title": "Управління пошуком в Інтернеті",
+    "heading": "Налаштуйте пошук в Інтернеті"
+  },
+  "about": {
+    "title": "Про додаток",
+    "heading": "Інформація про цей додаток",
+    "chromeVersion": "Версія Page Assist",
+    "ollamaVersion": "Версія Ollama",
+    "support": "Ви можете підтримати проект Page Assist, зробивши пожертву або ставши спонсором через наступні платформи:",
+    "koFi": "Підтримка на Ko-fi",
+    "githubSponsor": "Стати спонсором на GitHub",
+    "githubRepo": "Репозиторій GitHub"
+  },
+  "manageKnowledge": {
+    "title": "Управління знаннями",
+    "heading": "Налаштування бази знань"
+  },
+  "rag": {
+    "title": "Налаштування RAG",
+    "ragSettings": {
+      "label": "Налаштування RAG",
+      "model": {
+        "label": "Модель вбудованих даних",
+        "required": "Будь ласка, виберіть модель",
+        "help": "Рекомендується використовувати моделі вбудованих даних, такі як `nomic-embed-text`.",
+        "placeholder": "Вибрати модель"
+      },
+      "chunkSize": {
+        "label": "Розмір шматка",
+        "placeholder": "Ввести розмір шматка",
+        "required": "Будь ласка, введіть розмір шматка"
+      },
+      "chunkOverlap": {
+        "label": "Перекриття шматків",
+        "placeholder": "Ввести перекриття шматків",
+        "required": "Будь ласка, введіть перекриття шматків"
+      },
+      "totalFilePerKB": {
+        "label": "Типовий ліміт кількості завантажень файлів до бази знань",
+        "placeholder": "Введіть типовий ліміт (напр. 10)",
+        "required": "Будь ласка, введіть типовий ліміт кількості файлів для завантаження"
+      },
+      "noOfRetrievedDocs": {
+        "label": "Кількість отриманих документів",
+        "placeholder": "Ввести кількість отриманих документів",
+        "required": "Будь ласка, введіть кількість документів"
+      }
+    },
+    "prompt": {
+      "label": "Налаштування запиту з RAG",
+      "option1": "Нормальний",
+      "option2": "З Веб-пошуком",
+      "alert": "Налаштування системного запиту тут застаріло. Використовуйте розділ «Управління запитами», щоб додавати або редагувати запити. Цей розділ буде видалено в майбутньому оновленні",
+      "systemPrompt": "Системний запит",
+      "systemPromptPlaceholder": "Ввести системний запит",
+      "webSearchPrompt": "Запит веб-пошуку",
+      "webSearchPromptHelp": "Не видаляйте `{search_results}` із запиту.",
+      "webSearchPromptError": "Будь ласка, введіть запит для веб-пошуку",
+      "webSearchPromptPlaceholder": "Ввести запит для веб-пошуку",
+      "webSearchFollowUpPrompt": "Запит для подальшого пошуку в мережі",
+      "webSearchFollowUpPromptHelp": "Не видаляйте `{chat_history}` і `{question}` із запиту.",
+      "webSearchFollowUpPromptError": "Будь ласка, введіть запит для подальшого пошуку в мережі!",
+      "webSearchFollowUpPromptPlaceholder": "Ваш запит для подальшого пошуку в мережі"
+    }
+  },
+  "chromeAiSettings": {
+    "title": "Налаштування Chrome AI"
+  }
+}
diff --git a/src/assets/locale/uk/sidepanel.json b/src/assets/locale/uk/sidepanel.json
new file mode 100644
index 0000000..ad8250a
--- /dev/null
+++ b/src/assets/locale/uk/sidepanel.json
@@ -0,0 +1,7 @@
+{
+    "tooltip": {
+        "embed": "Може знадобитися кілька хвилин для вкладення сторінки у базу. Будь ласка, зачекайте...",
+        "clear": "Стерти історію чату",
+        "history": "Історія чату"
+    }
+}
diff --git a/src/i18n/index.ts b/src/i18n/index.ts
index effddb4..5897a1e 100644
--- a/src/i18n/index.ts
+++ b/src/i18n/index.ts
@@ -3,6 +3,7 @@ import { initReactI18next } from "react-i18next";
 import { en } from "./lang/en";
 import { pt } from "./lang/pt";
 import { fr } from "./lang/fr";
+import { uk } from "./lang/uk";
 import { ru } from "./lang/ru";
 import { ml } from "./lang/ml";
 import { zh } from "./lang/zh";
@@ -28,6 +29,8 @@ i18n
             ml: ml,
             "pt-BR": pt,
             "zh-CN": zh,
+            uk: uk,
+            "uk-UA": uk,
             ru: ru,
             "ru-RU": ru,
             zh: zh,
diff --git a/src/i18n/lang/uk.ts b/src/i18n/lang/uk.ts
new file mode 100644
index 0000000..a1abf62
--- /dev/null
+++ b/src/i18n/lang/uk.ts
@@ -0,0 +1,19 @@
+import option from "@/assets/locale/uk/option.json";
+import playground from "@/assets/locale/uk/playground.json";
+import common from "@/assets/locale/uk/common.json";
+import sidepanel from "@/assets/locale/uk/sidepanel.json";
+import settings from "@/assets/locale/uk/settings.json";
+import knowledge from "@/assets/locale/uk/knowledge.json";
+import chrome from "@/assets/locale/uk/chrome.json";
+import openai from "@/assets/locale/uk/openai.json";
+
+export const uk = {
+    option,
+    playground,
+    common,
+    sidepanel,
+    settings,
+    knowledge,
+    chrome,
+    openai
+}
\ No newline at end of file
diff --git a/src/i18n/support-language.ts b/src/i18n/support-language.ts
index 273964d..4787ab6 100644
--- a/src/i18n/support-language.ts
+++ b/src/i18n/support-language.ts
@@ -16,6 +16,10 @@ export const supportLanguage = [
         label: "Italiano",
         value: "it"
     },
+    {
+        label: "Ukrainian",
+        value: "uk"
+    },
     {
         label: "Russian",
         value: "ru"

From edc5380a760b1c029a5a3bae58d4773a358d6ba2 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 23 Nov 2024 11:53:13 +0530
Subject: [PATCH 26/27] feat(i18n): add Ukrainian translations for Playground
 and Settings

- Add new "welcome" string to the Ukrainian Playground translation
- Add new "ollamaStatus" setting to the Ukrainian Settings translation
---
 src/assets/locale/uk/playground.json | 5 +++--
 src/assets/locale/uk/settings.json   | 3 +++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/assets/locale/uk/playground.json b/src/assets/locale/uk/playground.json
index a481804..29e244d 100644
--- a/src/assets/locale/uk/playground.json
+++ b/src/assets/locale/uk/playground.json
@@ -25,5 +25,6 @@
         "stopStreaming": "Зупинити трансляцію",
         "knowledge": "Знання"
     },
-    "sendWhenEnter": "Надсилати при натисканні Enter"
-}
+    "sendWhenEnter": "Надсилати при натисканні Enter",
+    "welcome": "Вітаю! Як я можу допомогти вам сьогодні?"
+}
\ No newline at end of file
diff --git a/src/assets/locale/uk/settings.json b/src/assets/locale/uk/settings.json
index 02dc3e8..db0695b 100644
--- a/src/assets/locale/uk/settings.json
+++ b/src/assets/locale/uk/settings.json
@@ -32,6 +32,9 @@
       },
       "generateTitle": {
         "label": "Створювати заголовок за допомогою AI"
+      },
+      "ollamaStatus": {
+        "label": "Увімкнути або вимкнути перевірку стану з'єднання Ollama"
       }
     },
     "sidepanelRag": {

From 2c12b17dda1bb68cfb6cba0bfd253b296b9087bb Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 23 Nov 2024 14:04:57 +0530
Subject: [PATCH 27/27] feat(vision): add vision chat mode - Add new "vision"
 chat mode to the application - Implement the `visionChatMode` function to
 handle vision-based chat interactions - Update the UI to include a new button
 to toggle the vision chat mode - Add new translations for the "vision" chat
 mode tooltip - Disable certain UI elements when the vision chat mode is
 active

---
 src/assets/locale/en/playground.json   |   3 +-
 src/components/Sidepanel/Chat/form.tsx |  70 ++++--
 src/hooks/useMessage.tsx               | 292 ++++++++++++++++++++++---
 src/libs/get-screenshot.ts             |  46 ++++
 src/store/index.tsx                    |   4 +-
 5 files changed, 368 insertions(+), 47 deletions(-)
 create mode 100644 src/libs/get-screenshot.ts

diff --git a/src/assets/locale/en/playground.json b/src/assets/locale/en/playground.json
index 2fc5041..c4f5f5c 100644
--- a/src/assets/locale/en/playground.json
+++ b/src/assets/locale/en/playground.json
@@ -23,7 +23,8 @@
         "speechToText": "Speech to Text",
         "uploadImage": "Upload Image",
         "stopStreaming": "Stop Streaming",
-        "knowledge": "Knowledge"
+        "knowledge": "Knowledge",
+        "vision": "[Experimental] Vision Chat"
     },
     "sendWhenEnter": "Send when Enter pressed",
     "welcome": "Hello! How can I help you today?"
diff --git a/src/components/Sidepanel/Chat/form.tsx b/src/components/Sidepanel/Chat/form.tsx
index 618431d..a402680 100644
--- a/src/components/Sidepanel/Chat/form.tsx
+++ b/src/components/Sidepanel/Chat/form.tsx
@@ -7,7 +7,14 @@ import { toBase64 } from "~/libs/to-base64"
 import { Checkbox, Dropdown, Image, Switch, Tooltip } from "antd"
 import { useWebUI } from "~/store/webui"
 import { defaultEmbeddingModelForRag } from "~/services/ollama"
-import { ImageIcon, MicIcon, StopCircleIcon, X } from "lucide-react"
+import {
+  ImageIcon,
+  MicIcon,
+  StopCircleIcon,
+  X,
+  EyeIcon,
+  EyeOffIcon
+} from "lucide-react"
 import { useTranslation } from "react-i18next"
 import { ModelSelect } from "@/components/Common/ModelSelect"
 import { useSpeechRecognition } from "@/hooks/useSpeechRecognition"
@@ -36,7 +43,7 @@ export const SidepanelForm = ({ dropedFile }: Props) => {
     resetTranscript,
     start: startListening,
     stop: stopSpeechRecognition,
-    supported: browserSupportsSpeechRecognition, 
+    supported: browserSupportsSpeechRecognition
   } = useSpeechRecognition()
 
   const stopListening = async () => {
@@ -237,7 +244,10 @@ export const SidepanelForm = ({ dropedFile }: Props) => {
                 }
               }
               await stopListening()
-              if (value.message.trim().length === 0 && value.image.length === 0) {
+              if (
+                value.message.trim().length === 0 &&
+                value.image.length === 0
+              ) {
                 return
               }
               form.reset()
@@ -281,20 +291,22 @@ export const SidepanelForm = ({ dropedFile }: Props) => {
                 {...form.getInputProps("message")}
               />
               <div className="flex mt-4 justify-end gap-3">
-                <Tooltip title={t("tooltip.searchInternet")}>
-                  <button
-                    type="button"
-                    onClick={() => setWebSearch(!webSearch)}
-                    className={`inline-flex items-center gap-2   ${
-                      chatMode === "rag" ? "hidden" : "block"
-                    }`}>
-                    {webSearch ? (
-                      <PiGlobe className="h-5 w-5  dark:text-gray-300" />
-                    ) : (
-                      <PiGlobeX className="h-5 w-5 text-gray-600 dark:text-gray-400" />
-                    )}
-                  </button>
-                </Tooltip>
+                {chatMode !== "vision" && (
+                  <Tooltip title={t("tooltip.searchInternet")}>
+                    <button
+                      type="button"
+                      onClick={() => setWebSearch(!webSearch)}
+                      className={`inline-flex items-center gap-2   ${
+                        chatMode === "rag" ? "hidden" : "block"
+                      }`}>
+                      {webSearch ? (
+                        <PiGlobe className="h-5 w-5  dark:text-gray-300" />
+                      ) : (
+                        <PiGlobeX className="h-5 w-5 text-gray-600 dark:text-gray-400" />
+                      )}
+                    </button>
+                  </Tooltip>
+                )}
                 <ModelSelect />
                 {browserSupportsSpeechRecognition && (
                   <Tooltip title={t("tooltip.speechToText")}>
@@ -323,13 +335,35 @@ export const SidepanelForm = ({ dropedFile }: Props) => {
                     </button>
                   </Tooltip>
                 )}
+                <Tooltip title={t("tooltip.vision")}>
+                  <button
+                    type="button"
+                    onClick={() => {
+                      if (chatMode === "vision") {
+                        setChatMode("normal")
+                      } else {
+                        setChatMode("vision")
+                      }
+                    }}
+                    disabled={chatMode === "rag"}
+                    className={`flex items-center justify-center dark:text-gray-300 ${
+                      chatMode === "rag" ? "hidden" : "block"
+                    } disabled:opacity-50`}>
+                    {chatMode === "vision" ? (
+                      <EyeIcon className="h-5 w-5" />
+                    ) : (
+                      <EyeOffIcon className="h-5 w-5" />
+                    )}
+                  </button>
+                </Tooltip>
                 <Tooltip title={t("tooltip.uploadImage")}>
                   <button
                     type="button"
                     onClick={() => {
                       inputRef.current?.click()
                     }}
-                    className={`flex items-center justify-center dark:text-gray-300 ${
+                    disabled={chatMode === "vision"}
+                    className={`flex items-center justify-center disabled:opacity-50 dark:text-gray-300 ${
                       chatMode === "rag" ? "hidden" : "block"
                     }`}>
                     <ImageIcon className="h-5 w-5" />
diff --git a/src/hooks/useMessage.tsx b/src/hooks/useMessage.tsx
index 6fb2175..eb17f3e 100644
--- a/src/hooks/useMessage.tsx
+++ b/src/hooks/useMessage.tsx
@@ -36,6 +36,7 @@ import { humanMessageFormatter } from "@/utils/human-message"
 import { pageAssistEmbeddingModel } from "@/models/embedding"
 import { PageAssistVectorStore } from "@/libs/PageAssistVectorStore"
 import { PAMemoryVectorStore } from "@/libs/PAMemoryVectorStore"
+import { getScreenshotFromCurrentTab } from "@/libs/get-screenshot"
 
 export const useMessage = () => {
   const {
@@ -136,8 +137,9 @@ export const useMessage = () => {
       seed: currentChatModelSettings?.seed,
       numGpu:
         currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
-
+      numPredict:
+        currentChatModelSettings?.numPredict ??
+        userDefaultModelSettings?.numPredict
     })
 
     let newMessage: Message[] = []
@@ -265,9 +267,11 @@ export const useMessage = () => {
             userDefaultModelSettings?.numCtx,
           seed: currentChatModelSettings?.seed,
           numGpu:
-            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
-
+            currentChatModelSettings?.numGpu ??
+            userDefaultModelSettings?.numGpu,
+          numPredict:
+            currentChatModelSettings?.numPredict ??
+            userDefaultModelSettings?.numPredict
         })
         const response = await questionOllama.invoke(promptForQuestion)
         query = response.content.toString()
@@ -342,9 +346,7 @@ export const useMessage = () => {
           signal: signal,
           callbacks: [
             {
-              handleLLMEnd(
-                output: any,
-              ): any {
+              handleLLMEnd(output: any): any {
                 try {
                   generationInfo = output?.generations?.[0][0]?.generationInfo
                 } catch (e) {
@@ -450,6 +452,236 @@ export const useMessage = () => {
     }
   }
 
+  const visionChatMode = async (
+    message: string,
+    image: string,
+    isRegenerate: boolean,
+    messages: Message[],
+    history: ChatHistory,
+    signal: AbortSignal
+  ) => {
+    setStreaming(true)
+    const url = await getOllamaURL()
+    const userDefaultModelSettings = await getAllDefaultModelSettings()
+
+    const ollama = await pageAssistModel({
+      model: selectedModel!,
+      baseUrl: cleanUrl(url),
+      keepAlive:
+        currentChatModelSettings?.keepAlive ??
+        userDefaultModelSettings?.keepAlive,
+      temperature:
+        currentChatModelSettings?.temperature ??
+        userDefaultModelSettings?.temperature,
+      topK: currentChatModelSettings?.topK ?? userDefaultModelSettings?.topK,
+      topP: currentChatModelSettings?.topP ?? userDefaultModelSettings?.topP,
+      numCtx:
+        currentChatModelSettings?.numCtx ?? userDefaultModelSettings?.numCtx,
+      seed: currentChatModelSettings?.seed,
+      numGpu:
+        currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
+      numPredict:
+        currentChatModelSettings?.numPredict ??
+        userDefaultModelSettings?.numPredict
+    })
+
+    let newMessage: Message[] = []
+    let generateMessageId = generateID()
+
+    if (!isRegenerate) {
+      newMessage = [
+        ...messages,
+        {
+          isBot: false,
+          name: "You",
+          message,
+          sources: [],
+          images: []
+        },
+        {
+          isBot: true,
+          name: selectedModel,
+          message: "▋",
+          sources: [],
+          id: generateMessageId
+        }
+      ]
+    } else {
+      newMessage = [
+        ...messages,
+        {
+          isBot: true,
+          name: selectedModel,
+          message: "▋",
+          sources: [],
+          id: generateMessageId
+        }
+      ]
+    }
+    setMessages(newMessage)
+    let fullText = ""
+    let contentToSave = ""
+
+    try {
+      const prompt = await systemPromptForNonRag()
+      const selectedPrompt = await getPromptById(selectedSystemPrompt)
+
+      const applicationChatHistory = generateHistory(history, selectedModel)
+
+      const data = await getScreenshotFromCurrentTab()
+      console.log(
+        data?.success
+          ? `[PageAssist] Screenshot is taken`
+          : `[PageAssist] Screenshot is not taken`
+      )
+      const visionImage = data?.screenshot || ""
+
+      if (visionImage === "") {
+        throw new Error(
+          "Please close and reopen the side panel. This is a bug that will be fixed soon."
+        )
+      }
+
+      if (prompt && !selectedPrompt) {
+        applicationChatHistory.unshift(
+          new SystemMessage({
+            content: prompt
+          })
+        )
+      }
+      if (selectedPrompt) {
+        applicationChatHistory.unshift(
+          new SystemMessage({
+            content: selectedPrompt.content
+          })
+        )
+      }
+
+      let humanMessage = humanMessageFormatter({
+        content: [
+          {
+            text: message,
+            type: "text"
+          },
+          {
+            image_url: visionImage,
+            type: "image_url"
+          }
+        ],
+        model: selectedModel
+      })
+
+      let generationInfo: any | undefined = undefined
+
+      const chunks = await ollama.stream(
+        [...applicationChatHistory, humanMessage],
+        {
+          signal: signal,
+          callbacks: [
+            {
+              handleLLMEnd(output: any): any {
+                try {
+                  generationInfo = output?.generations?.[0][0]?.generationInfo
+                } catch (e) {
+                  console.log("handleLLMEnd error", e)
+                }
+              }
+            }
+          ]
+        }
+      )
+      let count = 0
+      for await (const chunk of chunks) {
+        contentToSave += chunk?.content
+        fullText += chunk?.content
+        if (count === 0) {
+          setIsProcessing(true)
+        }
+        setMessages((prev) => {
+          return prev.map((message) => {
+            if (message.id === generateMessageId) {
+              return {
+                ...message,
+                message: fullText + "▋"
+              }
+            }
+            return message
+          })
+        })
+        count++
+      }
+      setMessages((prev) => {
+        return prev.map((message) => {
+          if (message.id === generateMessageId) {
+            return {
+              ...message,
+              message: fullText,
+              generationInfo
+            }
+          }
+          return message
+        })
+      })
+
+      setHistory([
+        ...history,
+        {
+          role: "user",
+          content: message
+        },
+        {
+          role: "assistant",
+          content: fullText
+        }
+      ])
+
+      await saveMessageOnSuccess({
+        historyId,
+        setHistoryId,
+        isRegenerate,
+        selectedModel: selectedModel,
+        message,
+        image,
+        fullText,
+        source: [],
+        message_source: "copilot",
+        generationInfo
+      })
+
+      setIsProcessing(false)
+      setStreaming(false)
+    } catch (e) {
+      const errorSave = await saveMessageOnError({
+        e,
+        botMessage: fullText,
+        history,
+        historyId,
+        image,
+        selectedModel,
+        setHistory,
+        setHistoryId,
+        userMessage: message,
+        isRegenerating: isRegenerate,
+        message_source: "copilot"
+      })
+
+      if (!errorSave) {
+        notification.error({
+          message: t("error"),
+          description: e?.message || t("somethingWentWrong")
+        })
+      }
+      setIsProcessing(false)
+      setStreaming(false)
+      setIsProcessing(false)
+      setStreaming(false)
+      setIsEmbedding(false)
+    } finally {
+      setAbortController(null)
+      setEmbeddingController(null)
+    }
+  }
+
   const normalChatMode = async (
     message: string,
     image: string,
@@ -482,8 +714,9 @@ export const useMessage = () => {
       seed: currentChatModelSettings?.seed,
       numGpu:
         currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
-
+      numPredict:
+        currentChatModelSettings?.numPredict ??
+        userDefaultModelSettings?.numPredict
     })
 
     let newMessage: Message[] = []
@@ -577,9 +810,7 @@ export const useMessage = () => {
           signal: signal,
           callbacks: [
             {
-              handleLLMEnd(
-                output: any,
-              ): any {
+              handleLLMEnd(output: any): any {
                 try {
                   generationInfo = output?.generations?.[0][0]?.generationInfo
                 } catch (e) {
@@ -711,8 +942,9 @@ export const useMessage = () => {
       seed: currentChatModelSettings?.seed,
       numGpu:
         currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
-
+      numPredict:
+        currentChatModelSettings?.numPredict ??
+        userDefaultModelSettings?.numPredict
     })
 
     let newMessage: Message[] = []
@@ -787,9 +1019,11 @@ export const useMessage = () => {
             userDefaultModelSettings?.numCtx,
           seed: currentChatModelSettings?.seed,
           numGpu:
-            currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
-
+            currentChatModelSettings?.numGpu ??
+            userDefaultModelSettings?.numGpu,
+          numPredict:
+            currentChatModelSettings?.numPredict ??
+            userDefaultModelSettings?.numPredict
         })
         const response = await questionOllama.invoke(promptForQuestion)
         query = response.content.toString()
@@ -842,9 +1076,7 @@ export const useMessage = () => {
           signal: signal,
           callbacks: [
             {
-              handleLLMEnd(
-                output: any,
-              ): any {
+              handleLLMEnd(output: any): any {
                 try {
                   generationInfo = output?.generations?.[0][0]?.generationInfo
                 } catch (e) {
@@ -977,8 +1209,9 @@ export const useMessage = () => {
       seed: currentChatModelSettings?.seed,
       numGpu:
         currentChatModelSettings?.numGpu ?? userDefaultModelSettings?.numGpu,
-       numPredict: currentChatModelSettings?.numPredict ?? userDefaultModelSettings?.numPredict,
-
+      numPredict:
+        currentChatModelSettings?.numPredict ??
+        userDefaultModelSettings?.numPredict
     })
 
     let newMessage: Message[] = []
@@ -1052,9 +1285,7 @@ export const useMessage = () => {
         signal: signal,
         callbacks: [
           {
-            handleLLMEnd(
-              output: any,
-            ): any {
+            handleLLMEnd(output: any): any {
               try {
                 generationInfo = output?.generations?.[0][0]?.generationInfo
               } catch (e) {
@@ -1216,6 +1447,15 @@ export const useMessage = () => {
             signal
           )
         }
+      } else if (chatMode === "vision") {
+        await visionChatMode(
+          message,
+          image,
+          isRegenerate,
+          chatHistory || messages,
+          memory || history,
+          signal
+        )
       } else {
         const newEmbeddingController = new AbortController()
         let embeddingSignal = newEmbeddingController.signal
diff --git a/src/libs/get-screenshot.ts b/src/libs/get-screenshot.ts
new file mode 100644
index 0000000..d3cbb33
--- /dev/null
+++ b/src/libs/get-screenshot.ts
@@ -0,0 +1,46 @@
+const captureVisibleTab = () => {
+  const result = new Promise<string>((resolve) => {
+    if (import.meta.env.BROWSER === "chrome") {
+      chrome.tabs.query({ active: true, currentWindow: true }, async (tabs) => {
+        const tab = tabs[0]
+        chrome.tabs.captureVisibleTab(null, { format: "png" }, (dataUrl) => {
+          resolve(dataUrl)
+        })
+      })
+    } else {
+      browser.tabs
+        .query({ active: true, currentWindow: true })
+        .then(async (tabs) => {
+          const dataUrl = (await Promise.race([
+            browser.tabs.captureVisibleTab(null, { format: "png" }),
+            new Promise((_, reject) =>
+              setTimeout(
+                () => reject(new Error("Screenshot capture timed out")),
+                10000
+              )
+            )
+          ])) as string
+          resolve(dataUrl)
+        })
+    }
+  })
+  return result
+}
+
+export const getScreenshotFromCurrentTab = async () => {
+  try {
+    const screenshotDataUrl = await captureVisibleTab()
+    return {
+      success: true,
+      screenshot: screenshotDataUrl,
+      error: null
+    }
+  } catch (error) {
+    return {
+      success: false,
+      screenshot: null,
+      error:
+        error instanceof Error ? error.message : "Failed to capture screenshot"
+    }
+  }
+}
diff --git a/src/store/index.tsx b/src/store/index.tsx
index 02bf2cb..9bdee7b 100644
--- a/src/store/index.tsx
+++ b/src/store/index.tsx
@@ -32,8 +32,8 @@ type State = {
   setIsProcessing: (isProcessing: boolean) => void
   selectedModel: string | null
   setSelectedModel: (selectedModel: string) => void
-  chatMode: "normal" | "rag"
-  setChatMode: (chatMode: "normal" | "rag") => void
+  chatMode: "normal" | "rag" | "vision"
+  setChatMode: (chatMode: "normal" | "rag" | "vision") => void
   isEmbedding: boolean
   setIsEmbedding: (isEmbedding: boolean) => void
   speechToTextLanguage: string