wip

README.md

@@ -132,6 +132,12 @@ INFO: Application startup complete.
 INFO: Uvicorn running on http://0.0.0.0:4501 (Press CTRL+C to quit)
 ```
 
+> [!TIP]
+> By default, CORS is enabled. If you want to disable it, you can set the `DISABLE_CORS` environment variable:
+> ```
+> $ DISABLE_CORS=true python -m llama_deploy.apiserver
+> ```
+
 From another shell, use `llamactl` to create the deployment:
 
 ```

llama_deploy/apiserver/app.py

@@ -1,7 +1,9 @@
 import logging
+import os
 
 from fastapi import FastAPI
 from fastapi.responses import JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
 
 from .routers import status_router, deployments_router
 from .server import lifespan
@@ -10,6 +12,17 @@
 
 
 app = FastAPI(root_path="/", lifespan=lifespan)
+
+# Configure CORS middleware if the environment variable is set
+if not os.environ.get("DISABLE_CORS", False):
+ app.add_middleware(
+ CORSMiddleware,
+ allow_origins=["*"], # Allows all origins
+ allow_credentials=True,
+ allow_methods=["GET", "POST"],
+ allow_headers=["Content-Type", "Authorization"],
+ )
+
 app.include_router(deployments_router)
 app.include_router(status_router)
 

llama_deploy/cli/run.py

@@ -27,7 +27,7 @@ def run(
  service: str,
 ) -> None:
  server_url, disable_ssl, timeout = global_config
- deploy_url = f"{server_url}/deployments/{deployment}/tasks/create"
+ deploy_url = f"{server_url}/deployments/{deployment}/tasks/run"
  payload = {"input": json.dumps(dict(arg))}
  if service:
  payload["agent_id"] = service

pyproject.toml

@@ -13,7 +13,7 @@ omit = [
 
 [tool.poetry]
 name = "llama-deploy"
-version = "0.2.1"
+version = "0.2.3"
 description = ""
 authors = ["Logan Markewich <[email protected]>", "Andrei Fajardo <[email protected]>"]
 maintainers = [

tests/cli/test_run.py

@@ -13,7 +13,7 @@ def test_run(runner: CliRunner) -> None:
  llamactl, ["run", "-d", "deployment_name", "-s", "service_name"]
  )
  mocked_post.assert_called_with(
- "http://localhost:4501/deployments/deployment_name/tasks/create",
+ "http://localhost:4501/deployments/deployment_name/tasks/run",
  verify=True,
  json={"input": "{}", "agent_id": "service_name"},
  timeout=None,
@@ -51,7 +51,7 @@ def test_run_args(runner: CliRunner) -> None:
  ],
  )
  mocked_post.assert_called_with(
- "http://localhost:4501/deployments/deployment_name/tasks/create",
+ "http://localhost:4501/deployments/deployment_name/tasks/run",
  verify=True,
  json={
  "input": '{"first_arg": "first_value", "second_arg": "\\"second value with spaces\\""}',