From 7c72ccfe34439aa9315c25d63f9f1110542b545b Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Thu, 27 Mar 2025 23:51:40 -0700
Subject: [PATCH] refac

---
 servers/filesystem/main.py | 16 +++++++++++++++-
 servers/time/main.py       | 15 +++++++++++++++
 2 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/servers/filesystem/main.py b/servers/filesystem/main.py
index 6d12935..5644750 100644
--- a/servers/filesystem/main.py
+++ b/servers/filesystem/main.py
@@ -1,5 +1,8 @@
 from fastapi import FastAPI, HTTPException, Body
 from fastapi.responses import PlainTextResponse
+from fastapi.middleware.cors import CORSMiddleware
+
+
 from pydantic import BaseModel, Field
 import os
 import pathlib
@@ -9,10 +12,21 @@ import difflib
 
 app = FastAPI(
     title="Secure Filesystem API",
-    version="0.2.0",
+    version="0.1.0",
     description="A secure file manipulation server for reading, editing, writing, listing, and searching files with access restrictions.",
 )
 
+origins = ["*"]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
 # Constants
 ALLOWED_DIRECTORIES = [
     str(pathlib.Path(os.path.expanduser("~/mydir")).resolve())
diff --git a/servers/time/main.py b/servers/time/main.py
index 24c550d..ee6e96b 100644
--- a/servers/time/main.py
+++ b/servers/time/main.py
@@ -1,4 +1,7 @@
 from fastapi import FastAPI, HTTPException, Body
+from fastapi.middleware.cors import CORSMiddleware
+
+
 from pydantic import BaseModel, Field
 from datetime import datetime, timezone
 from typing import Literal
@@ -11,6 +14,18 @@ app = FastAPI(
     description="Provides secure UTC/local time retrieval, formatting, timezone conversion, and comparison.",
 )
 
+
+origins = ["*"]
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
 # -------------------------------
 # Pydantic models
 # -------------------------------