Update prompt index with new prompt data name (#714)

Update prompt index with new prompt data name Fixing issue from #710 (comment)
lastmile-ai · Jan 2, 2024 · 811046e · 811046e
2 parents dbd96d4 + b3b2e76
commit 811046e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/src/aiconfig/schema.py b/python/src/aiconfig/schema.py
@@ -586,6 +586,8 @@ def update_prompt(self, prompt_name: str, prompt_data: Prompt):
         for i, prompt in enumerate(self.prompts):
             if prompt.name == prompt_name:
                 self.prompts[i] = prompt_data
+                del self.prompt_index[prompt_name]
+                self.prompt_index[prompt_data.name] = prompt_data
                 break
 
     def delete_prompt(self, prompt_name: str):