From ecb5fe3026809675e647fefd055dca09f0faf621 Mon Sep 17 00:00:00 2001
From: "Lennart J. Kurzweg (Nx2)" <git@nx2.site>
Date: Fri, 30 Aug 2024 18:09:42 +0200
Subject: [PATCH] needs experimental

---
 requirements.txt        | 1 +
 suite_settings/tests.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 22897d4..b39e670 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,6 +2,7 @@ langchain
 langchain-core
 langchain-ollama
 langchain-community
+langchain-experimental
 langgraph
 seaborn
 pandas
diff --git a/suite_settings/tests.py b/suite_settings/tests.py
index 2190cd8..b67c731 100644
--- a/suite_settings/tests.py
+++ b/suite_settings/tests.py
@@ -55,7 +55,7 @@ tests = {
         runnable_input={
             "system_msg": 'You are a helpful assistant.',
             "human_msg": 'Is 31515261 divisible by 425? If not, whats the remainder?',
-            "tools": { "python_repl": save_python_repl },
+            "tools": { "save_python_repl": save_python_repl },
         },
         validator=regex_match_any,
         validation_input={