b7427e2de7
* init server refactor * refactored websockets server/client code to use internal server API * added intentional fail on test * update workflow to try and get test to pass remotely * refactor to put websocket code in a separate subdirectory * added fastapi rest server * add error handling * modified interface return style * disabled certain tests on remote * added SSE response option for user_message * fix ws interface test * fallback for oai key * add soft fail for test when localhost is borked * add step_yield for all server related interfaces * extra catch * update toml + lock with server add-ons (add uvicorn+fastapi, move websockets to server extra) * regen lock file * added pytest-asyncio as an extra in dev * add pydantic to deps * renamed CreateConfig to CreateAgentConfig * fixed POST request for creating agent + tested it
67 lines
2.1 KiB
TOML
67 lines
2.1 KiB
TOML
[tool.poetry]
|
|
name = "pymemgpt"
|
|
version = "0.2.7"
|
|
packages = [
|
|
{include = "memgpt"}
|
|
]
|
|
description = "Teaching LLMs memory management for unbounded context"
|
|
authors = [
|
|
"Charles Packer <contact@charlespacker.com>",
|
|
"Vivian Fang <hi@vivi.sh>",
|
|
"Sarah Wooders <sarahwooders@gmail.com>",
|
|
"Shishir Patil <shishirpatil@berkeley.edu>",
|
|
"Kevin Lin <kevinlin@eecs.berkeley.edu>"
|
|
]
|
|
license = "Apache License"
|
|
readme = "README.md"
|
|
|
|
[tool.poetry.scripts]
|
|
memgpt = "memgpt.main:app"
|
|
|
|
[tool.poetry.dependencies]
|
|
python = "<3.12,>=3.9"
|
|
typer = {extras = ["all"], version = "^0.9.0"}
|
|
questionary = "^2.0.1"
|
|
pytz = "^2023.3.post1"
|
|
tqdm = "^4.66.1"
|
|
black = { version = "^23.10.1", optional = true }
|
|
pytest = { version = "^7.4.3", optional = true }
|
|
llama-index = "^0.9.13"
|
|
setuptools = "^68.2.2"
|
|
datasets = { version = "^2.14.6", optional = true}
|
|
prettytable = "^3.9.0"
|
|
pgvector = { version = "^0.2.3", optional = true }
|
|
psycopg = { version = "^3.1.12", optional = true }
|
|
psycopg-binary = { version = "^3.1.12", optional = true }
|
|
psycopg2-binary = { version = "^2.9.9", optional = true }
|
|
huggingface-hub = { version = "0.17.3", optional = true }
|
|
transformers = { version = "4.34.1", optional = true }
|
|
pre-commit = {version = "^3.5.0", optional = true }
|
|
pg8000 = {version = "^1.30.3", optional = true}
|
|
torch = {version = ">=2.0.0, !=2.0.1, !=2.1.0", optional = true}
|
|
websockets = {version = "^12.0", optional = true}
|
|
docstring-parser = "^0.15"
|
|
lancedb = "^0.3.3"
|
|
httpx = "^0.25.2"
|
|
numpy = "^1.26.2"
|
|
demjson3 = "^3.0.6"
|
|
tiktoken = "^0.5.1"
|
|
python-box = "^7.1.1"
|
|
pypdf = "^3.17.1"
|
|
pyyaml = "^6.0.1"
|
|
fastapi = {version = "^0.104.1", optional = true}
|
|
uvicorn = {version = "^0.24.0.post1", optional = true}
|
|
chromadb = "^0.4.18"
|
|
pytest-asyncio = {version = "^0.23.2", optional = true}
|
|
pydantic = "^2.5.2"
|
|
|
|
[tool.poetry.extras]
|
|
local = ["torch", "huggingface-hub", "transformers"]
|
|
postgres = ["pgvector", "psycopg", "psycopg-binary", "psycopg2-binary", "pg8000"]
|
|
dev = ["pytest", "pytest-asyncio", "black", "pre-commit", "datasets"]
|
|
server = ["websockets", "fastapi", "uvicorn"]
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|