Merge pull request #1 from NivekT/LanceDB

Touch up LanceDB after refactor
hegelai · Aug 11, 2023 · ad99417 · ad99417
2 parents bf475ac + 9668c82
commit ad99417
Show file tree

Hide file tree

Showing 16 changed files with 1,298 additions and 11 deletions.
diff --git a/.gitignore b/.gitignore
@@ -175,3 +175,5 @@ cython_debug/
 
 **/.chroma/**
 /ptvenv
+
+**/lancedb/**
diff --git a/README.md b/README.md
@@ -107,7 +107,7 @@ Vector Databases and Data Utility
 - Chroma - **Supported**
 - Weaviate - **Supported**
 - Qdrant - _In Progress_
-- LanceDB - Exploratory
+- LanceDB - **Supported**
 - Milvus - Exploratory
 - Pinecone - Exploratory
 - LlamaIndex - Exploratory

diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -11,7 +11,7 @@
 project = "prompttools"
 copyright = "2023, Hegel AI"
 author = "Hegel AI"
-release = "0.0.31"
+release = "0.0.32"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

diff --git a/docs/source/experiment.rst b/docs/source/experiment.rst
@@ -44,4 +44,6 @@ Integrated Experiment APIs
 
 .. autoclass:: WeaviateExperiment
 
+.. autoclass:: LanceDBExperiment
+
 .. autoclass:: MindsDBExperiment
diff --git a/examples/notebooks/LanceDBExperiment.ipynb b/examples/notebooks/LanceDBExperiment.ipynb
diff --git a/examples/notebooks/LangChainRouterChainExperiment.ipynb b/examples/notebooks/LangChainRouterChainExperiment.ipynb
@@ -56,7 +56,7 @@
  " \"name\": \"restaurant\",\n",
  " \"description\": \"Good for building a restaurant\",\n",
  " \"prompt_template\": restaurant_template,\n",
- " }\n",
+ " },\n",
  " ],\n",
  "]\n",
  "\n",
@@ -77,7 +77,7 @@
  "\n",
  "experiment.evaluate(\"similar_to_expected\", semantic_similarity, expected=[expected] * 2)\n",
  "\n",
- "experiment.visualize()\n"
+ "experiment.visualize()"
  ]
  }
  ],