Fix unit test

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

@@ -43,8 +43,10 @@ def sample_chat_completions_from_input_prompt_string():
 
 
     prompt_template = """
+system:
 You are an AI assistant in a hotel. You help guests with their requests and provide information about the hotel and its services.
 
+user:
 {input}
 """
     prompt_config = PromptyTemplate.from_message(

sdk/ai/azure-ai-inference/tests/sample1.prompty

@@ -0,0 +1,30 @@
+---
+name: Basic Prompt
+description: A basic prompt that uses the GPT-3 chat API to answer questions
+authors:
+  - author_1
+  - author_2
+model:
+  api: chat
+  configuration:
+    azure_deployment: gpt-4o-mini
+  parameters:
+    temperature: 1
+    frequency_penalty: 0.5
+    presence_penalty: 0.5
+sample:
+  firstName: Jane
+  lastName: Doe
+  question: What is the meaning of life?
+  chat_history: []
+---
+system:
+You are an AI assistant in a hotel. You help guests with their requests and provide information about the hotel and its services.
+
+{{#chat_history}}
+{{role}}:
+{{content}}
+{{/chat_history}}
+
+user:
+{{input}}

sdk/ai/azure-ai-inference/tests/test_prompts.py

@@ -3,62 +3,48 @@
 # Licensed under the MIT License.
 # ------------------------------------
 import os
-import json
-import azure.ai.inference as sdk
-import azure.ai.inference.prompts as prompts
+from azure.ai.inference.prompts import PromptyTemplate
+from devtools_testutils import AzureRecordedTestCase
 
-from model_inference_test_base import (
-    ModelClientTestBase,
-    ServicePreparerChatCompletions,
-    ServicePreparerAOAIChatCompletions,
-    ServicePreparerEmbeddings,
-)
-from azure.core.pipeline.transport import RequestsTransport
-from devtools_testutils import recorded_by_proxy
-from azure.core.exceptions import AzureError, ServiceRequestError
-from azure.core.credentials import AzureKeyCredential
 
-
-
-class TestModelClient(ModelClientTestBase):
+class TestPrompts(AzureRecordedTestCase):
 
     # **********************************************************************************
     #
     #                               UNIT TESTS
     #
     # **********************************************************************************
 
-    def test_prompty(self, **kwargs):
-        path = "/Users/weiwu/Workspace/1_Testing/TestAI/test-prompty/test.prompty"
-        p = prompts.load(path)
-
-        inputs = {
-            "input": "my first question",
+    def test_prompt_config_from_prompty(self, **kwargs):
+        script_dir = os.path.dirname(os.path.abspath(__file__))
+        prompty_file_path = os.path.join(script_dir, "sample1.prompty")
+        prompt_config = PromptyTemplate.load(prompty_file_path)
+        assert prompt_config.model_name == "gpt-4o-mini"
+        assert prompt_config.config["temperature"] == 1
+        assert prompt_config.config["frequency_penalty"] == 0.5
+        assert prompt_config.config["presence_penalty"] == 0.5
+        input_variables = {
+            "input": "please tell me a joke about cats",
         }
-
-        print(p)
-
-        parsed = prompts.prepare(p, inputs)
-
-        lc_messages = [] # TODO: will be removed
-        for message in parsed:
-            message_class = prompts.RoleMap.get_message_class(message["role"])
-            lc_messages.append(message_class(content=message["content"]))
-
-        print(lc_messages)
-
-        assert True
-
-
-    def test_prompt_config(self, **kwargs):
-        path = "/Users/weiwu/Workspace/1_Testing/TestAI/test-prompty/test.prompty"
-        prompt_config = prompts.get_prompt_config(file_path=path)
-
-        inputs = {
-            "input": "my first question",
+        messages = prompt_config.render(input_variables=input_variables)
+        assert len(messages) == 2
+        assert messages[0]["role"] == "system"
+        assert messages[1]["role"] == "user"
+        assert messages[1]["content"] == "please tell me a joke about cats"
+
+    def test_prompt_config_from_message(self, **kwargs):
+        prompt_config = PromptyTemplate.from_message(
+            api = "chat",
+            model_name = "gpt-4o-mini",
+            prompt_template = "system prompt template {input}"
+        )
+        assert prompt_config.model_name == "gpt-4o-mini"
+        input_variables = {
+            "input": "please tell me a joke about cats",
         }
-
-        messages = prompt_config.render(inputs)
-        print(messages)
-
-        assert True
+        messages = prompt_config.render(input_variables=input_variables)
+        assert len(messages) == 1
+        assert messages[0]["role"] == "system"
+        # TODO: need to separate the system prompt from the user input
+        # assert messages[1]["role"] == "user"
+        # assert messages[1]["content"] == "please tell me a joke about cats"