seekrai 0.0.1__py3-none-any.whl → 0.1.1__py3-none-any.whl

seekrai-0.0.1.dist-info/METADATA

@@ -1,401 +0,0 @@
-Metadata-Version: 2.1
-Name: seekrai
-Version: 0.0.1
-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.8,<4.0
-Classifier: License :: OSI Approved :: Apache Software License
-Classifier: Operating System :: POSIX :: Linux
-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
-Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: aiohttp (>=3.9.3,<4.0.0)
-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: numpy (>=1.23.5) ; python_version < "3.12"
-Requires-Dist: numpy (>=1.26.0) ; python_version >= "3.12"
-Requires-Dist: openai (>=1.25.0,<2.0.0)
-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](https://pypi.org/project/seekrflow/) 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
-
-> 🚧
-> The library was rewritten in v1.0.0 released in April of 2024. There were significant changes made.
-
-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://api.seekrflow.xyz/) to obtain a SeekrFlow API Key.
-
-Once logged in to the SeekrFlow Playground, you can find available API keys in [this settings page](https://api.seekrflow.xyz/settings/api-keys).
-
-### Setting environment variable
-
-```shell
-export SEEKRFLOW_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("SEEKRFLOW_API_KEY"))
-
-response = client.chat.completions.create(
-    model="mistralai/Mixtral-8x7B-Instruct-v0.1",
-    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("SEEKRFLOW_API_KEY"))
-stream = client.chat.completions.create(
-    model="mistralai/Mixtral-8x7B-Instruct-v0.1",
-    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("SEEKRFLOW_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("SEEKRFLOW_API_KEY"))
-    tasks = [
-        async_client.chat.completions.create(
-            model="mistralai/Mixtral-8x7B-Instruct-v0.1",
-            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))
-```
-
-## Completions
-
-Completions are for code and language models shown [here](https://docs.seekrflow.ai/docs/inference-models). Below, a code model example is shown.
-
-```python
-import os
-from seekrai import SeekrFlow
-
-client = SeekrFlow(api_key=os.environ.get("SEEKRFLOW_API_KEY"))
-
-response = client.completions.create(
-    model="codellama/CodeLlama-34b-Python-hf",
-    prompt="Write a Next.js component with TailwindCSS for a header component.",
-)
-print(response.choices[0].text)
-```
-
-### Streaming
-
-```python
-import os
-from seekrai import SeekrFlow
-
-client = SeekrFlow(api_key=os.environ.get("SEEKRFLOW_API_KEY"))
-stream = client.completions.create(
-    model="codellama/CodeLlama-34b-Python-hf",
-    prompt="Write a Next.js component with TailwindCSS for a header component.",
-    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("SEEKRFLOW_API_KEY"))
-prompts = [
-    "Write a Next.js component with TailwindCSS for a header component.",
-    "Write a python function for the fibonacci sequence",
-]
-
-
-async def async_chat_completion(prompts):
-    async_client = AsyncSeekrFlow(api_key=os.environ.get("SEEKRFLOW_API_KEY"))
-    tasks = [
-        async_client.completions.create(
-            model="codellama/CodeLlama-34b-Python-hf",
-            prompt=prompt,
-        )
-        for prompt in prompts
-    ]
-    responses = await asyncio.gather(*tasks)
-
-    for response in responses:
-        print(response.choices[0].text)
-
-
-asyncio.run(async_chat_completion(prompts))
-```
-
-## Image generation
-
-```python
-import os
-from seekrai import SeekrFlow
-
-client = SeekrFlow(api_key=os.environ.get("SEEKRFLOW_API_KEY"))
-
-response = client.images.generate(
-    prompt="space robots",
-    model="stabilityai/stable-diffusion-xl-base-1.0",
-    steps=10,
-    n=4,
-)
-print(response.data[0].b64_json)
-```
-
-## Embeddings
-
-```python
-from typing import List
-from seekrai import SeekrFlow
-
-client = SeekrFlow(api_key=os.environ.get("SEEKRFLOW_API_KEY"))
-
-
-def get_embeddings(texts: List[str], model: str) -> List[List[float]]:
-    texts = [text.replace("\n", " ") for text in texts]
-    outputs = client.embeddings.create(model=model, input=texts)
-    return [outputs.data[i].embedding for i in range(len(texts))]
-
-
-input_texts = ['Our solar system orbits the Milky Way galaxy at about 515,000 mph']
-embeddings = get_embeddings(input_texts, model='seekrflowcomputer/m2-bert-80M-8k-retrieval')
-
-print(embeddings)
-```
-
-## 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("SEEKRFLOW_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.retrieve_content(id="file-d0d318cb-b7d9-493a-bd70-1cfe089d3815")  # retrieves content of 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("SEEKRFLOW_API_KEY"))
-
-client.fine_tuning.create(
-    training_file='file-d0d318cb-b7d9-493a-bd70-1cfe089d3815',
-    model='mistralai/Mixtral-8x7B-Instruct-v0.1',
-    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.cancel(id="ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b")  # Cancels a fine-tuning job
-client.fine_tuning.list_events(id="ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b")  # Lists events of a fine-tune job
-client.fine_tuning.download(
-    id="ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b")  # downloads compressed fine-tuned model or checkpoint to local disk
-```
-
-## Models
-
-This lists all the models that SeekrFlow supports.
-
-```python
-import os
-from seekrai import SeekrFlow
-
-client = SeekrFlow(api_key=os.environ.get("SEEKRFLOW_API_KEY"))
-
-models = client.models.list()
-
-for model in models:
-    print(model)
-```
-
-# Usage – CLI
-
-## Chat Completions
-
-```bash
-seekrai chat.completions \
-  --message "system" "You are a helpful assistant named SeekrFlow" \
-  --message "user" "What is your name?" \
-  --model mistralai/Mixtral-8x7B-Instruct-v0.1
-```
-
-The Chat Completions CLI enables streaming tokens to stdout by default. To disable streaming, use `--no-stream`.
-
-## Completions
-
-```bash
-seekrai completions \
-  "Large language models are " \
-  --model mistralai/Mixtral-8x7B-v0.1 \
-  --max-tokens 512 \
-  --stop "."
-```
-
-The Completions CLI enables streaming tokens to stdout by default. To disable streaming, use `--no-stream`.
-
-## Image Generations
-
-```bash
-seekrai images generate \
-  "space robots" \
-  --model stabilityai/stable-diffusion-xl-base-1.0 \
-  --n 4
-```
-
-The image is opened in the default image viewer by default. To disable this, use `--no-show`.
-
-## Files
-
-```bash
-# Help
-seekrai files --help
-
-# Check file
-seekrai files check example.jsonl
-
-# Upload file
-seekrai files upload example.jsonl
-
-# List files
-seekrai files list
-
-# Retrieve file metadata
-seekrai files retrieve file-6f50f9d1-5b95-416c-9040-0799b2b4b894
-
-# Retrieve file content
-seekrai files retrieve-content file-6f50f9d1-5b95-416c-9040-0799b2b4b894
-
-# Delete remote file
-seekrai files delete file-6f50f9d1-5b95-416c-9040-0799b2b4b894
-```
-
-## Fine-tuning
-
-```bash
-# Help
-seekrai fine-tuning --help
-
-# Create fine-tune job
-seekrai fine-tuning create \
-  --model seekrflowcomputer/llama-2-7b-chat \
-  --training-file file-711d8724-b3e3-4ae2-b516-94841958117d
-
-# List fine-tune jobs
-seekrai fine-tuning list
-
-# Retrieve fine-tune job details
-seekrai fine-tuning retrieve ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b
-
-# List fine-tune job events
-seekrai fine-tuning list-events ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b
-
-# Cancel running job
-seekrai fine-tuning cancel ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b
-
-# Download fine-tuned model weights
-seekrai fine-tuning download ft-c66a5c18-1d6d-43c9-94bd-32d756425b4b
-```
-
-## Models
-
-```bash
-# Help
-seekrai models --help
-
-# List models
-seekrai models list
-```
-
-## Contributing
-
-Refer to the [Contributing Guide](CONTRIBUTING.md)
-

seekrai-0.0.1.dist-info/RECORD

@@ -1,56 +0,0 @@
-seekrai/__init__.py,sha256=4Z_VK9I4lqX3r_1vtcUMPHRNm_9rkqjWJUzWOSlpHp0,1398
-seekrai/abstract/__init__.py,sha256=wNiOTW9TJpUgfCJCG-wAbhhWWH2PtoVpAuL3nxvQGps,56
-seekrai/abstract/api_requestor.py,sha256=fiXXaUxlOZeMlz62WfjpHSI1GUTTEvjGp1CxwbmHVag,25140
-seekrai/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-seekrai/cli/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-seekrai/cli/api/chat.py,sha256=yrEhs1XBFRztttOmha-hhshI0mLJsxjSvEakx-HuYcE,7840
-seekrai/cli/api/completions.py,sha256=rJkN8IWOD52J0q4w74elNfVOBDvyPJbnIGx_XgU7WSw,3690
-seekrai/cli/api/files.py,sha256=bP-0MlG5RhCAxiuLC5jwe303iXUceouLjRWGHMlBCwg,3188
-seekrai/cli/api/finetune.py,sha256=jzNA1WVVjDh6ttQpb-AVBG0Sd7Bj6KfWxZbhW3Xg510,4959
-seekrai/cli/api/images.py,sha256=D_L2shR1ki35jgjxiKaYitnjHG1ZZe6fWcDZwbFO8i4,2362
-seekrai/cli/api/models.py,sha256=gVa0o1FA5WaRfl84xkaqeOI__lU367xmEM9Rwror0xo,1126
-seekrai/cli/cli.py,sha256=kiW-QT8mPMujt80bptlaG8TMnfsAKl_a62BcFPLwois,1970
-seekrai/client.py,sha256=2WIsiZiKoneTQFWafVSyVbJtcO0UOER3HsvnRzgCZeU,4873
-seekrai/constants.py,sha256=uNl_HF0SPC3J0atqA1AluOuAGdaHgepaPxV_Ox6-j5k,955
-seekrai/error.py,sha256=HB7w4GkJj-wvC9tLWhYeHMIarQG088cSJ2GWr4N93Bs,5364
-seekrai/filemanager.py,sha256=3WWnaYoneMgfcweRzjNbqtRsPsPEaTSASdtUse93tjQ,11499
-seekrai/legacy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-seekrai/legacy/base.py,sha256=OSVtz7aKyeOoLv8sN2CqVwJUFOYcTCGeRhxkkCaEUEc,727
-seekrai/legacy/complete.py,sha256=a68qMLIrDxXENCPM2OB8Z4evxYRx43RwTFpyKGUQ-EE,2332
-seekrai/legacy/embeddings.py,sha256=dUqroJ1LrQs0kJX4guNDyZgIQm6cRBxzh6QtQvLCras,587
-seekrai/legacy/files.py,sha256=GqdhSzLPXQEQNTbXvIYSjEvKgYIFTHCeBCs68Yi4nlA,3849
-seekrai/legacy/finetune.py,sha256=nBWRqTpg6lP9E1jdrnrE9X2SgXTJf2J-l3MhV2h9zOc,4903
-seekrai/legacy/images.py,sha256=0lktig74NgdAhVCZediuVjTRV0hGuyKY6DWQ-mZxoQg,578
-seekrai/legacy/models.py,sha256=9OFssojcC6naHUIyNwsvwDEkmBVB3qjAjFphFSNzYd8,1047
-seekrai/resources/__init__.py,sha256=6m1J5kXqJFHqDzQkWy5Us-q3Ug1YZWWP1oQtJ_IGsRM,694
-seekrai/resources/chat/__init__.py,sha256=KmtPupgECtEN80NyvcnSmieTAFXhwmVxhMHP0qhspA4,618
-seekrai/resources/chat/completions.py,sha256=HAo43u1bMSIoIiiIt8ISpCBUGtTXJK6sF7KK74vYocw,11093
-seekrai/resources/completions.py,sha256=NVMSF_goqMLq0ynjXuEmi-wNN3FnmjO7FZhMmD-UuPQ,8414
-seekrai/resources/embeddings.py,sha256=RJaY2gu-U-z1eCda3igCtXfMZFya9jZ8je37oTIt5KY,2553
-seekrai/resources/files.py,sha256=qgX8Q60gBinuqDLPYzHp73kgqmKh7uxBbH2adUH7Cnk,4771
-seekrai/resources/finetune.py,sha256=cJhaxMOi-LspxghjwsJjyaRM0USQfwxti6nzvE5u6lg,12657
-seekrai/resources/images.py,sha256=ZFH0RQCRbppWr9WGszdqB6d1McWSh6EVKPMjRGByvZo,4782
-seekrai/resources/models.py,sha256=PJotW78B1wB7J_kXbx0TioC0z9uisU7Xt4BvMsPqSqo,1793
-seekrai/seekrflow_response.py,sha256=5RFEQzamDy7sTSDkxSsZQThZ3biNmeCPeHWdrFId5Go,1320
-seekrai/types/__init__.py,sha256=PJggVUXnzdgNB_4QktHjgSf3DS7Spkk21oVpXe5J3y8,1515
-seekrai/types/abstract.py,sha256=TqWFQV_6bPblywfCH-r8FCkXWvPkc9KlJ4QVgyrnaMc,642
-seekrai/types/chat_completions.py,sha256=GgqEcfdjkWecPVZEHRENctQ1SQ6KB4CculxyKhkvnHo,3571
-seekrai/types/common.py,sha256=2zdSkEVfQW7nnNM2fe5nU36ihJQpWLwZRD05XM_nrqM,1491
-seekrai/types/completions.py,sha256=lm9AFdZR3Xg5AHPkV-qETHikkwMJmkHrLGr5GG-YR-M,2171
-seekrai/types/embeddings.py,sha256=OANoLNOs0aceS8NppVvvcNYQbF7-pAOAmcr30pw64OU,749
-seekrai/types/error.py,sha256=uTKISs9aRC4_6zwirtNkanxepN8KY-SqCq0kNbfZylQ,370
-seekrai/types/files.py,sha256=fLxHH9yTD7u-8JtpQZWF0dSjYLo3MfLKvQFlhr9PnvU,1908
-seekrai/types/finetune.py,sha256=JbAuo1UE0QFIZbvb8u3S1oTBgdTIOQpyNwScU06y3sI,6020
-seekrai/types/images.py,sha256=Fusj8OhVYFsT8kz636lRGGivLbPXo_ZNgakKwmzJi3U,914
-seekrai/types/models.py,sha256=x8YG2MIaP3G1jiJKtuftRbmiUO3-Tt22c1f-GUaczP8,1011
-seekrai/utils/__init__.py,sha256=dfbiYEc47EBVRkq6C4O9y6tTGuPuV3LbV3__v01Mbds,658
-seekrai/utils/_log.py,sha256=IjKp_i0Q5Ryqrt-rcq7_Y0F4F9XdjQhx_fs5xyMQqCc,1665
-seekrai/utils/api_helpers.py,sha256=R4169B55RNorjTKuolaeu6rGvq4w9gPppolnXoSEf-I,2393
-seekrai/utils/files.py,sha256=B61Pwra49MVVWjPtdkx4hBtAuUe9UI63hdNus87Uq0o,7164
-seekrai/utils/tools.py,sha256=Lg_4y1Bi-EVTr5CXR7jg87IWUFdheMmhMM44QfAe72M,1789
-seekrai/version.py,sha256=q6iGQVFor8zXiPP5F-3vy9TndOxKv5JXbaNJ2kdOQws,125
-seekrai-0.0.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-seekrai-0.0.1.dist-info/METADATA,sha256=tD3ThHfxNaieESaDxXxTENA10YO5c88y8u9TFBFsCQU,11094
-seekrai-0.0.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-seekrai-0.0.1.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
-seekrai-0.0.1.dist-info/RECORD,,