Address PR comment

sdk/ai/azure-ai-inference/azure/ai/inference/prompts/__init__.py

@@ -7,15 +7,13 @@
 
 from ._renderers import MustacheRenderer
 from ._parsers import PromptyChatParser
-from ._utils import load
 from ._patch import patch_sdk as _patch_sdk, PromptTemplate
 
 # Register the Mustache renderer and parser
 InvokerFactory().register_renderer("mustache", MustacheRenderer)
 InvokerFactory().register_parser("prompty.chat", PromptyChatParser)
 
 __all__ = [
-    "load",
     "Prompty",
     "PromptTemplate",
 ]

sdk/ai/azure-ai-inference/azure/ai/inference/prompts/_patch.py

@@ -7,9 +7,9 @@
 Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize
 """
 
-import azure.ai.inference.prompts as prompts
+from typing import Any, Dict, List
 from ._core import Prompty
-from ._utils import prepare
+from ._utils import load, prepare
 from ._mustache import render
 
 
@@ -26,15 +26,16 @@ class PromptTemplate:
     :type model_name: str
     """
 
-    @staticmethod
-    def from_prompty(file_path: str):
+    @classmethod
+    def from_prompty(cls, file_path: str):
         if not file_path:
             raise ValueError("Please provide file_path")
-        prompty = prompts.load(file_path)
+        prompty = load(file_path)
         return PromptTemplate(prompty=prompty)        
 
-    @staticmethod
+    @classmethod
     def from_message(
+        cls,
         prompt_template: str,
         api: str = "chat",
         model_name: str | None = None
@@ -64,7 +65,7 @@ def __init__(
         else:
             raise ValueError("Please invalid arguments for PromptConfig")
 
-    def render(self, data: dict[str, any] | None = None, **kwargs):
+    def render(self, data: dict[str, Any] | None = None, **kwargs) -> List[Dict[str, Any]]:
         if data is None:
             data = kwargs
 
@@ -74,6 +75,8 @@ def render(self, data: dict[str, any] | None = None, **kwargs):
         elif "prompt_template" in self._parameters:
             system_prompt = render(self._parameters["prompt_template"], data)
             return [{"role": "system", "content": system_prompt}]
+        else:
+            raise ValueError("Please provide valid prompt template")
 
 
 def patch_sdk():

sdk/ai/azure-ai-inference/samples/sample_chat_completions_from_input_prompt_string.py

@@ -23,8 +23,6 @@
         `your-azure-region` is the Azure region where your model is deployed.
     2) AZURE_AI_CHAT_KEY - Your model key (a 32-character string). Keep it secret.
 """
-# mypy: disable-error-code="union-attr"
-# pyright: reportAttributeAccessIssue=false
 
 
 def sample_chat_completions_from_input_prompt_string():

sdk/ai/azure-ai-inference/samples/sample_chat_completions_from_input_prompty.py

@@ -23,8 +23,6 @@
         `your-azure-region` is the Azure region where your model is deployed.
     2) AZURE_AI_CHAT_KEY - Your model key (a 32-character string). Keep it secret.
 """
-# mypy: disable-error-code="union-attr"
-# pyright: reportAttributeAccessIssue=false
 
 
 def sample_chat_completions_from_input_prompty():