feat: add cors support

.env.example

@@ -2,6 +2,7 @@
 LOG_LEVEL=INFO
 API_HOST=127.0.0.1
 API_PORT=8000
+CORS_ALLOWED_ORIGINS="*" # comma-separated list of allowed origins
 AGENTS_REF='{"agent_uuid": "agent_module_name:agent_var"}'
 AGENT_MANIFEST_PATH=manifest.json
 AGWS_STORAGE_PERSIST=True

src/agent_workflow_server/agents/load.py

@@ -127,7 +127,7 @@ def _resolve_agent(name: str, path: str) -> AgentInfo:
     # Load manifest. Check in paths below (in order)
     manifest_paths = [
         os.path.join(os.path.dirname(module.__file__), "manifest.json"),
-        os.environ.get("AGENT_MANIFEST_PATH", "manifest.json") or "manifest.json",
+        os.getenv("AGENT_MANIFEST_PATH", "manifest.json") or "manifest.json",
     ]
 
     for manifest_path in manifest_paths:

src/agent_workflow_server/logging/logger.py

@@ -14,5 +14,5 @@
 
 CustomLoggerHandler = handler
 
-LOG_LEVEL = os.environ.get("LOG_LEVEL", "INFO").upper()
+LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO").upper()
 logging.basicConfig(level=LOG_LEVEL, handlers=[CustomLoggerHandler], force=True)

src/agent_workflow_server/main.py

@@ -11,6 +11,7 @@
 import uvicorn.logging
 from dotenv import load_dotenv
 from fastapi import Depends, FastAPI
+from fastapi.middleware.cors import CORSMiddleware
 
 import agent_workflow_server.logging.logger  # noqa: F401
 from agent_workflow_server.agents.load import load_agents
@@ -49,6 +50,14 @@
     dependencies=[Depends(authentication_with_api_key)],
 )
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=os.getenv("CORS_ALLOWED_ORIGINS", "*").split(","),
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 
 def signal_handler(sig, frame):
     logger.warning(f"Received {signal.Signals(sig).name}. Exiting...")
@@ -60,7 +69,7 @@ def start():
         signal.signal(signal.SIGINT, signal_handler)
         signal.signal(signal.SIGTERM, signal_handler)
         load_agents()
-        n_workers = int(os.environ.get("NUM_WORKERS", 5))
+        n_workers = int(os.getenv("NUM_WORKERS", 5))
 
         loop = asyncio.get_event_loop()
         loop.create_task(start_workers(n_workers))