From 9624493b55dca92cc814815830099d70a4e79f26 Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Tue, 4 Feb 2025 23:44:51 -0800
Subject: [PATCH] refac

---
 backend/open_webui/main.py                        | 6 +++++-
 scripts/prepare-pyodide.js                        | 3 ++-
 src/lib/components/chat/Messages/CodeBlock.svelte | 3 ++-
 src/routes/+layout.svelte                         | 3 ++-
 4 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/backend/open_webui/main.py b/backend/open_webui/main.py
index 1707c8299..863f58dea 100644
--- a/backend/open_webui/main.py
+++ b/backend/open_webui/main.py
@@ -883,7 +883,11 @@ async def chat_completion(
             **(
                 {"function_calling": "native"}
                 if form_data.get("params", {}).get("function_calling") == "native"
-                or model_info.params.model_dump().get("function_calling") == "native"
+                or (
+                    model_info
+                    and model_info.params.model_dump().get("function_calling")
+                    == "native"
+                )
                 else {}
             ),
         }
diff --git a/scripts/prepare-pyodide.js b/scripts/prepare-pyodide.js
index ccc21b452..71f2a2cb2 100644
--- a/scripts/prepare-pyodide.js
+++ b/scripts/prepare-pyodide.js
@@ -11,7 +11,8 @@ const packages = [
 	'regex',
 	'sympy',
 	'tiktoken',
-	'seaborn'
+	'seaborn',
+	'pytz'
 ];
 
 import { loadPyodide } from 'pyodide';
diff --git a/src/lib/components/chat/Messages/CodeBlock.svelte b/src/lib/components/chat/Messages/CodeBlock.svelte
index 02715f31f..f4efaba73 100644
--- a/src/lib/components/chat/Messages/CodeBlock.svelte
+++ b/src/lib/components/chat/Messages/CodeBlock.svelte
@@ -132,7 +132,8 @@
 			code.includes('seaborn') ? 'seaborn' : null,
 			code.includes('sympy') ? 'sympy' : null,
 			code.includes('tiktoken') ? 'tiktoken' : null,
-			code.includes('matplotlib') ? 'matplotlib' : null
+			code.includes('matplotlib') ? 'matplotlib' : null,
+			code.includes('pytz') ? 'pytz' : null
 		].filter(Boolean);
 
 		console.log(packages);
diff --git a/src/routes/+layout.svelte b/src/routes/+layout.svelte
index b3d793ff6..76323f772 100644
--- a/src/routes/+layout.svelte
+++ b/src/routes/+layout.svelte
@@ -118,7 +118,8 @@
 			code.includes('re') ? 'regex' : null,
 			code.includes('seaborn') ? 'seaborn' : null,
 			code.includes('sympy') ? 'sympy' : null,
-			code.includes('tiktoken') ? 'tiktoken' : null
+			code.includes('tiktoken') ? 'tiktoken' : null,
+			code.includes('pytz') ? 'pytz' : null
 		].filter(Boolean);
 
 		const pyodideWorker = new PyodideWorker();