Adding mandatory model name to llm gateway call

Author: ritaagarwala-sf

src/datacustomcode/llm_gateway/types/generate_text_request.py

@@ -39,7 +39,7 @@ class GenerateTextRequest(BaseModel):
         default="v1", description="API version, must be 'v1'"
     )
     model_name: str = Field(..., min_length=1, description="Name of the model to use")
-    prompt: str = Field(..., min_length=1, max_length=1000, description="Input prompt")
+    prompt: str = Field(..., description="Input prompt")
     localization: Optional[Dict[str, Any]] = Field(
         default=None, description="Localization settings"
     )

src/datacustomcode/templates/function/payload/entrypoint.py

@@ -47,7 +47,7 @@ def function(request: dict, runTime: Runtime) -> dict:
     current_seq_no = 1  # Start sequence number from 1
 
     builder = GenerateTextRequestBuilder()
-    llm_request = builder.set_prompt("Hello").set_model("").build()
+    llm_request = builder.set_prompt("Hello").set_model("modelName").build()
     llm_response = runTime.llm_gateway.generate_text(llm_request)
 
     if llm_response.is_success: