From a8f76d178fd406d42805759e762ac2ebd591c2b1 Mon Sep 17 00:00:00 2001
From: Ty <124617566+tmichaeldb@users.noreply.github.com>
Date: Fri, 19 Apr 2024 15:06:23 -0700
Subject: [PATCH] Update using_agents.py

---
 examples/using_agents.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/using_agents.py b/examples/using_agents.py
index cb1b2ea..3504079 100644
--- a/examples/using_agents.py
+++ b/examples/using_agents.py
@@ -14,6 +14,7 @@
 langchain_model = con.models.create(
     'agent_model',
     predict='answer',
+    mode='retrieval', # Use retrieval mode if using knowledge bases.
     engine='langchain',
     prompt_template='You are a spicy, cheeky assistant. Add some personality and flare when responding to the user question: {{question}}',
     model_name='gpt-4-0125-preview' # This is the underlying LLM. Can use OpenAI, Claude, local Ollama, etc