added http errors

api_contract.py

@@ -0,0 +1,54 @@
+from error_handling import ErrorCodes
+
+def get_error_codes_description():
+    description = "| code | description |\n| - | - |\n"
+    error_codes_description = []
+    for error_code in ErrorCodes:
+        error_codes_description.append(
+            f"| {error_code.value[0]} | {error_code.value[1]} |\n"
+        )
+    list.sort(error_codes_description)
+    description = f"{description}{''.join(error_codes_description)}"
+    return description
+
+
+components_dict = {
+    "schemas": {
+        "ErrorMessage": {
+            "required": ["errors"],
+            "description": get_error_codes_description(),
+            "properties": {
+                "errors": {
+                    "type": "array",
+                    "title": "Error array",
+                    "items": {
+                        "type": "object",
+                        "required": ["code", "message"],
+                        "properties": {
+                            "code": {"type": "integer", "example": 400},
+                            "message": {
+                                "type": "string",
+                                "example": "'input_text' is a required property",
+                                "maxLength": 256,
+                            },
+                        },
+                    },
+                }
+            },
+        }
+    },
+    "paths": {
+        "/api/generatel-language": {
+            "parameters": [
+                {
+                    "name": "petId",
+                    "in": "path",
+                    "schema": {
+                        "type": "integer",
+                        "format": "int64"
+                    }
+                }
+            ]
+        }
+    }
+}

error_handling.py

@@ -0,0 +1,8 @@
+from enum import Flag
+
+class ErrorCodes(Flag):
+    INTERNAL_SERVER_ERROR = (500, "internal server error")
+    REQUEST_VALIDATION_ERROR = (400, "request validation error")
+    JWT_VALIDATION_ERROR = (401, "jwt validation error")
+    TEXT_VALIDATION_ERROR = (490, "text validation error")
+    OTHER_ERROR = (999, "other error")

main.py

@@ -1,22 +1,37 @@
 from transformers import pipeline
-from fastapi import FastAPI
+from fastapi import FastAPI, Request, HTTPException
 from payload import SomeText
-
-
+from api_contract import components_dict as api_components
+from error_handling import ErrorCodes
 
 app = FastAPI()
 
+app=FastAPI(title="Huggingface Gen LLM gest",
+            version="1.0",
+            debug=True,
+            components=api_components,
+            swagger_ui_bundle_js= "//unpkg.com/swagger-ui-dist@3/swagger-ui-bundle.js",
+            swagger_ui_standalone_preset_js= "//unpkg.com/swagger-ui-dist@3/swagger-ui-standalone-preset.js",
+            summary="API to perform generative prompt completion using small LLM (without GPU).",
+)
+
 pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
 
 @app.get("/")
 def read_root():
-    return {'detail': 'API seems to be working'}
+    return {'detail': 'API running. Try out the endpoints in swagger'}
 
 @app.get("/api/check-heartbeat")
 def get_heartbeat():
-    return {"Status": "Running. Try out the endpoints in swagger"}
+    return {"detail": "seems to be working"}
+
+@app.post("/api/generatel-language", summary="Generate text from prompt", tags=["Generate"])
+def inference(request: Request, input_prompt: SomeText):
+
+    if len(input_prompt.text) >0:
+
+        output = pipe_flan(input_prompt.text)
+        return {"output": output[0]["generated_text"]}
 
-@app.post("/api/generate", summary="Generate text from prompt", tags=["Generate"], response_model=SomeText)
-def inference(input_prompt: SomeText):
-    output = pipe_flan(input_prompt)
-    return {"output": output[0]["generated_text"]}
+    else:
+        raise HTTPException(status_code=400, detail = ErrorCodes.REQUEST_VALIDATION_ERROR.value[1])

payload.py

@@ -12,6 +12,4 @@ class SomeText(BaseModel):
                 }
             ]
         }
-    }
-
-
+    }