Merge pull request #370 from miurla/fix-multiple-tool-calls

Fix display of multiple tool calls
miurla · Oct 24, 2024 · 69e8c94 · 69e8c94
2 parents df38a2c + 393e596
commit 69e8c94
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/components/default-skeleton.tsx b/components/default-skeleton.tsx
@@ -5,7 +5,7 @@ import { Skeleton } from './ui/skeleton'
 
 export const DefaultSkeleton = () => {
   return (
-    <div className="flex flex-col gap-2">
+    <div className="flex flex-col gap-2 pb-4">
       <Skeleton className="h-6 w-48" />
       <Skeleton className="w-full h-6" />
     </div>

diff --git a/lib/actions/workflow.tsx b/lib/actions/workflow.tsx
@@ -59,6 +59,9 @@ export async function workflow(
   // Set the collapsed state to true
   isCollapsed.done(true)
 
+  // Remove the spinner
+  uiStream.update(null)
+
   const useOllama = process.env.OLLAMA_MODEL && process.env.OLLAMA_BASE_URL
   // Select the appropriate researcher function based on the environment variables
   const { text, toolResults } = useOllama

diff --git a/lib/agents/tools/search.tsx b/lib/agents/tools/search.tsx
@@ -27,7 +27,7 @@ export const searchTool = ({ uiStream, fullResponse }: ToolProps) =>
       let hasError = false
       // Append the search section
       const streamResults = createStreamableValue<string>()
-      uiStream.update(
+      uiStream.append(
         <SearchSection
           result={streamResults.value}
           includeDomains={include_domains}