chore: add comments and clean unused imports
This commit is contained in:
parent
8f778ee90f
commit
90c7420c34
@ -200,7 +200,6 @@ const ModelDropdown = ({
|
|||||||
if (model)
|
if (model)
|
||||||
updateModelParameter(activeThread, {
|
updateModelParameter(activeThread, {
|
||||||
params: modelParams,
|
params: modelParams,
|
||||||
// modelPath: model.file_path,
|
|
||||||
modelId: model.id,
|
modelId: model.id,
|
||||||
engine: model.engine,
|
engine: model.engine,
|
||||||
})
|
})
|
||||||
|
|||||||
@ -1,6 +1,5 @@
|
|||||||
import { memo } from 'react'
|
import { memo } from 'react'
|
||||||
|
|
||||||
import { InferenceEngine } from '@janhq/core'
|
|
||||||
import { Button } from '@janhq/joi'
|
import { Button } from '@janhq/joi'
|
||||||
import { useAtomValue, useSetAtom } from 'jotai'
|
import { useAtomValue, useSetAtom } from 'jotai'
|
||||||
|
|
||||||
|
|||||||
@ -1,6 +1,6 @@
|
|||||||
/* eslint-disable @typescript-eslint/naming-convention */
|
/* eslint-disable @typescript-eslint/naming-convention */
|
||||||
|
|
||||||
import { Fragment, useEffect, useState } from 'react'
|
import { useEffect, useState } from 'react'
|
||||||
|
|
||||||
import { Accept, useDropzone } from 'react-dropzone'
|
import { Accept, useDropzone } from 'react-dropzone'
|
||||||
|
|
||||||
|
|||||||
@ -1,5 +1,13 @@
|
|||||||
import { AllQuantizations, getFileSize, HuggingFaceRepoData } from '@janhq/core'
|
import { AllQuantizations, getFileSize, HuggingFaceRepoData } from '@janhq/core'
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Fetches data from a Hugging Face repository.
|
||||||
|
*
|
||||||
|
* @param repoId - The ID of the Hugging Face repository.
|
||||||
|
* @param huggingFaceAccessToken - Optional access token for Hugging Face API.
|
||||||
|
* @returns A promise that resolves to the HuggingFaceRepoData.
|
||||||
|
* @throws Will throw an error if the repository is not supported or if there is an error in the response.
|
||||||
|
*/
|
||||||
export const fetchHuggingFaceRepoData = async (
|
export const fetchHuggingFaceRepoData = async (
|
||||||
repoId: string,
|
repoId: string,
|
||||||
huggingFaceAccessToken?: string
|
huggingFaceAccessToken?: string
|
||||||
@ -60,31 +68,53 @@ export const fetchHuggingFaceRepoData = async (
|
|||||||
return data
|
return data
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Converts a repository ID or URL to a valid Hugging Face API URL.
|
||||||
|
*
|
||||||
|
* @param repoId - The repository ID or URL to convert.
|
||||||
|
* @returns A string representing the Hugging Face API URL.
|
||||||
|
* @throws {InvalidHostError} If the URL is invalid or not from huggingface.co.
|
||||||
|
* @throws {Error} If the URL cannot be parsed.
|
||||||
|
*/
|
||||||
export function toHuggingFaceUrl(repoId: string): string {
|
export function toHuggingFaceUrl(repoId: string): string {
|
||||||
try {
|
try {
|
||||||
|
// Attempt to create a URL object from the repoId
|
||||||
const url = new URL(repoId)
|
const url = new URL(repoId)
|
||||||
|
|
||||||
|
// Check if the host is huggingface.co
|
||||||
if (url.host !== 'huggingface.co') {
|
if (url.host !== 'huggingface.co') {
|
||||||
throw new InvalidHostError(`Invalid Hugging Face repo URL: ${repoId}`)
|
throw new InvalidHostError(`Invalid Hugging Face repo URL: ${repoId}`)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Split the pathname into parts and filter out empty strings
|
||||||
const paths = url.pathname.split('/').filter((e) => e.trim().length > 0)
|
const paths = url.pathname.split('/').filter((e) => e.trim().length > 0)
|
||||||
|
|
||||||
|
// Ensure there are at least two parts in the path (user/repo)
|
||||||
if (paths.length < 2) {
|
if (paths.length < 2) {
|
||||||
throw new InvalidHostError(`Invalid Hugging Face repo URL: ${repoId}`)
|
throw new InvalidHostError(`Invalid Hugging Face repo URL: ${repoId}`)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Construct and return the API URL
|
||||||
return `${url.origin}/api/models/${paths[0]}/${paths[1]}`
|
return `${url.origin}/api/models/${paths[0]}/${paths[1]}`
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
|
// Re-throw InvalidHostError if it was caught
|
||||||
if (err instanceof InvalidHostError) {
|
if (err instanceof InvalidHostError) {
|
||||||
throw err
|
throw err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// If repoId starts with 'https' but couldn't be parsed, throw an error
|
||||||
if (repoId.startsWith('https')) {
|
if (repoId.startsWith('https')) {
|
||||||
throw new Error(`Cannot parse url: ${repoId}`)
|
throw new Error(`Cannot parse url: ${repoId}`)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// If repoId is not a URL, assume it's a valid repo ID and construct the API URL
|
||||||
return `https://huggingface.co/api/models/${repoId}`
|
return `https://huggingface.co/api/models/${repoId}`
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Error thrown when the host of a URL is invalid or not from huggingface.co.
|
||||||
|
*/
|
||||||
export class InvalidHostError extends Error {
|
export class InvalidHostError extends Error {
|
||||||
constructor(message: string) {
|
constructor(message: string) {
|
||||||
super(message)
|
super(message)
|
||||||
|
|||||||
@ -1,3 +1,9 @@
|
|||||||
|
/**
|
||||||
|
* Extracts and normalizes the model ID from a given download URL.
|
||||||
|
*
|
||||||
|
* @param downloadUrl - The URL from which to extract the model ID.
|
||||||
|
* @returns The extracted model ID, or the original URL if extraction fails.
|
||||||
|
*/
|
||||||
export const normalizeModelId = (downloadUrl: string): string => {
|
export const normalizeModelId = (downloadUrl: string): string => {
|
||||||
return downloadUrl.split('/').pop() ?? downloadUrl
|
return downloadUrl.split('/').pop() ?? downloadUrl
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user