diff --git a/packages/frontend/pages/prompts/[[...id]].tsx b/packages/frontend/pages/prompts/[[...id]].tsx
index 05e193ba..4a630831 100644
--- a/packages/frontend/pages/prompts/[[...id]].tsx
+++ b/packages/frontend/pages/prompts/[[...id]].tsx
@@ -374,7 +374,11 @@ function Playground() {
               citations = parsedLine.citations;
             }
 
-            setOutput({ ...parsedLine.choices[0]?.message, citations });
+            let output = parsedLine.choices[0]?.message;
+            if (citations) {
+              output.citations = citations;
+            }
+            setOutput(output);
             setOutputTokens(parsedLine.usage?.completion_tokens || 0);
             setError(null);
           } catch (error) {
@@ -383,8 +387,6 @@ function Playground() {
         },
       );
 
-      // setOutput({ ...output, citations });
-
       // scroll template-input-area to the end
       const element = document.getElementById("template-input-area");
       element.scrollTop = element.scrollHeight;
diff --git a/packages/shared/models.ts b/packages/shared/models.ts
index 6fbe1e4e..c1b0c937 100644
--- a/packages/shared/models.ts
+++ b/packages/shared/models.ts
@@ -203,4 +203,14 @@ export const MODELS = [
     name: "perplexity-llama-3.1-sonar-large-128k-online",
     provider: "openrouter",
   },
+  {
+    id: "deepseek/deepseek-r1",
+    name: "deepseek-r1",
+    provider: "openrouter",
+  },
+  {
+    id: "microsoft/phi-4",
+    name: "phi-4",
+    provider: "openrouter",
+  },
 ];