update default collection name to ChatState

Author: manvkaur

azure/functions/decorators/function_app.py

@@ -3267,7 +3267,7 @@ def assistant_query_input(self,
                               id: str,
                               timestamp_utc: str,
                               chat_storage_connection_setting: Optional[str] = "AzureWebJobsStorage",       # noqa: E501
-                              collection_name: Optional[str] = "SampleChatState",       # noqa: E501
+                              collection_name: Optional[str] = "ChatState",       # noqa: E501
                               data_type: Optional[
                                   Union[DataType, str]] = None,
                               **kwargs) \
@@ -3284,7 +3284,7 @@ def assistant_query_input(self,
         for the table settings for assistant chat storage. The default value is
         "AzureWebJobsStorage".
         :param collection_name:  The table collection name for assistant chat
-        storage. The default value is "SampleChatState".
+        storage. The default value is "ChatState".
         :param id: The ID of the Assistant to query.
         :param data_type: Defines how Functions runtime should treat the
         parameter value
@@ -3318,7 +3318,7 @@ def assistant_post_input(self, arg_name: str,
                              user_message: str,
                              model: Optional[str] = None,
                              chat_storage_connection_setting: Optional[str] = "AzureWebJobsStorage",       # noqa: E501
-                             collection_name: Optional[str] = "SampleChatState",       # noqa: E501
+                             collection_name: Optional[str] = "ChatState",       # noqa: E501
                              data_type: Optional[
                                  Union[DataType, str]] = None,
                              **kwargs) \
@@ -3336,7 +3336,7 @@ def assistant_post_input(self, arg_name: str,
         for the table settings for assistant chat storage. The default value is
         "AzureWebJobsStorage".
         :param collection_name:  The table collection name for assistant chat
-        storage. The default value is "SampleChatState".
+        storage. The default value is "ChatState".
         :param data_type: Defines how Functions runtime should treat the
         parameter value
         :param kwargs: Keyword arguments for specifying additional binding

azure/functions/decorators/openai.py

@@ -78,7 +78,7 @@ def __init__(self,
                  id: str,
                  timestamp_utc: str,
                  chat_storage_connection_setting: Optional[str] = "AzureWebJobsStorage",        # noqa: E501
-                 collection_name: Optional[str] = "SampleChatState",
+                 collection_name: Optional[str] = "ChatState",
                  data_type: Optional[DataType] = None,
                  **kwargs):
         self.id = id
@@ -170,7 +170,7 @@ def __init__(self, name: str,
                  user_message: str,
                  model: Optional[str] = None,
                  chat_storage_connection_setting: Optional[str] = "AzureWebJobsStorage",       # noqa: E501
-                 collection_name: Optional[str] = "SampleChatState",
+                 collection_name: Optional[str] = "ChatState",
                  data_type: Optional[DataType] = None,
                  **kwargs):
         self.name = name

tests/decorators/test_openai.py

@@ -58,7 +58,7 @@ def test_assistant_query_input_valid_creation(self):
         input = AssistantQueryInput(name="test",
                                     timestamp_utc="timestamp_utc",
                                     chat_storage_connection_setting="AzureWebJobsStorage",       # noqa: E501
-                                    collection_name="SampleChatState",
+                                    collection_name="ChatState",
                                     data_type=DataType.UNDEFINED,
                                     id="test_id",
                                     type="assistantQueryInput",
@@ -69,7 +69,7 @@ def test_assistant_query_input_valid_creation(self):
                          {"name": "test",
                           "timestampUtc": "timestamp_utc",
                           "chatStorageConnectionSetting": "AzureWebJobsStorage",       # noqa: E501
-                          "collectionName": "SampleChatState",
+                          "collectionName": "ChatState",
                           "dataType": DataType.UNDEFINED,
                           "direction": BindingDirection.IN,
                           "type": "assistantQuery",
@@ -116,7 +116,7 @@ def test_assistant_post_input_valid_creation(self):
                                    id="test_id",
                                    model="test_model",
                                    chat_storage_connection_setting="AzureWebJobsStorage",       # noqa: E501
-                                   collection_name="SampleChatState",
+                                   collection_name="ChatState",
                                    user_message="test_message",
                                    data_type=DataType.UNDEFINED,
                                    dummy_field="dummy")
@@ -127,7 +127,7 @@ def test_assistant_post_input_valid_creation(self):
                           "id": "test_id",
                           "model": "test_model",
                           "chatStorageConnectionSetting": "AzureWebJobsStorage",       # noqa: E501
-                          "collectionName": "SampleChatState",
+                          "collectionName": "ChatState",
                           "userMessage": "test_message",
                           "dataType": DataType.UNDEFINED,
                           "direction": BindingDirection.IN,