Merge pull request #31 from wri/fix-api-deploy

Fix api deploy

.gitignore

@@ -171,4 +171,7 @@ data/
 .DS_Store
 
 # VS Code
-.vscode
+.vscode
+
+# Custom zeno
+ee-zeno-service-account.json

Dockerfile

@@ -26,4 +26,4 @@ WORKDIR /app
 RUN uv sync --frozen
 
 # Command to run the application.
-CMD ["uv", "run", "uvicorn", "api:app", "--reload"]
+CMD ["uv", "run", "uvicorn", "api:app", "--reload", "--host", "0.0.0.0"]

api.py

@@ -2,16 +2,21 @@
 from typing import Annotated
 
 from fastapi import Body, FastAPI
+from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
 
 from zeno.agents.maingraph.utils.state import GraphState
 from zeno.agents.maingraph.agent import graph
 
 app = FastAPI()
 
-# https://www.softgrade.org/sse-with-fastapi-react-langgraph/
-# https://www.workfall.com/learning/blog/how-to-stream-json-data-using-server-sent-events-and-fastapi-in-python-over-http/
-
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
 def pack(data):
     return json.dumps(data) + "\n"
@@ -36,8 +41,6 @@ def event_stream(query: str):
                             yield pack({"tool_calls": msg.tool_calls})
                         if hasattr(msg, "artifact"):
                             yield pack({"artifact": msg.artifact})
-                else:
-                    yield pack({key2: val2})
 
 
 @app.post("/stream")

docker-compose.yaml

@@ -1,7 +1,7 @@
 services:
   api:
     build:
-      context: ./zeno
+      context: .
     ports:
       - "8000:8000"
     environment:
@@ -15,8 +15,7 @@ services:
       - .env
 
     volumes:
-      - ./zeno:/app
-      - ./data:/app/data
+      - .:/app
 
   frontend: 
     build: