Merge pull request #12 from ESGF/replication-flow-type-correction

Speculative correction for static type issues

README.md

@@ -251,7 +251,7 @@ This UI provides a complete view of the stage of your Kafka service and is descr
 
 ### STAC Browser
 
-A STAC browser simulating the **East** ESGF index is available at http://localhost:9011. This service should be listening 
+A STAC browser simulating the **East** ESGF index is available at http://localhost:9011. This service should be listening  
 to all the publication, retraction and update events in the Kafka queue.
 
 A STAC browser simulating the **West** ESGF index is available at http://localhost:9015. This service should be listening 

esgf-transaction-api/esgf_transaction_api/main.py

@@ -47,7 +47,9 @@ async def lifespan(app: FastAPI) -> AsyncGenerator[Any, Any]:
 app = FastAPI(lifespan=lifespan)
 
 
-def item_body(payload) -> KafkaEvent:
+def item_body(
+    payload: Union[RevokePayload, UpdatePayload, CreatePayload]
+) -> KafkaEvent:
     data = Data(type="STAC", version="1.0.0", payload=payload)
     auth = Auth(client_id="esgf-generator", server="docker-compose-local")
     publisher = Publisher(package="esgf-generator", version="0.1.0")
@@ -158,9 +160,7 @@ async def create_item(
 
 
 @app.put("/{collection_id}/items/{item_id}")
-async def update_item(
-    collection_id: str, item_id: str, item: Union[Item, ItemCollection]
-) -> Union[Item, ItemCollection]:
+async def update_item(collection_id: str, item_id: str, item: Item) -> Item:
     """Add UPDATE message to kafka event stream.
 
     Args: