Bump to 0.3.0 and update docs (#189)

Version bump to 0.3.0 and updates documentation/README!
redis · Aug 2, 2024 · c852e3d · c852e3d
1 parent 8abec54
commit c852e3d
Show file tree

Hide file tree

Showing 13 changed files with 265 additions and 119 deletions.
diff --git a/README.md b/README.md
diff --git a/docs/_static/js/sidebar.js b/docs/_static/js/sidebar.js
@@ -4,29 +4,26 @@ const toc = [
         { title: "Install", path: "/overview/installation.html" },
         { title: "CLI", path: "/overview/cli.html" },
     ]},
-    { header: "User Guides", toc: [
-        { title: "Getting Started", path: "/user_guide/getting_started_01.html" },
-        { title: "Query and Filter", path: "/user_guide/hybrid_queries_02.html" },
-        { title: "JSON vs Hash Storage", path: "/user_guide/hash_vs_json_05.html" },
-        { title: "Vectorizers", path: "/user_guide/vectorizers_04.html" },
-        { title: "Rerankers", path: "/user_guide/rerankers_06.html" },
-        { title: "Semantic Caching", path: "/user_guide/llmcache_03.html" },
-        { title: "Semantic Routing", path: "/user_guide/semantic_router_08.html" },
-    ]},
     { header: "API", toc: [
         { title: "Schema", path: "/api/schema.html"},
         { title: "Search Index", path: "/api/searchindex.html" },
         { title: "Query", path: "/api/query.html" },
         { title: "Filter", path: "/api/filter.html" },
-    ]},
-    { header: "Utils", toc: [
         { title: "Vectorizers", path: "/api/vectorizer.html" },
         { title: "Rerankers", path: "/api/reranker.html" },
-    ]},
-    { header: "Extensions", toc: [
         { title: "LLM Cache", path: "/api/cache.html" },
+        { title: "LLM Session Manager", path: "/api/session_manager.html" },
         { title: "Semantic Router", path: "/api/router.html" },
-    ]}
+    ]},
+    { header: "User Guides", toc: [
+        { title: "Getting Started", path: "/user_guide/getting_started_01.html" },
+        { title: "Query and Filter", path: "/user_guide/hybrid_queries_02.html" },
+        { title: "JSON vs Hash Storage", path: "/user_guide/hash_vs_json_05.html" },
+        { title: "Using Vectorizers", path: "/user_guide/vectorizers_04.html" },
+        { title: "Using Rerankers", path: "/user_guide/rerankers_06.html" },
+        { title: "Semantic Caching", path: "/user_guide/llmcache_03.html" },
+        { title: "Semantic Routing", path: "/user_guide/semantic_router_08.html" },
+    ]},
 ];
 
 document.addEventListener('DOMContentLoaded', function() {

diff --git a/docs/api/cache.rst b/docs/api/cache.rst
@@ -1,4 +1,3 @@
-
 *********
 LLM Cache
 *********

diff --git a/docs/api/index.md b/docs/api/index.md
@@ -18,6 +18,7 @@ filter
 vectorizer
 reranker
 cache
+session_manager
 router
 ```
 
diff --git a/docs/api/session_manager.rst b/docs/api/session_manager.rst
@@ -0,0 +1,29 @@
+*******************
+LLM Session Manager
+*******************
+
+
+SemanticSessionManager
+======================
+
+.. _semantic_session_manager_api:
+
+.. currentmodule:: redisvl.extensions.session_manager
+
+.. autoclass:: SemanticSessionManager
+   :show-inheritance:
+   :members:
+   :inherited-members:
+
+
+StandardSessionManager
+======================
+
+.. _standard_session_manager_api:
+
+.. currentmodule:: redisvl.extensions.session_manager
+
+.. autoclass:: StandardSessionManager
+   :show-inheritance:
+   :members:
+   :inherited-members:
diff --git a/docs/api/vectorizer.rst b/docs/api/vectorizer.rst
@@ -26,6 +26,18 @@ OpenAITextVectorizer
    :members:
 
 
+AzureOpenAITextVectorizer
+=========================
+
+.. _azureopenaitextvectorizer_api:
+
+.. currentmodule:: redisvl.utils.vectorize.text.azureopenai
+
+.. autoclass:: AzureOpenAITextVectorizer
+   :show-inheritance:
+   :members:
+
+
 VertexAITextVectorizer
 ======================
 
@@ -49,3 +61,14 @@ CohereTextVectorizer
    :show-inheritance:
    :members:
 
+
+CustomTextVectorizer
+====================
+
+.. _customtextvectorizer_api:
+
+.. currentmodule:: redisvl.utils.vectorize.text.custom
+
+.. autoclass:: CustomTextVectorizer
+   :show-inheritance:
+   :members:
diff --git a/docs/overview/cli.ipynb b/docs/overview/cli.ipynb
@@ -24,7 +24,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[32m09:44:14\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   RedisVL version 0.2.3\n"
+      "\u001b[32m22:33:29\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   RedisVL version 0.3.0\n"
      ]
     }
    ],
@@ -46,9 +46,17 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Overwriting schema.yaml\n"
+     ]
+    }
+   ],
    "source": [
     "%%writefile schema.yaml\n",
     "\n",
@@ -79,7 +87,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[32m14:22:33\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Index created successfully\n"
+      "\u001b[32m22:33:31\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Index created successfully\n"
      ]
     }
    ],
@@ -97,8 +105,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[32m14:22:34\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Indices:\n",
-      "\u001b[32m14:22:34\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   1. vectorizers\n"
+      "\u001b[32m22:33:33\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Indices:\n",
+      "\u001b[32m22:33:33\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   1. vectorizers\n"
      ]
     }
    ],
@@ -125,12 +133,12 @@
       "│ vectorizers  │ HASH           │ ['doc']    │ []              │          0 │\n",
       "╰──────────────┴────────────────┴────────────┴─────────────────┴────────────╯\n",
       "Index Fields:\n",
-      "╭───────────┬─────────────┬────────┬────────────────┬────────────────╮\n",
-      "│ Name      │ Attribute   │ Type   │ Field Option   │   Option Value │\n",
-      "├───────────┼─────────────┼────────┼────────────────┼────────────────┤\n",
-      "│ sentence  │ sentence    │ TEXT   │ WEIGHT         │              1 │\n",
-      "│ embedding │ embedding   │ VECTOR │                │                │\n",
-      "╰───────────┴─────────────┴────────┴────────────────┴────────────────╯\n"
+      "╭───────────┬─────────────┬────────┬────────────────┬────────────────┬────────────────┬────────────────┬────────────────┬────────────────┬─────────────────┬────────────────╮\n",
+      "│ Name      │ Attribute   │ Type   │ Field Option   │ Option Value   │ Field Option   │ Option Value   │ Field Option   │   Option Value │ Field Option    │ Option Value   │\n",
+      "├───────────┼─────────────┼────────┼────────────────┼────────────────┼────────────────┼────────────────┼────────────────┼────────────────┼─────────────────┼────────────────┤\n",
+      "│ sentence  │ sentence    │ TEXT   │ WEIGHT         │ 1              │                │                │                │                │                 │                │\n",
+      "│ embedding │ embedding   │ VECTOR │ algorithm      │ FLAT           │ data_type      │ FLOAT32        │ dim            │            768 │ distance_metric │ COSINE         │\n",
+      "╰───────────┴─────────────┴────────┴────────────────┴────────────────┴────────────────┴────────────────┴────────────────┴────────────────┴─────────────────┴────────────────╯\n"
      ]
     }
    ],
@@ -148,7 +156,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[32m14:22:36\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Index deleted successfully\n"
+      "\u001b[32m22:33:36\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Index deleted successfully\n"
      ]
     }
    ],
@@ -166,7 +174,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[32m14:22:38\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Indices:\n"
+      "\u001b[32m22:33:37\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Indices:\n"
      ]
     }
    ],
@@ -193,7 +201,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[32m14:22:39\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Index created successfully\n"
+      "\u001b[32m22:33:40\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Index created successfully\n"
      ]
     }
    ],
@@ -211,8 +219,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[32m14:22:40\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Indices:\n",
-      "\u001b[32m14:22:40\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   1. vectorizers\n"
+      "\u001b[32m22:33:42\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Indices:\n",
+      "\u001b[32m22:33:42\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   1. vectorizers\n"
      ]
     }
    ],
@@ -272,7 +280,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "\u001b[32m14:22:42\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Index deleted successfully\n"
+      "\u001b[32m22:33:45\u001b[0m \u001b[34m[RedisVL]\u001b[0m \u001b[1;30mINFO\u001b[0m   Index deleted successfully\n"
      ]
     }
    ],
@@ -297,7 +305,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "version": "3.11.9"
   }
  },
  "nbformat": 4,

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "redisvl"
-version = "0.2.3"
+version = "0.3.0"
 description = "Python client library and CLI for using Redis as a vector database"
 authors = ["Redis Inc. <[email protected]>"]
 license = "MIT"

diff --git a/redisvl/extensions/llmcache/base.py b/redisvl/extensions/llmcache/base.py
@@ -54,6 +54,13 @@ def store(
         metadata."""
         raise NotImplementedError
 
-    def hash_input(self, prompt: str):
-        """Hashes the input using SHA256."""
+    def hash_input(self, prompt: str) -> str:
+        """Hashes the input prompt using SHA256.
+
+        Args:
+            prompt (str): Input string to be hashed.
+
+        Returns:
+            str: Hashed string.
+        """
         return hashify(prompt)
diff --git a/redisvl/extensions/llmcache/semantic.py b/redisvl/extensions/llmcache/semantic.py
@@ -200,7 +200,7 @@ def drop(
 
         Args:
             ids (Optional[str]): The document ID or IDs to remove from the cache.
-            keys (Optional[str]):
+            keys (Optional[str]): The Redis keys to remove from the cache.
         """
         if ids is not None:
             self._index.drop_keys([self._index.key(id) for id in ids])
@@ -300,6 +300,8 @@ def check(
             key = cache_search_result["id"]
             self._refresh_ttl(key)
 
+            print(cache_search_result, flush=True)
+
             # Create cache hit
             cache_hit = CacheHit(**cache_search_result)
             cache_hit_dict = {

diff --git a/redisvl/extensions/session_manager/semantic_session.py b/redisvl/extensions/session_manager/semantic_session.py
@@ -151,7 +151,7 @@ def get_relevant(
             as_text (bool): Whether to return the prompts and responses as text
             or as JSON
             top_k (int): The number of previous messages to return. Default is 5.
-            fallback (bool): Whether to drop back to recent conversation history
+            fall_back (bool): Whether to drop back to recent conversation history
                 if no relevant context is found.
             session_tag (Optional[str]): Tag to be added to entries to link to a specific
                 session. Defaults to instance uuid.

diff --git a/redisvl/utils/vectorize/text/custom.py b/redisvl/utils/vectorize/text/custom.py
@@ -44,7 +44,6 @@ def __init__(
         embed_many: Optional[Callable] = None,
         aembed: Optional[Callable] = None,
         aembed_many: Optional[Callable] = None,
-        model: str = "custom_vectorizer",
     ):
         """Initialize the Custom vectorizer.
 
@@ -72,7 +71,7 @@ def __init__(
             self._validate_aembed_many(aembed_many)
             self._aembed_many_func = aembed_many
 
-        super().__init__(model=model, dims=self._set_model_dims())
+        super().__init__(model=self.type, dims=self._set_model_dims())
 
     def _validate_embed(self, func: Callable):
         """calls the func with dummy input and validates that it returns a vector"""

diff --git a/redisvl/version.py b/redisvl/version.py
@@ -1 +1 @@
-__version__ = "0.2.3"
+__version__ = "0.3.0"