jan/web/hooks/useGetHuggingFaceModel.ts

import { useState } from 'react'
import { useSetAtom } from 'jotai'
import { modelLoadMoreAtom } from '@helpers/atoms/ExploreModelLoading.atom'

export default function useGetHuggingFaceModel() {
  const setLoadMoreInProgress = useSetAtom(modelLoadMoreAtom)
  const [modelList, setModelList] = useState<Product[]>([])
  const [currentOwner, setCurrentOwner] = useState<string | undefined>(
    undefined
  )

  const getHuggingFaceModel = async (owner?: string) => {
    if (!owner) {
      setModelList([])
      return
    }

    const searchParams: SearchModelParamHf = {
      search: { owner },
      limit: 5,
    }
    // const result = await searchModels(searchParams);
    // console.debug("result", JSON.stringify(result));
    // if (owner !== currentOwner) {
    //   setModelList(result.data);
    //   setCurrentOwner(owner);
    // } else {
    //   setModelList([...modelList, ...result.data]);
    // }
    setLoadMoreInProgress(false)
  }

  return { modelList, getHuggingFaceModel }
}