update assistant decortators for table storage optional params

Azure · Sep 23, 2024 · ae34695 · ae34695
1 parent 2e5ea4f
commit ae34695
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/azure/functions/decorators/openai.py b/azure/functions/decorators/openai.py
@@ -77,10 +77,14 @@ def __init__(self,
                  name: str,
                  id: str,
                  timestamp_utc: str,
+                 chat_storage_connection_setting: Optional[str] = "AzureWebJobsStorage",
+                 collection_name: Optional[str] = "SampleChatState",
                  data_type: Optional[DataType] = None,
                  **kwargs):
         self.id = id
         self.timestamp_utc = timestamp_utc
+        self.chat_storage_connection_setting = chat_storage_connection_setting
+        self.collection_name = collection_name
         super().__init__(name=name, data_type=data_type)
 
 
@@ -165,12 +169,16 @@ def __init__(self, name: str,
                  id: str,
                  user_message: str,
                  model: Optional[str] = None,
+                 chat_storage_connection_setting: Optional[str] = "AzureWebJobsStorage",
+                 collection_name: Optional[str] = "SampleChatState",
                  data_type: Optional[DataType] = None,
                  **kwargs):
         self.name = name
         self.id = id
         self.user_message = user_message
         self.model = model
+        self.chat_storage_connection_setting = chat_storage_connection_setting
+        self.collection_name = collection_name
         super().__init__(name=name, data_type=data_type)