seekrai 0.0.1__tar.gz → 0.1.0__tar.gz

seekrai-0.1.0/PKG-INFO

@@ -0,0 +1,169 @@
+Metadata-Version: 2.1
+Name: seekrai
+Version: 0.1.0
+Summary: Python client for SeekrAI
+Home-page: https://gitlab.cb.ntent.com/ml/seekr-py
+License: Apache-2.0
+Author: SeekrFlow
+Author-email: support@seekr.com
+Requires-Python: >=3.9,<4.0
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: click (>=8.1.7,<9.0.0)
+Requires-Dist: eval-type-backport (>=0.1.3,<0.3.0)
+Requires-Dist: filelock (>=3.13.1,<4.0.0)
+Requires-Dist: httpx (>=0.27.0,<0.28.0)
+Requires-Dist: numpy (>=1.23.5) ; python_version < "3.12"
+Requires-Dist: numpy (>=1.26.0) ; python_version >= "3.12"
+Requires-Dist: pillow (>=10.3.0,<11.0.0)
+Requires-Dist: pyarrow (>=10.0.1)
+Requires-Dist: pydantic (>=2.7,<3.0)
+Requires-Dist: requests (>=2.31.0,<3.0.0)
+Requires-Dist: tabulate (>=0.9.0,<0.10.0)
+Requires-Dist: tqdm (>=4.66.2,<5.0.0)
+Requires-Dist: typer (>=0.9,<0.13)
+Project-URL: Homepage, https://www.seekr.com/
+Project-URL: Repository, https://gitlab.cb.ntent.com/ml/seekr-py
+Description-Content-Type: text/markdown
+
+The Seekr Python Library is the official Python client for SeekrFlow's API platform, providing a convenient way for interacting with the REST APIs and enables easy integrations with Python 3.8+ applications with easy to use synchronous and asynchronous clients.
+
+# Installation
+
+To install Seekr Python Library from PyPi, simply run:
+
+```shell Shell
+pip install --upgrade seekrai
+```
+
+## Setting up API Key
+
+> 🚧 You will need to create an account with [Seekr.com](https://seekr.com/) to obtain a SeekrFlow API Key.
+
+### Setting environment variable
+
+```shell
+export SEEKR_API_KEY=xxxxx
+```
+
+### Using the client
+
+```python
+from seekrai import SeekrFlow
+
+client = SeekrFlow(api_key="xxxxx")
+```
+
+This library contains both a python library and a CLI. We'll demonstrate how to use both below.
+
+# Usage – Python Client
+
+## Chat Completions
+
+```python
+import os
+from seekrai import SeekrFlow
+
+client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+
+response = client.chat.completions.create(
+    model="meta-llama-3-8b-instruct",
+    messages=[{"role": "user", "content": "tell me about new york"}],
+)
+print(response.choices[0].message.content)
+```
+
+### Streaming
+
+```python
+import os
+from seekrai import SeekrFlow
+
+client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+stream = client.chat.completions.create(
+    model="meta-llama-3-8b-instruct",
+    messages=[{"role": "user", "content": "tell me about new york"}],
+    stream=True,
+)
+
+for chunk in stream:
+    print(chunk.choices[0].delta.content or "", end="", flush=True)
+```
+
+### Async usage
+
+```python
+import os, asyncio
+from seekrai import AsyncSeekrFlow
+
+async_client = AsyncSeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+messages = [
+    "What are the top things to do in San Francisco?",
+    "What country is Paris in?",
+]
+
+
+async def async_chat_completion(messages):
+    async_client = AsyncSeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+    tasks = [
+        async_client.chat.completions.create(
+            model="meta-llama-3-8b-instruct",
+            messages=[{"role": "user", "content": message}],
+        )
+        for message in messages
+    ]
+    responses = await asyncio.gather(*tasks)
+
+    for response in responses:
+        print(response.choices[0].message.content)
+
+
+asyncio.run(async_chat_completion(messages))
+```
+
+## Files
+
+The files API is used for fine-tuning and allows developers to upload data to fine-tune on. It also has several methods to list all files, retrive files, and delete files. Please refer to our fine-tuning docs [here](https://docs.seekrflow.ai/docs/fine-tuning-python).
+
+```python
+import os
+from seekrai import SeekrFlow
+
+client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+
+client.files.upload(file="somedata.jsonl")  # uploads a file
+client.files.list()  # lists all uploaded files
+client.files.retrieve(id="file-d0d318cb-b7d9-493a-bd70-1cfe089d3815")  # retrieves a specific file
+client.files.delete(id="file-d0d318cb-b7d9-493a-bd70-1cfe089d3815")  # deletes a file
+```
+
+## Fine-tunes
+
+The finetune API is used for fine-tuning and allows developers to create finetuning jobs. It also has several methods to list all jobs, retrive statuses and get checkpoints. Please refer to our fine-tuning docs [here](https://docs.seekrflow.ai/docs/fine-tuning-python).
+
+```python
+import os
+from seekrai import SeekrFlow
+
+client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+
+client.fine_tuning.create(
+    training_file='file-d0d318cb-b7d9-493a-bd70-1cfe089d3815',
+    model='meta-llama-3-8b-instruct',
+    n_epochs=3,
+    n_checkpoints=1,
+    batch_size=4,
+    learning_rate=1e-5,
+    suffix='my-demo-finetune',
+    wandb_api_key='1a2b3c4d5e.......',
+)
+client.fine_tuning.list()  # lists all fine-tuned jobs
+client.fine_tuning.retrieve(id="ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b")  # retrieves information on finetune event
+client.fine_tuning.list_events(id="ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b")  # Lists events of a fine-tune job
+```
+

seekrai-0.1.0/README.md

@@ -0,0 +1,135 @@
+The Seekr Python Library is the official Python client for SeekrFlow's API platform, providing a convenient way for interacting with the REST APIs and enables easy integrations with Python 3.8+ applications with easy to use synchronous and asynchronous clients.
+
+# Installation
+
+To install Seekr Python Library from PyPi, simply run:
+
+```shell Shell
+pip install --upgrade seekrai
+```
+
+## Setting up API Key
+
+> 🚧 You will need to create an account with [Seekr.com](https://seekr.com/) to obtain a SeekrFlow API Key.
+
+### Setting environment variable
+
+```shell
+export SEEKR_API_KEY=xxxxx
+```
+
+### Using the client
+
+```python
+from seekrai import SeekrFlow
+
+client = SeekrFlow(api_key="xxxxx")
+```
+
+This library contains both a python library and a CLI. We'll demonstrate how to use both below.
+
+# Usage – Python Client
+
+## Chat Completions
+
+```python
+import os
+from seekrai import SeekrFlow
+
+client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+
+response = client.chat.completions.create(
+    model="meta-llama-3-8b-instruct",
+    messages=[{"role": "user", "content": "tell me about new york"}],
+)
+print(response.choices[0].message.content)
+```
+
+### Streaming
+
+```python
+import os
+from seekrai import SeekrFlow
+
+client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+stream = client.chat.completions.create(
+    model="meta-llama-3-8b-instruct",
+    messages=[{"role": "user", "content": "tell me about new york"}],
+    stream=True,
+)
+
+for chunk in stream:
+    print(chunk.choices[0].delta.content or "", end="", flush=True)
+```
+
+### Async usage
+
+```python
+import os, asyncio
+from seekrai import AsyncSeekrFlow
+
+async_client = AsyncSeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+messages = [
+    "What are the top things to do in San Francisco?",
+    "What country is Paris in?",
+]
+
+
+async def async_chat_completion(messages):
+    async_client = AsyncSeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+    tasks = [
+        async_client.chat.completions.create(
+            model="meta-llama-3-8b-instruct",
+            messages=[{"role": "user", "content": message}],
+        )
+        for message in messages
+    ]
+    responses = await asyncio.gather(*tasks)
+
+    for response in responses:
+        print(response.choices[0].message.content)
+
+
+asyncio.run(async_chat_completion(messages))
+```
+
+## Files
+
+The files API is used for fine-tuning and allows developers to upload data to fine-tune on. It also has several methods to list all files, retrive files, and delete files. Please refer to our fine-tuning docs [here](https://docs.seekrflow.ai/docs/fine-tuning-python).
+
+```python
+import os
+from seekrai import SeekrFlow
+
+client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+
+client.files.upload(file="somedata.jsonl")  # uploads a file
+client.files.list()  # lists all uploaded files
+client.files.retrieve(id="file-d0d318cb-b7d9-493a-bd70-1cfe089d3815")  # retrieves a specific file
+client.files.delete(id="file-d0d318cb-b7d9-493a-bd70-1cfe089d3815")  # deletes a file
+```
+
+## Fine-tunes
+
+The finetune API is used for fine-tuning and allows developers to create finetuning jobs. It also has several methods to list all jobs, retrive statuses and get checkpoints. Please refer to our fine-tuning docs [here](https://docs.seekrflow.ai/docs/fine-tuning-python).
+
+```python
+import os
+from seekrai import SeekrFlow
+
+client = SeekrFlow(api_key=os.environ.get("SEEKR_API_KEY"))
+
+client.fine_tuning.create(
+    training_file='file-d0d318cb-b7d9-493a-bd70-1cfe089d3815',
+    model='meta-llama-3-8b-instruct',
+    n_epochs=3,
+    n_checkpoints=1,
+    batch_size=4,
+    learning_rate=1e-5,
+    suffix='my-demo-finetune',
+    wandb_api_key='1a2b3c4d5e.......',
+)
+client.fine_tuning.list()  # lists all fine-tuned jobs
+client.fine_tuning.retrieve(id="ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b")  # retrieves information on finetune event
+client.fine_tuning.list_events(id="ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b")  # Lists events of a fine-tune job
+```

{seekrai-0.0.1 → seekrai-0.1.0}/pyproject.toml

@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.0.1"
+version = "0.1.0"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]
@@ -31,13 +31,12 @@ homepage = "https://gitlab.cb.ntent.com/ml/seekr-py"
 
 
 [tool.poetry.dependencies]
-python = "^3.8"
+python = "^3.9"
 typer = ">=0.9,<0.13"
 requests = "^2.31.0"
 tqdm = "^4.66.2"
 tabulate = "^0.9.0"
 pydantic = "^2.7"
-aiohttp = "^3.9.3"
 filelock = "^3.13.1"
 eval-type-backport = ">=0.1.3,<0.3.0"
 click = "^8.1.7"
@@ -47,7 +46,7 @@ numpy = [
     { version = ">=1.23.5", python = "<3.12" },
     { version = ">=1.26.0", python = ">=3.12" },
 ]
-openai = "^1.25.0"
+httpx = "^0.27.0"
 
 [tool.poetry.group.quality]
 optional = true
@@ -57,7 +56,6 @@ black = ">=23.1,<25.0"
 ruff = ">=0.3.2,<0.5.0"
 types-tqdm = "^4.65.0.0"
 types-tabulate = "^0.9.0.3"
-pre-commit = "3.5.0"
 types-requests = "^2.31.0.20240218"
 pyarrow-stubs = "^10.0.1.7"
 mypy = "^1.9.0"
@@ -74,6 +72,7 @@ tox = "^4.14.1"
 
 [tool.poetry.group.dev.dependencies]
 ipdb = "^0.13.13"
+pre-commit = "^3.7.1"
 
 [tool.poetry.urls]
 "Homepage" = "https://www.seekr.com/"

{seekrai-0.0.1 → seekrai-0.1.0}/src/seekrai/__init__.py

@@ -23,30 +23,15 @@ log: str | None = None  # Set to either 'debug' or 'info', controls console logg
 
 if TYPE_CHECKING:
     import requests
-    from aiohttp import ClientSession
 
 requestssession: "requests.Session" | Callable[[], "requests.Session"] | None = None
 
-aiosession: ContextVar["ClientSession" | None] = ContextVar(
-    "aiohttp-session", default=None
-)
-
 from seekrai.client import AsyncClient, AsyncSeekrFlow, Client, SeekrFlow
 
 
 api_key: str | None = None  # To be deprecated in the next major release
 
-# Legacy functions
-from seekrai.legacy.complete import AsyncComplete, Complete, Completion
-from seekrai.legacy.embeddings import Embeddings
-from seekrai.legacy.files import Files
-from seekrai.legacy.finetune import Finetune
-from seekrai.legacy.images import Image
-from seekrai.legacy.models import Models
-
-
 __all__ = [
-    "aiosession",
     "constants",
     "version",
     "SeekrFlow",