}>
- Default} key={"0"}>
- {[...(adapterData || [])].map((adapter, index) => (
+ {!isSinglePass && (
+ Default} key={"0"}>
+ )}
+ {adapterData.map((adapter, index) => (
{adapter.llm_model}}
key={(index + 1)?.toString()}
@@ -54,6 +57,7 @@ JsonView.propTypes = {
selectedProfile: PropTypes.string,
llmProfiles: PropTypes.array,
activeKey: PropTypes.string,
+ isSinglePass: PropTypes.bool,
};
export { JsonView };
diff --git a/frontend/src/components/custom-tools/manage-llm-profiles/ManageLlmProfiles.jsx b/frontend/src/components/custom-tools/manage-llm-profiles/ManageLlmProfiles.jsx
index 643abe2e3..31e9f5304 100644
--- a/frontend/src/components/custom-tools/manage-llm-profiles/ManageLlmProfiles.jsx
+++ b/frontend/src/components/custom-tools/manage-llm-profiles/ManageLlmProfiles.jsx
@@ -1,5 +1,5 @@
import { DeleteOutlined, EditOutlined } from "@ant-design/icons";
-import { Button, Radio, Table, Typography } from "antd";
+import { Button, Radio, Table, Tooltip, Typography } from "antd";
import { useEffect, useState } from "react";
import { useAxiosPrivate } from "../../../hooks/useAxiosPrivate";
@@ -75,6 +75,8 @@ function ManageLlmProfiles() {
const { setAlertDetails } = useAlertStore();
const handleException = useExceptionHandler();
const { setPostHogCustomEvent } = usePostHogEvents();
+ const MAX_PROFILE_COUNT = 4;
+ const isMaxProfile = llmProfiles.length >= MAX_PROFILE_COUNT;
const handleDefaultLlm = (profileId) => {
try {
@@ -239,13 +241,21 @@ function ManageLlmProfiles() {
-
- Add New LLM Profile
-
+
+ Add New LLM Profile
+
+
);
diff --git a/frontend/src/components/custom-tools/output-for-doc-modal/OutputForDocModal.jsx b/frontend/src/components/custom-tools/output-for-doc-modal/OutputForDocModal.jsx
index 0d053e4f3..446a9ffb3 100644
--- a/frontend/src/components/custom-tools/output-for-doc-modal/OutputForDocModal.jsx
+++ b/frontend/src/components/custom-tools/output-for-doc-modal/OutputForDocModal.jsx
@@ -31,25 +31,6 @@ try {
} catch {
// The component will remain null of it is not available
}
-const columns = [
- {
- title: "Document",
- dataIndex: "document",
- key: "document",
- },
- {
- title: "Token Count",
- dataIndex: "token_count",
- key: "token_count",
- width: 200,
- },
- {
- title: "Value",
- dataIndex: "value",
- key: "value",
- width: 600,
- },
-];
const outputStatus = {
yet_to_process: "YET_TO_PROCESS",
@@ -75,13 +56,14 @@ function OutputForDocModal({
details,
listOfDocs,
selectedDoc,
- defaultLlmProfile,
disableLlmOrDocChange,
singlePassExtractMode,
isSinglePassExtractLoading,
isPublicSource,
llmProfiles,
+ defaultLlmProfile,
} = useCustomToolStore();
+ const [selectedProfile, setSelectedProfile] = useState(defaultLlmProfile);
const { sessionDetails } = useSessionStore();
const axiosPrivate = useAxiosPrivate();
const navigate = useNavigate();
@@ -89,13 +71,12 @@ function OutputForDocModal({
const { setAlertDetails } = useAlertStore();
const handleException = useExceptionHandler();
const { tokenUsage } = useTokenUsageStore();
- const [selectedProfile, setSelectedProfile] = useState(defaultLlmProfile);
useEffect(() => {
if (!open) {
return;
}
- handleGetOutputForDocs();
+ handleGetOutputForDocs(selectedProfile || profileManagerId);
getAdapterInfo();
}, [open, singlePassExtractMode, isSinglePassExtractLoading]);
@@ -139,34 +120,6 @@ function OutputForDocModal({
// If data is provided, use it; otherwise, create a copy of the previous state
const updatedPromptOutput = data || [...prev];
- // Get the keys of docOutputs
- const keys = Object.keys(docOutputs);
-
- keys.forEach((key) => {
- // Find the index of the prompt output corresponding to the document manager key
- const index = updatedPromptOutput.findIndex(
- (promptOutput) => promptOutput?.document_manager === key
- );
-
- let promptOutputInstance = {};
- // If the prompt output for the current key doesn't exist, skip it
- if (index > -1) {
- promptOutputInstance = updatedPromptOutput[index];
- promptOutputInstance["output"] = docOutputs[key]?.output;
- }
-
- // Update output and isLoading properties based on docOutputs
- promptOutputInstance["document_manager"] = key;
- promptOutputInstance["isLoading"] = docOutputs[key]?.isLoading || false;
-
- // Update the prompt output instance in the array
- if (index > -1) {
- updatedPromptOutput[index] = promptOutputInstance;
- } else {
- updatedPromptOutput.push(promptOutputInstance);
- }
- });
-
return updatedPromptOutput;
});
};
@@ -181,10 +134,6 @@ function OutputForDocModal({
};
const handleGetOutputForDocs = (profile = profileManagerId) => {
- if (singlePassExtractMode) {
- profile = defaultLlmProfile;
- }
-
if (!profile) {
setRows([]);
return;
@@ -243,10 +192,14 @@ function OutputForDocModal({
const result = {
key: item?.document_id,
document: item?.document_name,
- token_count: (
+ token_count: !singlePassExtractMode && (