pre-commit fixes

Zipstack · Jun 20, 2024 · 87472f4 · 87472f4
1 parent 6a9ffe7
commit 87472f4
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/backend/prompt_studio/prompt_studio_output_manager/output_manager_helper.py b/backend/prompt_studio/prompt_studio_output_manager/output_manager_helper.py
@@ -37,7 +37,8 @@ def handle_prompt_output_update(
             outputs (Any): Outputs corresponding to the prompts.
             document_id (str): ID of the document.
             profile_manager_id (Optional[str]): UUID of the profile manager.
-            is_single_pass_extract (bool): Flag indicating if single pass extract is active.
+            is_single_pass_extract (bool): 
+            Flag indicating if single pass extract is active.
             tool (CustomTool, optional): Custom tool used for extracting.
         """
 
@@ -75,11 +76,13 @@ def update_or_create_prompt_output(
 
                 if success:
                     logger.info(
-                        f"Created record for prompt_id: {prompt.prompt_id} and profile {profile_manager.profile_id}"
+                        f"Created record for prompt_id: {prompt.prompt_id} and "
+                        f"profile {profile_manager.profile_id}"                    
                     )
                 else:
                     logger.info(
-                        f"Updated record for prompt_id: {prompt.prompt_id} and profile {profile_manager.profile_id}"
+                        f"Updated record for prompt_id: {prompt.prompt_id} and "
+                        f"profile {profile_manager.profile_id}"
                     )
 
                 args: dict[str, str] = {

diff --git a/frontend/src/components/custom-tools/prompt-card/OutputForIndex.jsx b/frontend/src/components/custom-tools/prompt-card/OutputForIndex.jsx
@@ -7,12 +7,16 @@ import TabPane from "antd/es/tabs/TabPane";
 import { TextViewerPre } from "../text-viewer-pre/TextViewerPre";
 import "./PromptCard.css";
 import { SpinnerLoader } from "../../widgets/spinner-loader/SpinnerLoader";
+import { useExceptionHandler } from "../../../hooks/useExceptionHandler";
+import { useAlertStore } from "../../../store/alert-store";
 
 function OutputForIndex({ llmProfileId, isIndexOpen, setIsIndexOpen }) {
   const [data, setData] = useState(null);
   const [loading, setLoading] = useState(false);
   const { sessionDetails } = useSessionStore();
   const axiosPrivate = useAxiosPrivate();
+  const handleException = useExceptionHandler();
+  const { setAlertDetails } = useAlertStore();
 
   const fetchData = () => {
     setLoading(true);

diff --git a/frontend/src/components/custom-tools/prompt-card/PromptCardItems.jsx b/frontend/src/components/custom-tools/prompt-card/PromptCardItems.jsx
@@ -37,6 +37,8 @@ import { useAxiosPrivate } from "../../../hooks/useAxiosPrivate";
 import { useSessionStore } from "../../../store/session-store";
 import { motion, AnimatePresence } from "framer-motion";
 import { OutputForIndex } from "./OutputForIndex";
+import { useExceptionHandler } from "../../../hooks/useExceptionHandler";
+import { useAlertStore } from "../../../store/alert-store";
 
 const EvalBtn = null;
 const EvalMetrics = null;
@@ -107,6 +109,8 @@ function PromptCardItems({
   const privateAxios = useAxiosPrivate();
   const { sessionDetails } = useSessionStore();
   const { width: windowWidth } = useWindowDimensions();
+  const handleException = useExceptionHandler();
+  const { setAlertDetails } = useAlertStore();
   const componentWidth = windowWidth * 0.4;
 
   const divRef = useRef(null);