Update UpTrainCallback.ipynb

run-llama · Feb 14, 2024 · df135a3 · df135a3
1 parent 26ffd8e
commit df135a3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 322 deletions.
diff --git a/docs/examples/callbacks/UpTrainCallback.ipynb b/docs/examples/callbacks/UpTrainCallback.ipynb
@@ -67,6 +67,7 @@
    "source": [
     "%pip install llama-index-postprocessor-cohere-rerank\n",
     "%pip install llama-index-readers-web\n",
+    "%pip install llama-index-callback-uptrain\n",
     "%pip install -q html2text llama-index pandas tqdm uptrain cohere"
    ]
   },
@@ -83,11 +84,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from llama_index.core import Settings, VectorStoreIndex\n",
+    "from llama_index.core.settings import Settings\n",
+    "from llama_index.core import VectorStoreIndex\n",
     "from llama_index.core.node_parser import SentenceSplitter\n",
     "from llama_index.readers.web import SimpleWebPageReader\n",
     "from llama_index.core.callbacks import CallbackManager\n",
-    "from llama_index.legacy.callbacks import UpTrainCallbackHandler\n",
+    "from llama_index.callbacks.uptrain.base import UpTrainCallbackHandler\n",
     "from llama_index.core.query_engine import SubQuestionQueryEngine\n",
     "from llama_index.core.tools import QueryEngineTool, ToolMetadata\n",
     "from llama_index.core.postprocessor.llm_rerank import LLMRerank\n",

diff --git a/llama-index-legacy/llama_index/legacy/callbacks/uptrain_callback.py b/llama-index-legacy/llama_index/legacy/callbacks/uptrain_callback.py