semvec/pypi-packages huggingface dataset:
h2o-helium 0.8.2
Client library for H2O Helium
Requires Python: >=3.8
Keywords: information-retrieval,LLM,large-language-models,question-answering,search,semantic-search,analytical-search,lexical-search,document-search,natural-language-querying

# H2O Helium Python Client

## Install

```bash
pip install h2o-helium
```

## Usage

```python
from h2o_helium import Helium

helium = Helium(address='https://helium.h2o.ai', api_key='sk-xxxxxx')

# Create a new collection
collection_id = helium.create_collection(name='Contracts', description='Paper clip supply contracts')

# Upload documents
with open('/path/to/dunder_mifflin.pdf', 'rb') as f:
    dunder_mifflin = helium.upload('Dunder Mifflin.pdf', f)
with open('/path/to/wernham_hogg.pdf', 'rb') as f:
    initech = helium.upload('Wernham Hogg.pdf', f)

# Ingest documents
helium.ingest_uploads(collection_id, [dunder_mifflin, initech])

# Create a chat session
chat_session_id = helium.create_chat_session(collection_id)

# Query the collection
with helium.connect(chat_session_id) as session:
    reply = session.query('How many paper clips were shipped to Scranton?', timeout=10)
    print(reply.content)
    
    reply = session.query('Did David Brent co-sign the contract with Initech?', timeout=10)
    print(reply.content)
```