Always create the ~/r2ai.models symlink

radareorg · May 14, 2024 · 1d96a4f · 1d96a4f
1 parent b5a0dee
commit 1d96a4f
Showing 1 changed file with 12 additions and 2 deletions.
diff --git 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: