From 000c863e95e5ed73c005a7e255f42369f155de43 Mon Sep 17 00:00:00 2001
From: Saoud Rizwan <7799382+saoudrizwan@users.noreply.github.com>
Date: Tue, 10 Sep 2024 18:31:55 -0400
Subject: [PATCH] Fix task header for ollama/openai

---
 webview-ui/src/components/TaskHeader.tsx | 35 +++++++++++++-----------
 1 file changed, 19 insertions(+), 16 deletions(-)

diff --git a/webview-ui/src/components/TaskHeader.tsx b/webview-ui/src/components/TaskHeader.tsx
index c7b66be..f05e86d 100644
--- a/webview-ui/src/components/TaskHeader.tsx
+++ b/webview-ui/src/components/TaskHeader.tsx
@@ -140,22 +140,25 @@ const TaskHeader: React.FC<TaskHeaderProps> = ({
 							{!isTaskExpanded && <span style={{ marginLeft: 4 }}>{task.text}</span>}
 						</div>
 					</div>
-					{!isTaskExpanded && (
-						<div
-							style={{
-								marginLeft: 10,
-								backgroundColor: "color-mix(in srgb, var(--vscode-badge-foreground) 70%, transparent)",
-								color: "var(--vscode-badge-background)",
-								padding: "2px 4px",
-								borderRadius: "500px",
-								fontSize: "11px",
-								fontWeight: 500,
-								display: "inline-block",
-								flexShrink: 0,
-							}}>
-							${totalCost?.toFixed(4)}
-						</div>
-					)}
+					{!isTaskExpanded &&
+						apiConfiguration?.apiProvider !== "openai" &&
+						apiConfiguration?.apiProvider !== "ollama" && (
+							<div
+								style={{
+									marginLeft: 10,
+									backgroundColor:
+										"color-mix(in srgb, var(--vscode-badge-foreground) 70%, transparent)",
+									color: "var(--vscode-badge-background)",
+									padding: "2px 4px",
+									borderRadius: "500px",
+									fontSize: "11px",
+									fontWeight: 500,
+									display: "inline-block",
+									flexShrink: 0,
+								}}>
+								${totalCost?.toFixed(4)}
+							</div>
+						)}
 					<VSCodeButton appearance="icon" onClick={onClose} style={{ marginLeft: 6, flexShrink: 0 }}>
 						<span className="codicon codicon-close"></span>
 					</VSCodeButton>