1 files changed, 2 insertions, 3 deletions
diff --git a/packages/web/src/pages/s/[id].astro b/packages/web/src/pages/s/[id].astro
index a81d2e786..46cdd2484 100644
--- a/packages/web/src/pages/s/[id].astro
+++ b/packages/web/src/pages/s/[id].astro
@@ -23,9 +23,8 @@ const models: Set<string> = new Set();
 const version = data.info.version ? `v${data.info.version}` : "v0.0.1";
 
 Object.values(data.messages).forEach((d) => {
-  const assistant = d.metadata?.assistant;
-  if (assistant) {
-    models.add(assistant.modelID);
+  if (d.role === "assistant" && d.modelID) {
+    models.add(d.modelID);
   }
 });