Update dependencies and refactor useMessage hook

This commit is contained in:
n4ze3m
2024-03-24 23:56:41 +05:30
parent 01d27fd1c2
commit 2381588e72
6 changed files with 752 additions and 518 deletions

View File

@@ -16,7 +16,7 @@ import {
type MessageContent,
SystemMessage
} from "@langchain/core/messages"
import { getHtmlOfCurrentTab } from "~/libs/get-html"
import { getDataFromCurrentTab } from "~/libs/get-html"
import { PageAssistHtmlLoader } from "~/loader/html"
import { RecursiveCharacterTextSplitter } from "langchain/text_splitter"
import { OllamaEmbeddings } from "@langchain/community/embeddings/ollama"
@@ -167,7 +167,7 @@ export const useMessage = () => {
let isAlreadyExistEmbedding: MemoryVectorStore
let embedURL: string, embedHTML: string
if (messages.length === 0) {
const { html, url } = await getHtmlOfCurrentTab()
const { content: html, url, type } = await getDataFromCurrentTab()
embedHTML = html
embedURL = url
setCurrentURL(url)