fix(desktop): Make Desktop Great Again! (#3537)

Co-authored-by: Charles Packer <packercharles@gmail.com>
Co-authored-by: Kian Jones <kian@Kians-MacBook-Pro.local>
This commit is contained in:
Kian Jones
2025-07-30 10:17:35 -07:00
committed by GitHub
parent 6ebbaed8c4
commit 1df22d0c71
3 changed files with 588 additions and 558 deletions

View File

@@ -154,10 +154,10 @@ def register_functions(dbapi_connection, connection_record):
print("Detected aiosqlite connection - sqlite-vec will be loaded per-query")
else:
# For sync connections
dbapi_connection.enable_load_extension(True)
sqlite_vec.load(dbapi_connection)
dbapi_connection.enable_load_extension(False)
print("Successfully loaded sqlite-vec extension (sync)")
# dbapi_connection.enable_load_extension(True)
# sqlite_vec.load(dbapi_connection)
# dbapi_connection.enable_load_extension(False)
print("sqlite3 extension loading") # Successfully loaded sqlite-vec extension (sync)")
except Exception as e:
raise RuntimeError(f"Failed to load sqlite-vec extension: {e}")

1127
poetry.lock generated

File diff suppressed because it is too large Load Diff

View File

@@ -93,8 +93,8 @@ matplotlib = "^3.10.1"
asyncpg = {version = "^0.30.0", optional = true}
tavily-python = "^0.7.2"
mistralai = "^1.8.1"
uvloop = {version = "^0.21.0", optional = true}
granian = {version = "^2.3.2", extras = ["uvloop", "reload"], optional = true}
uvloop = {version = "^0.21.0", optional = true, markers = "sys_platform != 'win32'"}
granian = {version = "^2.3.2", extras = ["reload"], optional = true}
redis = {version = "^6.2.0", optional = true}
structlog = "^25.4.0"
certifi = "^2025.6.15"
@@ -102,6 +102,7 @@ aioboto3 = {version = "^14.3.0", optional = true}
pinecone = {extras = ["asyncio"], version = "^7.3.0", optional = true}
markitdown = {extras = ["docx", "pdf", "pptx"], version = "^0.1.2"}
google-cloud-profiler = {version = "^4.1.0", optional = true}
sqlite-vec = {version = "^0.1.7a2", optional = true}
orjson = "^3.11.1"
@@ -117,7 +118,7 @@ external-tools = ["docker", "langchain", "wikipedia", "langchain-community", "fi
tests = ["wikipedia"]
bedrock = ["boto3", "aioboto3"]
google = ["google-genai"]
desktop = ["pyright", "websockets", "fastapi", "uvicorn", "docker", "langchain", "wikipedia", "langchain-community", "locust"]
desktop = ["pyright", "websockets", "fastapi", "uvicorn", "docker", "langchain", "wikipedia", "langchain-community", "locust", "sqlite-vec"]
all = ["pgvector", "pg8000", "psycopg2-binary", "psycopg2", "pytest", "pytest-asyncio", "pexpect", "black", "pre-commit", "pyright", "pytest-order", "autoflake", "isort", "websockets", "fastapi", "uvicorn", "docker", "langchain", "wikipedia", "langchain-community", "locust", "uvloop", "granian", "redis", "pinecone", "google-cloud-profiler"]
[tool.poetry.group.dev.dependencies]
@@ -135,6 +136,10 @@ aiosqlite = "^0.21.0"
# https://github.com/asg017/sqlite-vec/issues/148
sqlite-vec = "^0.1.7a2"
[tool.poetry.group.desktop.dependencies]
sqlite-vec = "^0.1.7a2"
[tool.black]
line-length = 140
target-version = ['py310', 'py311', 'py312', 'py313']