Add caching to agent options in mixtureOfAgents function

microsoft · Sep 24, 2024 · e0d536d · e0d536d
1 parent bf36fd9
commit e0d536d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/sample/genaisrc/mixture-of-agents.genai.mts b/packages/sample/genaisrc/mixture-of-agents.genai.mts
@@ -21,7 +21,7 @@ export async function mixtureOfAgents(
                     ctx.writeText(query)
                     ctx.assistant(`What do you think?`)
                 },
-                { ...agent, label: agent.label || agent.model }
+                { ...agent, cache: "moa", label: agent.label || agent.model }
             )
         )
     )
@@ -39,7 +39,7 @@ export async function mixtureOfAgents(
             }
             _.$`Please provide your critique for each candidate:`
         },
-        { ...(options ?? {}), label: "critique" }
+        { ...(options ?? {}), cache: "moa", label: "critique" }
     )
 
     // final prompt
@@ -55,7 +55,7 @@ export async function mixtureOfAgents(
             _.def(`Critique`, critique)
             _.$`Please provide a final, optimized response to the original query:`
         },
-        { ...(options ?? {}), label: "final" }
+        { ...(options ?? {}), cache: "moa", label: "final" }
     )
     return result
 }