From 1d96a4fe0145ecd4cce8f4744619b006ecc5e410 Mon Sep 17 00:00:00 2001
From: pancake <pancake@nopcode.org>
Date: Tue, 14 May 2024 19:20:06 +0200
Subject: [PATCH] Always create the ~/r2ai.models symlink

---
 r2ai/main.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/r2ai/main.py b/r2ai/main.py
index 6da5301f..a5b601ba 100755
--- a/r2ai/main.py
+++ b/r2ai/main.py
@@ -19,8 +19,18 @@ def __main__():
 except:
 	pass
 
-# run `make vectordb` because pip install vectordb2 is too old
-sys.path.append(f"vectordb")
+
+# create symlink if it doesnt exist
+try:
+    import appdirs
+    dst = os.environ["HOME"] + "/.r2ai.models"
+    udd = appdirs.user_data_dir("r2ai")
+    src = os.path.join(udd, "models")
+    if not os.path.exists(dst):
+        os.symlink(src, dst)
+except:
+    traceback.print_exc()
+    pass
 
 OPENAI_KEY = ""
 try: