wehub-resource-sync
94 lines
3.1 KiB
TOML
94 lines
3.1 KiB
TOML
[tool.poetry]
|
|
name = "aisuite"
|
|
version = "0.1.14"
|
|
description = "Uniform access layer for LLMs"
|
|
authors = ["Andrew Ng <ng@deeplearning.ai>", "Rohit P <rohit.prasad15@gmail.com>"]
|
|
maintainers = ["Andrew Ng <ng@deeplearning.ai>", "Rohit P <rohit.prasad15@gmail.com>"]
|
|
readme = "README.md"
|
|
include = ["aisuite/tracing/static/viewer/**/*"]
|
|
|
|
[tool.poetry.dependencies]
|
|
python = "^3.10"
|
|
anthropic = { version = "^0.30.1", optional = true }
|
|
boto3 = { version = "^1.34.144", optional = true }
|
|
cohere = { version = "^5.12.0", optional = true }
|
|
vertexai = { version = "^1.63.0", optional = true }
|
|
google-cloud-speech = { version = "^2.33.0", optional = true }
|
|
deepgram-sdk = { version = "^5.0.0", optional = true }
|
|
soundfile = { version = "^0.12.1", optional = true }
|
|
scipy = { version = "^1.11.0", optional = true }
|
|
numpy = { version = "^1.24.0", optional = true }
|
|
groq = { version = "^0.9.0", optional = true }
|
|
mistralai = { version = "^1.0.3", optional = true }
|
|
ibm-watsonx-ai = { version = "^1.1.16", optional = true }
|
|
docstring-parser = { version = ">=0.16,<1.0" }
|
|
pydantic = { version = "^2.10.0" }
|
|
cerebras_cloud_sdk = { version = "^1.19.0", optional = true }
|
|
openai = { version = "^1.107.0", optional = true }
|
|
mcp = { version = "^1.1.2", optional = true }
|
|
nest-asyncio = { version = "^1.6.0", optional = true }
|
|
psycopg = { version = "^3.3.4", optional = true }
|
|
|
|
# Optional dependencies for different providers
|
|
httpx = "~0.27.0"
|
|
[tool.poetry.extras]
|
|
anthropic = ["anthropic"]
|
|
aws = ["boto3"]
|
|
azure = []
|
|
cerebras = ["cerebras_cloud_sdk"]
|
|
cohere = ["cohere"]
|
|
deepgram = ["deepgram-sdk", "soundfile", "scipy", "numpy"]
|
|
deepseek = ["openai"]
|
|
google = ["vertexai", "google-cloud-speech"]
|
|
groq = ["groq"]
|
|
huggingface = []
|
|
mistral = ["mistralai"]
|
|
ollama = ["openai"]
|
|
lmstudio = ["openai"]
|
|
openai = ["openai"]
|
|
watsonx = ["ibm-watsonx-ai"]
|
|
mcp = ["mcp", "nest-asyncio"]
|
|
postgres = ["psycopg"]
|
|
all = ["anthropic", "boto3", "cerebras_cloud_sdk", "vertexai", "google-cloud-speech", "groq", "mistralai", "openai", "cohere", "ibm-watsonx-ai", "deepgram-sdk", "soundfile", "scipy", "numpy", "mcp", "nest-asyncio", "psycopg"] # To install all providers
|
|
|
|
[tool.poetry.group.dev.dependencies]
|
|
pre-commit = "^3.7.1"
|
|
black = "^24.4.2"
|
|
python-dotenv = "^1.0.1"
|
|
openai = "^1.107.0"
|
|
groq = "^0.9.0"
|
|
anthropic = "^0.30.1"
|
|
notebook = "^7.2.1"
|
|
ollama = "^0.2.1"
|
|
mistralai = "^1.0.3"
|
|
boto3 = "^1.34.144"
|
|
fireworks-ai = "^0.14.0"
|
|
chromadb = "^0.5.4"
|
|
sentence-transformers = "^3.0.1"
|
|
datasets = "^2.20.0"
|
|
vertexai = "^1.63.0"
|
|
google-cloud-speech = "^2.33.0"
|
|
deepgram-sdk = "^5.0.0"
|
|
ibm-watsonx-ai = "^1.1.16"
|
|
cerebras_cloud_sdk = "^1.19.0"
|
|
|
|
[tool.poetry.group.test]
|
|
optional = true
|
|
|
|
[tool.poetry.group.test.dependencies]
|
|
pytest = "^8.2.2"
|
|
pytest-cov = "^6.0.0"
|
|
pytest-asyncio = "^0.24.0"
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|
|
|
|
[tool.pytest.ini_options]
|
|
testpaths="tests"
|
|
markers = [
|
|
"integration: marks tests as integration tests that interact with external services",
|
|
"llm: marks tests that make real LLM API calls and incur costs (subset of integration)",
|
|
"mcp_server: marks tests that require MCP server functionality (e.g., npx, external MCP servers)",
|
|
]
|