Update main.py

main.py

@@ -7,6 +7,7 @@ app = FastAPI()
 
 class EatcHeadersResponse(BaseModel):
     eatc_code: str
+    eatc_match_asignation_rule: str
 
 @app.get("/eatc-headers", response_model=EatcHeadersResponse)
 async def get_eatc_headers(eatc_code: str = Query(..., description="EATC code to retrieve headers for")):
@@ -15,9 +16,11 @@ async def get_eatc_headers(eatc_code: str = Query(..., description="EATC code to
     if response.status_code == 200:
         response_json = json.loads(response.text)
         if len(response_json) >= 2 and "res" in response_json and response_json["res"]:
-            return {"eatc_code": response_json["res"][0]["eatc-code"]}
+            return {
+                "eatc_code": response_json["res"][0]["eatc-code"],
+                "eatc_match_asignation_rule": response_json["res"][0].get("eatc_match_asignation_rule", "")
+            }
         else:
             return {"error": "Invalid response from API"}
     else:
-        return {"error": "API returned error status code"}
-
+        return {"error": "API returned error status code"}