Add support for anthropic (#1819)

This commit is contained in:
Dev Khant
2024-09-07 02:12:22 +05:30
committed by GitHub
parent 965f7a3735
commit d32ae1a0b1
13 changed files with 82 additions and 84 deletions

View File

@@ -9,19 +9,7 @@ try:
import psycopg2
from psycopg2.extras import execute_values
except ImportError:
user_input = input("The 'psycopg2' library is required. Install it now? [y/N]: ")
if user_input.lower() == 'y':
try:
subprocess.check_call([sys.executable, "-m", "pip", "install", "psycopg2"])
import psycopg2
from psycopg2.extras import execute_values
except subprocess.CalledProcessError:
print("Failed to install 'psycopg2'. Please install it manually using 'pip install psycopg2'.")
sys.exit(1)
else:
print("The required 'psycopg2' library is not installed.")
sys.exit(1)
raise ImportError("The 'psycopg2' library is required. Please install it using 'pip install psycopg2'.")
from mem0.vector_stores.base import VectorStoreBase