feat: exposes memory option to users in gateway definition.

Snowflake-Labs · Dec 6, 2024 · 8d75e0a · 8d75e0a
1 parent c1a2f25
commit 8d75e0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agent_gateway/gateway/gateway.py b/agent_gateway/gateway/gateway.py
@@ -156,7 +156,7 @@ def __init__(
             max_retries: Maximum number of replans to do. Defaults to 2.
             planner_llm: Name of Snowflake Cortex LLM to use for planning.
             agent_llm: Name of Snowflake Cortex LLM to use for planning.
-            memory: Boolean to turn on memory mechanism or not. Defaults to False.
+            memory: Boolean to turn on memory mechanism or not. Defaults to True.
             planner_example_prompt: Example prompt for planning. Defaults to SNOWFLAKE_PLANNER_PROMPT.
             planner_example_prompt_replan: Example prompt for replanning.
                 Assign this if you want to use different example prompt for replanning.