Merge pull request #2080 from solliancenet/cp-tool-prompt-template

Adds template replacement of tool prompts
solliancenet · Dec 15, 2024 · 1c03b02 · 1c03b02
2 parents 3504a54 + 4663bc1
commit 1c03b02
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/dotnet/Orchestration/Orchestration/OrchestrationBuilder.cs b/src/dotnet/Orchestration/Orchestration/OrchestrationBuilder.cs
@@ -442,6 +442,16 @@ await cosmosDBService.PatchOperationsItemPropertiesAsync<LongRunningOperationCon
                             var prompt = await promptResourceProvider.GetResourceAsync<PromptBase>(
                                 resourceObjectId.ObjectId,
                                 currentUserIdentity);
+                            if (prompt is MultipartPrompt multipartPrompt)
+                            {
+                                // prompt template token replacement
+                                if (multipartPrompt is not null)
+                                {
+
+                                    multipartPrompt.Prefix = templatingService.Transform(multipartPrompt.Prefix!);
+                                    multipartPrompt.Suffix = templatingService.Transform(multipartPrompt.Suffix!);
+                                }
+                            }
                             explodedObjectsManager.TryAdd(
                                 resourceObjectId.ObjectId,
                                 prompt);