feat: Add turbopuffer as dependency (#3655)

This commit is contained in:
Matthew Zhou
2025-07-30 11:37:48 -07:00
committed by GitHub
parent 73e74270f4
commit 3ad790b642
3 changed files with 322 additions and 556 deletions

View File

@@ -292,6 +292,11 @@ class Settings(BaseSettings):
pinecone_agent_index: Optional[str] = "recall"
upsert_pinecone_indices: bool = False
# For tpuf - currently only for archival memories
use_tpuf: bool = False
tpuf_api_key: Optional[str] = None
tpuf_region: str = "gcp-us-central1.turbopuffer.com"
# File processing timeout settings
file_processing_timeout_minutes: int = 30
file_processing_timeout_error_message: str = "File processing timed out after {} minutes. Please try again."

870
poetry.lock generated

File diff suppressed because it is too large Load Diff

View File

@@ -105,6 +105,7 @@ google-cloud-profiler = {version = "^4.1.0", optional = true}
sqlite-vec = {version = "^0.1.7a2", optional = true}
orjson = "^3.11.1"
modal = {version = "^1.1.0", optional = true}
turbopuffer = {version = "^0.5.17", optional = true}
[tool.poetry.extras]
@@ -120,7 +121,7 @@ tests = ["wikipedia"]
bedrock = ["boto3", "aioboto3"]
google = ["google-genai"]
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"]
all = ["pgvector", "turbopuffer", "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]
black = "^24.4.2"