voyageai 0.1.0__tar.gz

voyageai-0.1.0/LICENSE

@@ -0,0 +1,23 @@
+The MIT License
+
+Copyright (c) 2023 OpenAI (https://openai.com)
+
+Copyright (c) 2023 VoyageAI (https://voyageai.com)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

voyageai-0.1.0/PKG-INFO

@@ -0,0 +1,51 @@
+Metadata-Version: 2.1
+Name: voyageai
+Version: 0.1.0
+Summary: 
+Author: Yujie Qian
+Author-email: yujieq@voyageai.com
+Requires-Python: >=3.7.1,<4.0.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Dist: aiohttp (>=3.5,<4.0)
+Requires-Dist: pandas (>=1.2.3)
+Requires-Dist: plotly (>=5.0.0)
+Requires-Dist: requests (>=2.20,<3.0)
+Requires-Dist: scikit-learn (>=1.0.2)
+Requires-Dist: tenacity (>=8.0.1)
+Description-Content-Type: text/markdown
+
+# VoyageAI Python Library
+The VoyageAI Python library provides convenient access to the VoyageAI API.
+
+## Installation
+Use `pip` to install the package:
+```bash
+pip install voyageai
+```
+
+## Usage
+First, import the libary at the top of a file:
+```python
+import voyageai
+```
+The library needs to be configured with your VoyageAI API key. Either set it as the `VOYAGEAI_API_KEY` environment variable before using the library:
+```bash
+export VOYAGEAI_API_KEY="[ Your VoyageAI API key ]"
+```
+Or set voyageai.api_key to its value:
+```python
+voyageai.api_key = "[ Your VoyageAI API key ]"
+```
+
+### Embedding
+The VoyageAI embedding is accessible through our API (see documentation at ...). This library provides Python SDKs for calling the embedding API.
+```python
+text = "This is a sample text."
+model = "voyage-api-v0"
+embedding = voyageai.Embedding.create(input=text, model=model)["data"][0]["embedding"]
+```
+

voyageai-0.1.0/README.md

@@ -0,0 +1,30 @@
+# VoyageAI Python Library
+The VoyageAI Python library provides convenient access to the VoyageAI API.
+
+## Installation
+Use `pip` to install the package:
+```bash
+pip install voyageai
+```
+
+## Usage
+First, import the libary at the top of a file:
+```python
+import voyageai
+```
+The library needs to be configured with your VoyageAI API key. Either set it as the `VOYAGEAI_API_KEY` environment variable before using the library:
+```bash
+export VOYAGEAI_API_KEY="[ Your VoyageAI API key ]"
+```
+Or set voyageai.api_key to its value:
+```python
+voyageai.api_key = "[ Your VoyageAI API key ]"
+```
+
+### Embedding
+The VoyageAI embedding is accessible through our API (see documentation at ...). This library provides Python SDKs for calling the embedding API.
+```python
+text = "This is a sample text."
+model = "voyage-api-v0"
+embedding = voyageai.Embedding.create(input=text, model=model)["data"][0]["embedding"]
+```

voyageai-0.1.0/pyproject.toml

@@ -0,0 +1,23 @@
+[tool.poetry]
+name = "voyageai"
+version = "0.1.0"
+description = ""
+authors = ["Yujie Qian <yujieq@voyageai.com>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.7.1"
+requests = "^2.20"
+aiohttp = "^3.5"
+scikit-learn = ">=1.0.2"
+tenacity = ">=8.0.1"
+plotly = ">=5.0.0"
+pandas = ">=1.2.3"
+
+[tool.poetry.group.test.dependencies]
+pytest = "^7.4.2"
+pytest-asyncio = "^0.21.1"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

voyageai-0.1.0/voyageai/__init__.py

@@ -0,0 +1,55 @@
+# VoyageAI Python bindings.
+#
+# Originally forked from the MIT-licensed Stripe Python bindings.
+
+import os
+import sys
+from typing import TYPE_CHECKING, Optional, Union, Callable
+
+from contextvars import ContextVar
+
+if "pkg_resources" not in sys.modules:
+    # workaround for the following:
+    # https://github.com/benoitc/gunicorn/pull/2539
+    sys.modules["pkg_resources"] = object()  # type: ignore[assignment]
+    import aiohttp
+
+    del sys.modules["pkg_resources"]
+
+from voyageai.api_resources import Embedding
+from voyageai.error import APIError, InvalidRequestError, VoyageAIError
+from voyageai.version import VERSION
+from voyageai.embeddings_utils import get_embedding, get_embeddings, aget_embedding, aget_embeddings
+
+if TYPE_CHECKING:
+    import requests
+    from aiohttp import ClientSession
+
+api_key = os.environ.get("VOYAGEAI_API_KEY")
+# Path of a file with an API key, whose contents can change. Supercedes
+# `api_key` if set.  The main use case is volume-mounted Kubernetes secrets,
+# which are updated automatically.
+api_key_path: Optional[str] = os.environ.get("VOYAGEAI_API_KEY_PATH")
+
+organization = os.environ.get("VOYAGEAI_ORGANIZATION")
+api_base = os.environ.get("VOYAGEAI_API_BASE", "http://api.voyageai.com/stage/api/v0")
+api_type = os.environ.get("VOYAGEAI_API_TYPE", "voyageai")
+api_version = os.environ.get("VOYAGEAI_API_VERSION", None)
+verify_ssl_certs = True  # No effect. Certificates are always verified.
+proxy = None
+app_info = None
+enable_telemetry = False  # Ignored; the telemetry feature was removed.
+ca_bundle_path = None  # No longer used, feature was removed
+debug = False
+log = None  # Set to either 'debug' or 'info', controls console logging
+
+requestssession: Optional[
+    Union["requests.Session", Callable[[], "requests.Session"]]
+] = None # Provide a requests.Session or Session factory.
+
+aiosession: ContextVar[Optional["ClientSession"]] = ContextVar(
+    "aiohttp-session", default=None
+)  # Acts as a global aiohttp ClientSession that reuses connections.
+# This is user-supplied; otherwise, a session is remade for each request.
+
+__version__ = VERSION