fix: only access memory_context if memory arg is True

Snowflake-Labs · Dec 6, 2024 · 9937b97 · 9937b97
1 parent 6888016
commit 9937b97
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/agent_gateway/gateway/gateway.py b/agent_gateway/gateway/gateway.py
@@ -468,8 +468,8 @@ async def acall(
             inputs["context"] = formatted_contexts
 
         max_memory = 3  # TODO consider exposing this to users
-
-        if len(self.memory_context) <= max_memory:
-            self.memory_context.append({"Question:": input, "Answer": answer})
+        if self.memory:
+            if len(self.memory_context) <= max_memory:
+                self.memory_context.append({"Question:": input, "Answer": answer})
 
         return answer