bulk-chain 1.2.0

A lightweight, no-strings-attached Chain-of-Thought framework for your LLM, ensuring reliable results for bulk input requests.

bulk-chain 1.2.0

Third-party providers hosting↗️
👉demo👈

A no-strings-attached framework for your LLM that allows applying Chain-of-Thought-alike prompt schema towards a massive textual collections using custom third-party providers ↗️.

Main Features

• ✅ No-strings: you're free to LLM dependencies and flexible venv customization.
• ✅ Support schemas descriptions for Chain-of-Thought concept.
• ✅ Provides iterator over infinite amount of input contexts

Installation

From PyPI:

pip install --no-deps bulk-chain

or latest version from here:

pip install git+https://github.com/nicolay-r/bulk-chain@master

Chain-of-Thought Schema

To declare Chain-of-Though (CoT) schema we use JSON format. The field schema is a list of CoT instructions for the Large Language Model. Each item of the list represent a dictionary with prompt and out keys that corresponds to the input prompt and output variable name respectively. All the variable names should be mentioned in {}.

Example:

[
  {"prompt": "Given customer message: {text}, detect the customer's intent?", "out": "intent" },
  {"prompt": "Given customer message: {text}, extract relevant entities?", "out": "entities"},
  {"prompt": "Given intent: {intent} and entities: {entities}, generate a concise response or action recommendation for support agent.", "out": "action"}
]

Usage

🤖 Prepare

1. schema
   • Example for Sentiment Analysis
2. LLM model from the Third-party providers hosting↗️.
3. Data (iter of dictionaries)

🚀 Launch

API: For more details see the related Wiki page

from bulk_chain.core.utils import dynamic_init
from bulk_chain.api import iter_content

content_it = iter_content(
    # 1. Your schema.              
    schema=[
      {"prompt": "Given customer message: {text}, detect the customer's intent?", "out": "intent" },
      {"prompt": "Given customer message: {text}, extract relevant entities?", "out": "entities"},
      {"prompt": "Given intent: {intent} and entities: {entities}, generate a concise response or action recommendation for support agent.", "out": "action"}
    ],
    # 2. Your third-party model implementation.
    llm=dynamic_init(class_filepath="replicate_104.py")(
       api_token="<API-KEY>",
       model_name="meta/meta-llama-3-70b-instruct"),
    # 3. Customize your inference and result providing modes: 
    infer_mode="batch_async", 
    # 4. Your iterator of dictionaries
    input_dicts_it=YOUR_DATA_IT,
)
    
for content in content_it:
    # Handle your LLM responses here ...

Inference Modes

The type of the content variable depends on infer_mode declared above.

Below is the table that describes expected content type to the infer mode:

Infer Modes	Content Type
single	record
single_stream	chunk
batch	batch
batch_async	batch
batch_stream_async	chunk

API

Method	Mode	Description
ask(prompt)	Sync	Infers the model with a single prompt.
ask_stream(prompt)	Sync	Returns a generator that yields chunks of the inferred result.
ask_async(prompt)	Async	Asynchronously infers the model with a single prompt.
ask_stream_async(prompt)	Async	Asynchronously returns a generator of result chunks of the inferred result.

See examples with models at nlp-thirdgate 🌌.