fix hide cot hide pp (#638)

Chainlit · Jan 7, 2024 · 1981d2a · 1981d2a
1 parent 2ccd6f9
commit 1981d2a
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 7 deletions.
diff --git a/backend/chainlit/data/__init__.py b/backend/chainlit/data/__init__.py
@@ -383,6 +383,8 @@ async def get_thread(self, thread_id: str) -> "Optional[ThreadDict]":
                     continue
                 for attachment in step.attachments:
                     elements.append(self.attachment_to_element_dict(attachment))
+                if not config.features.prompt_playground and step.generation:
+                    step.generation = None
                 steps.append(self.step_to_step_dict(step))
 
         user = None  # type: Optional["UserDict"]

diff --git a/backend/chainlit/step.py b/backend/chainlit/step.py
@@ -257,7 +257,7 @@ async def update(self):
 
         if data_layer:
             try:
-                asyncio.create_task(data_layer.update_step(step_dict))
+                asyncio.create_task(data_layer.update_step(step_dict.copy()))
             except Exception as e:
                 if self.fail_on_persist_error:
                     raise e
@@ -313,7 +313,7 @@ async def send(self):
 
         if data_layer:
             try:
-                asyncio.create_task(data_layer.create_step(step_dict))
+                asyncio.create_task(data_layer.create_step(step_dict.copy()))
                 self.persisted = True
             except Exception as e:
                 if self.fail_on_persist_error:

diff --git a/libs/react-components/src/messages/components/DetailsButton.tsx b/libs/react-components/src/messages/components/DetailsButton.tsx
@@ -27,12 +27,11 @@ const DetailsButton = ({ message, opened, onClick, loading }: Props) => {
 
   const content = message.output;
 
-  const isRunningEmptyStep = loading && !content;
+  const showDefaultLoader = loading && (!content || messageContext.hideCot);
 
-  const show = tool || isRunningEmptyStep;
-  const hide = messageContext.hideCot && !isRunningEmptyStep;
+  const show = tool || showDefaultLoader;
 
-  if (!show || hide) {
+  if (!show) {
     return null;
   }
 

diff --git a/libs/react-components/src/playground/header.tsx b/libs/react-components/src/playground/header.tsx
@@ -34,7 +34,7 @@ export default function PlaygroundHeader({
         <AccentButton
           size="large"
           target="_blank"
-          href="https://docs.chainlit.io/concepts/prompt-playground"
+          href="https://docs-prerelease.chainlit.io/advanced-features/prompt-playground"
         >
           Help
         </AccentButton>