Merge pull request #269 from launchdarkly-labs/cko_2025

Cko 2025

Author: ahmedhqadri

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

@@ -539,6 +539,8 @@ def add_userid_to_flags(self):
         res = self.ldproject.add_maintainer_to_flag("limitedTimeOfferBanner")
         res = self.ldproject.add_maintainer_to_flag("earlyAccessFeatureToggle")
         res = self.ldproject.add_maintainer_to_flag("debuggingModeForDevelopers")
+        res = self.ldproject.add_maintainer_to_flag("release-new-search-engine")
+        res = self.ldproject.add_maintainer_to_flag("release-new-shorten-collections-page")
 
 # ############################################################################################################
 
@@ -1133,7 +1135,8 @@ def flag_database_migration(self):
             "This feature flag will trigger the database migration tool in LaunchAirways",
             purpose="migration",
             migration_stages=6,
-            tags=["release", "migration-assistant", "optional"]
+            tags=["release", "migration-assistant", "optional"],
+            temporary=True
         )
 
     def flag_database_guarded_release(self):
@@ -2263,7 +2266,6 @@ def setup_template_environment(self):
         self.ldproject.copy_flag_settings("financialDBMigration", "production", "template-env")
         self.ldproject.copy_flag_settings("investment-recent-trade-db", "production", "template-env")
         self.ldproject.copy_flag_settings("release-new-investment-stock-api", "production", "template-env")
-        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")        
         print("Done")

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

@@ -726,7 +726,7 @@ def create_release_pipeline(self, pipeline_key, pipeline_name):
                                 "releaseStrategy": "monitored-release",
                                 "requireApproval": False,
                                 "releaseGuardianConfiguration": {
-                                    "monitoringWindowMilliseconds": 3000,
+                                    "monitoringWindowMilliseconds": 1000,
                                     "rolloutWeight": 50000,
                                     "rollbackOnRegression": True,
                                 },
@@ -1578,6 +1578,6 @@ def advance_flag_phase(self, flag_key, status, pipeline_phase_id):
             if status_code != 200:
                 data = json.loads(response.text)
                 print("Error advancing flag phase: " + data["message"])
-                time.sleep(2)
+                time.sleep(3)
 
         return response

Diff for: components/chatbot/ChatBot.tsx

@@ -17,7 +17,7 @@ import { BatteryCharging } from "lucide-react";
 import {
   PERSONA_ROLE_DEVELOPER,
   COHERE,
-  CLAUDE,
+  ANTHROPIC,
   META,
   DEFAULT_AI_MODEL,
 } from "@/utils/constants";
@@ -237,7 +237,7 @@ export default function Chatbot({ vertical }: { vertical: string }) {
                           : ""
                       } 
                       ${
-                        aiNewModelChatbotFlag?.model?.name?.includes(CLAUDE)
+                        aiNewModelChatbotFlag?.model?.name?.includes(ANTHROPIC)
                           ? "!text-anthropicColor"
                           : ""
                       }

Diff for: pages/api/chat.ts

@@ -9,7 +9,6 @@ import { getCookie } from "cookies-next";
 import { initAi } from "@launchdarkly/server-sdk-ai";
 import { LD_CONTEXT_COOKIE_KEY } from "@/utils/constants";
 import { v4 as uuidv4 } from "uuid";
-import { DEFAULT_AI_MODEL } from "@/utils/constants";
 
 export default async function chatResponse(
   req: NextApiRequest,
@@ -53,44 +52,14 @@ export default async function chatResponse(
     const ldClient = await getServerClient(process.env.LD_SDK_KEY || "");
     const aiClient = initAi(ldClient);
     const context: any = clientSideAudienceContext || {
-      kind: "audience",
-      key: uuidv4().slice(0, 6),
+      kind: "user",
+      key: uuidv4(),
     };
 
-    const aiConfig = await aiClient.config(aiConfigKey!, context, {}, {});
+    const aiConfig = await aiClient.config(aiConfigKey, context, {}, {});
     if (!aiConfig.enabled) {
       throw new Error("AI config is disabled");
     } else {
-      const parameters = aiConfig.model?.parameters || {};
-      const objWithoutModelId2 = Object.keys(parameters).reduce(
-        (newObj: any, key) => {
-          if (
-            key === "maxTokens" &&
-            aiConfig.model &&
-            aiConfig.model?.parameters &&
-            aiConfig?.model.name.includes("cohere")
-          ) {
-            newObj["max_tokens"] = aiConfig.model.parameters[key];
-          }
-          if (
-            key === "maxTokens" &&
-            aiConfig.model &&
-            aiConfig.model?.parameters && 
-            aiConfig?.model.name.includes("anthropic")
-          ) {
-            newObj["max_tokens_to_sample"] =
-              aiConfig.model.parameters[key];
-          }
-          if (key !== "maxTokens" &&
-            aiConfig.model
-          ) {
-            newObj[key] = (aiConfig.model as any)[key];
-          }
-          return newObj;
-        },
-        {}
-      );
-
       if (!aiConfig.model) {
         throw new Error("AI model configuration is undefined");
       }