Fix location tool chroma db location

wri · Dec 5, 2024 · fbfc03b · fbfc03b
1 parent 80dad02
commit fbfc03b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/zeno/tools/location/location_tool.py b/zeno/tools/location/location_tool.py
@@ -10,8 +10,7 @@
 gadm = fiona.open("data/gadm_410_small.gpkg")
 
 vectorstore = Chroma(
-    # persist_directory="data/chroma_gadm",
-    persist_directory="/Users/tam/Desktop/chroma_gadm",
+    persist_directory="data/chroma_gadm",
     embedding_function=OllamaEmbeddings(
         model="nomic-embed-text", base_url=os.environ["OLLAMA_BASE_URL"]
     ),