5fc1ba7067
* @janhq/plugin-core module * refactor web to use exported services from module * refactor data-plugin to provide DAL & move model logics to model management plugin * model-management in TS * add ci auto package, increate version, and publish to npm repository * chore: storage operations * chore: hybrid data-plugin esm & cjs module * chore: PouchDB Driver * chore: documentation --------- Co-authored-by: Hien To <hien@jan.ai> Co-authored-by: Service Account <service@jan.ai>
72 lines
2.4 KiB
TypeScript
72 lines
2.4 KiB
TypeScript
import React, { useCallback } from "react";
|
|
import { ModelStatus, ModelStatusComponent } from "../ModelStatusComponent";
|
|
import ModelActionMenu from "../ModelActionMenu";
|
|
import { useAtomValue } from "jotai";
|
|
import ModelActionButton, { ModelActionType } from "../ModelActionButton";
|
|
import useStartStopModel from "@/_hooks/useStartStopModel";
|
|
import useDeleteModel from "@/_hooks/useDeleteModel";
|
|
import { AssistantModel } from "@/_models/AssistantModel";
|
|
import { activeAssistantModelAtom } from "@/_helpers/atoms/Model.atom";
|
|
import { toGigabytes } from "@/_utils/converter";
|
|
|
|
type Props = {
|
|
model: AssistantModel;
|
|
};
|
|
|
|
const ModelRow: React.FC<Props> = ({ model }) => {
|
|
const { startModel, stopModel } = useStartStopModel();
|
|
const activeModel = useAtomValue(activeAssistantModelAtom);
|
|
const { deleteModel } = useDeleteModel();
|
|
|
|
let status = ModelStatus.Installed;
|
|
if (activeModel && activeModel._id === model._id) {
|
|
status = ModelStatus.Active;
|
|
}
|
|
|
|
let actionButtonType = ModelActionType.Start;
|
|
if (activeModel && activeModel._id === model._id) {
|
|
actionButtonType = ModelActionType.Stop;
|
|
}
|
|
|
|
const onModelActionClick = (action: ModelActionType) => {
|
|
if (action === ModelActionType.Start) {
|
|
startModel(model._id);
|
|
} else {
|
|
stopModel(model._id);
|
|
}
|
|
};
|
|
|
|
const onDeleteClick = useCallback(() => {
|
|
deleteModel(model);
|
|
}, [model]);
|
|
|
|
return (
|
|
<tr className="border-b border-gray-200 last:border-b-0 last:rounded-lg">
|
|
<td className="flex flex-col whitespace-nowrap px-6 py-4 text-sm font-medium text-gray-900">
|
|
{model.name}
|
|
<span className="text-gray-500 font-normal">{model.version}</span>
|
|
</td>
|
|
<td className="whitespace-nowrap px-6 py-4 text-sm text-gray-500">
|
|
<div className="flex flex-col justify-start">
|
|
<span>GGUF</span>
|
|
</div>
|
|
</td>
|
|
<td className="whitespace-nowrap px-6 py-4 text-sm text-gray-500">
|
|
{toGigabytes(model.size)}
|
|
</td>
|
|
<td className="whitespace-nowrap px-6 py-4 text-sm text-gray-500">
|
|
<ModelStatusComponent status={status} />
|
|
</td>
|
|
<ModelActionButton
|
|
type={actionButtonType}
|
|
onActionClick={onModelActionClick}
|
|
/>
|
|
<td className="relative whitespace-nowrap px-6 py-4 w-fit text-right text-sm font-medium">
|
|
<ModelActionMenu onDeleteClick={onDeleteClick} />
|
|
</td>
|
|
</tr>
|
|
);
|
|
};
|
|
|
|
export default ModelRow;
|