Skip to content

Commit

Permalink
feat: openrouter with async, async streaming, added async on qdrant, …
Browse files Browse the repository at this point in the history
…added async on mysql using aiomysql
  • Loading branch information
vikyw89 committed Jul 5, 2024
1 parent 8cc20fb commit 7999266
Show file tree
Hide file tree
Showing 6 changed files with 1,295 additions and 230 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -18,3 +18,4 @@ htmlcov
chroma.sqlite3
*.bin
.coverage.*
.vscode
49 changes: 40 additions & 9 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -5,20 +5,26 @@ build-backend = "flit_core.buildapi"
[project]
name = "vanna"
version = "0.6.2"
authors = [
{ name="Zain Hoda", email="[email protected]" },
]
authors = [{ name = "Zain Hoda", email = "[email protected]" }]

description = "Generate SQL queries from natural language"
readme = "README.md"
requires-python = ">=3.9"
classifiers = [
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
]
dependencies = [
"requests", "tabulate", "plotly", "pandas", "sqlparse", "kaleido", "flask", "flask-sock", "sqlalchemy"
"requests",
"tabulate",
"plotly",
"pandas",
"sqlparse",
"kaleido",
"flask",
"flask-sock",
"sqlalchemy",
]

[project.urls]
Expand All @@ -27,13 +33,38 @@ dependencies = [

[project.optional-dependencies]
postgres = ["psycopg2-binary", "db-dtypes"]
mysql = ["PyMySQL"]
mysql = ["PyMySQL", "aiomysql"]
clickhouse = ["clickhouse_connect"]
bigquery = ["google-cloud-bigquery"]
snowflake = ["snowflake-connector-python"]
duckdb = ["duckdb"]
google = ["google-generativeai", "google-cloud-aiplatform"]
all = ["psycopg2-binary", "db-dtypes", "PyMySQL", "google-cloud-bigquery", "snowflake-connector-python", "duckdb", "openai", "mistralai", "chromadb", "anthropic", "zhipuai", "marqo", "google-generativeai", "google-cloud-aiplatform", "qdrant-client", "fastembed", "ollama", "httpx", "opensearch-py", "opensearch-dsl", "transformers", "pinecone-client", "pymilvus[model]"]
all = [
"psycopg2-binary",
"db-dtypes",
"PyMySQL",
"aiomysql",
"google-cloud-bigquery",
"snowflake-connector-python",
"duckdb",
"openai",
"mistralai",
"chromadb",
"anthropic",
"zhipuai",
"marqo",
"google-generativeai",
"google-cloud-aiplatform",
"qdrant-client",
"fastembed",
"ollama",
"httpx",
"opensearch-py",
"opensearch-dsl",
"transformers",
"pinecone-client",
"pymilvus[model]",
]
test = ["tox"]
chromadb = ["chromadb"]
openai = ["openai"]
Expand Down
Loading

0 comments on commit 7999266

Please sign in to comment.