FIxed get owner function to return correct owner email id

backend/api/routes.py

@@ -45,7 +45,10 @@ def analyze(request: AnalyzeRequest) -> dict:
 
 @router.post("/custom-task")
 def custom_task(request: CustomTaskRequest) -> dict:
-    owner = db.get_owner(request.owner_id)
+    try:
+        owner = db.get_owner(request.owner_id)
+    except KeyError as exc:
+        raise HTTPException(status_code=404, detail=str(exc)) from exc
     try:
         return analyze_custom_task(
             business_description=owner.get("business_description", ""),
@@ -59,7 +62,10 @@ def custom_task(request: CustomTaskRequest) -> dict:
 
 @router.post("/suggest-workflows")
 def suggest_workflows(request: WorkflowSuggestionRequest) -> dict:
-    owner = db.get_owner(request.owner_id)
+    try:
+        owner = db.get_owner(request.owner_id)
+    except KeyError as exc:
+        raise HTTPException(status_code=404, detail=str(exc)) from exc
     try:
         return suggest_workflow_options(
             task_name=request.task_name,
@@ -75,7 +81,10 @@ def suggest_workflows(request: WorkflowSuggestionRequest) -> dict:
 
 @router.post("/build-workflow", response_model=BuildWorkflowResponse)
 def build_workflow(request: BuildWorkflowRequest) -> BuildWorkflowResponse:
-    owner = db.get_owner(request.owner_id)
+    try:
+        owner = db.get_owner(request.owner_id)
+    except KeyError as exc:
+        raise HTTPException(status_code=404, detail=str(exc)) from exc
     try:
         workflow_json = build_workflow_definition(request=request, owner=owner)
         compiled = compile_workflow(workflow_json)
@@ -87,9 +96,10 @@ def build_workflow(request: BuildWorkflowRequest) -> BuildWorkflowResponse:
 
 @router.post("/deploy", response_model=DeploymentResponse)
 def deploy_workflow(request: DeployRequest) -> DeploymentResponse:
-    owner = db.get_owner(request.owner_id)
-    if not owner:
-        raise HTTPException(status_code=404, detail="owner not found")
+    try:
+        owner = db.get_owner(request.owner_id)
+    except KeyError as exc:
+        raise HTTPException(status_code=404, detail=str(exc)) from exc
     deployments = []
     for workflow in request.workflows:
         compiled = compile_workflow(workflow.model_dump())
@@ -119,7 +129,10 @@ def reset_data() -> dict[str, str]:
 
 @router.get("/status", response_model=OwnerStatusResponse)
 def status(owner_id: str) -> OwnerStatusResponse:
-    owner = db.get_owner(owner_id)
+    try:
+        owner = db.get_owner(owner_id)
+    except KeyError as exc:
+        raise HTTPException(status_code=404, detail=str(exc)) from exc
     return OwnerStatusResponse(
         owner=owner,
         workflows=db.list_workflows(owner_id),

backend/storage/database.py

@@ -21,6 +21,7 @@ class InMemoryDatabase:
     def ensure_owner(self, owner_id: str, email: str) -> dict:
         owner = self.owners.get(owner_id)
         if owner:
+            owner["email"] = email
             return owner
         owner = {
             "id": owner_id,
@@ -49,7 +50,7 @@ class InMemoryDatabase:
 
     def get_owner(self, owner_id: str) -> dict:
         if owner_id not in self.owners:
-            self.ensure_owner(owner_id, f"{owner_id}@example.com")
+            raise KeyError(f"owner {owner_id} not found")
         return deepcopy(self.owners[owner_id])
 
     def save_workflow(self, owner_id: str, workflow: dict) -> dict:

tests/test_flowpilot_api.py

@@ -15,7 +15,34 @@ def test_in_memory_database_creates_timezone_aware_owner_timestamp():
     assert owner["created_at"].endswith("+00:00")
 
 
+def test_status_returns_404_for_unknown_owner_without_creating_placeholder():
+    db.reset()
+
+    status_response = client.get("/api/status", params={"owner_id": "missing-owner"})
+
+    assert status_response.status_code == 404
+    assert "missing-owner" not in db.owners
+
+
+def test_analyze_updates_existing_owner_email():
+    db.reset()
+    db.ensure_owner("sushanth", "sushanth@example.com")
+
+    analyze_response = client.post(
+        "/api/analyze",
+        json={
+            "owner_id": "sushanth",
+            "owner_email": "real@example.com",
+            "description": "Weekly pricing and orders.",
+        },
+    )
+
+    assert analyze_response.status_code == 200
+    assert db.owners["sushanth"]["email"] == "real@example.com"
+
+
 def test_owner_flow_builds_and_deploys_workflows():
+    db.reset()
     analyze_response = client.post(
         "/api/analyze",
         json={
@@ -66,6 +93,7 @@ def test_owner_flow_builds_and_deploys_workflows():
 
 
 def test_upload_and_status_endpoints():
+    db.reset()
     client.post(
         "/api/analyze",
         json={"owner_id": "raj2", "owner_email": "raj2@orchard.com", "description": "Weekly pricing and orders."},
@@ -87,6 +115,7 @@ def test_upload_and_status_endpoints():
 
 
 def test_reset_endpoint_clears_all_in_memory_data():
+    db.reset()
     analyze_response = client.post(
         "/api/analyze",
         json={