Merge pull request #261 from launchdarkly-labs/ai_config_patch

Ai config patch

Author: ahmedhqadri

Diff for: .github/workflows/demo_provisioning_scripts/DemoBuilder.py

@@ -328,6 +328,8 @@ def add_targeting_rules(self):
         res = self.ldproject.add_segment_to_flag("federatedAccounts", "development-team", "production")
         res = self.ldproject.add_segment_to_flag("wealthManagement", "beta-users", "production")
         res = self.ldproject.add_segment_to_flag("cartSuggestedItems", "beta-users", "production")
+        res = self.ldproject.add_segment_to_flag("ai-config--destination-picker-new-ai-model", "launch-airways-all-members", "production")
+        
 
     def toggle_flags(self):
         res = self.ldproject.toggle_flag(
@@ -517,7 +519,7 @@ def create_destination_recommendation_ai_config(self):
             "claude-haiku",
             "Claude Haiku",
             {
-                "id": "anthropic.claude-instant-v1",
+                "modelName": "anthropic.claude-instant-v1",
                 "parameters": {
                     "temperature": 0.5,
                     "maxTokens": 150
@@ -535,7 +537,7 @@ def create_destination_recommendation_ai_config(self):
             "cohere-text",
             "Cohere Text",
             {
-                "id": "cohere.command-text-v14",
+                "modelName": "cohere.command-text-v14",
                 "parameters": {
                     "temperature": 0.7,
                     "maxTokens": 250
@@ -561,7 +563,7 @@ def create_travel_insights_ai_config(self):
             "general-travel",
             "General Travel",
             {
-                "id": "anthropic.claude-instant-v1",
+                "modelName": "anthropic.claude-instant-v1",
                 "parameters": {
                     "maxTokens": 200,
                     "temperature": 0.9
@@ -579,7 +581,7 @@ def create_travel_insights_ai_config(self):
             "historical-focus",
             "Historical Focus",
             {
-                "id": "anthropic.claude-instant-v1",
+                "modelName": "anthropic.claude-instant-v1",
                 "parameters": {
                     "maxTokens": 200,
                     "temperature": 0.9
@@ -597,7 +599,7 @@ def create_travel_insights_ai_config(self):
             "weather-focus",
             "Weather Focus",
             {
-                "id": "anthropic.claude-instant-v1",
+                "modelName": "anthropic.claude-instant-v1",
                 "parameters": {
                     "maxTokens": 200,
                     "temperature": 0.9
@@ -623,7 +625,7 @@ def create_ai_chatbot_ai_config(self):
             "claude-haiku",
             "Claude Haiku",
             {
-                "id": "anthropic.claude-instant-v1",
+                "modelName": "anthropic.claude-instant-v1",
                 "parameters": {
                     "maxTokens": 200,
                     "temperature": 0.5
@@ -641,7 +643,7 @@ def create_ai_chatbot_ai_config(self):
             "cohere-coral",
             "Cohere Coral",
             {
-                "id": "cohere.command-text-v14",
+                "modelName": "cohere.command-text-v14",
                 "parameters": {
                     "maxTokens": 200,
                     "temperature": 0.5
@@ -1178,6 +1180,7 @@ def setup_template_environment(self):
         self.ldproject.copy_flag_settings("ai-chatbot", "production", "template-env")
         self.ldproject.copy_flag_settings("storeAttentionCallout", "production", "template-env")
         self.ldproject.copy_flag_settings("cartSuggestedItems", "production", "template-env")
+        self.ldproject.copy_flag_settings("ai-config--destination-picker-new-ai-model", "production", "template-env")
 
         print("Done")
 

Diff for: .github/workflows/demo_provisioning_scripts/LDPlatform.py

@@ -289,7 +289,7 @@ def create_ai_config_versions(self, ai_config_key, ai_config_version_key, ai_con
 
         response = self.getrequest(
             "POST",
-            "https://app.launchdarkly.com/api/v2/projects/" + self.project_key + "/ai-configs/" + ai_config_key + "/versions",
+            "https://app.launchdarkly.com/api/v2/projects/" + self.project_key + "/ai-configs/" + ai_config_key + "/variations",
             json=payload,
             headers=headers,
         )