fxn-llm 0.0.1

Use local LLMs in your Python apps, with GPU acceleration and zero dependencies.

Function LLM for Python

Use local LLMs in your Python apps, with GPU acceleration and zero dependencies. This package is designed to patch OpenAI and Anthropic clients for running inference locally, using predictors hosted on Function.

[!IMPORTANT] This package is still a work-in-progress, so the API could change drastically between all releases.

Installing Function LLM

Function is distributed on PyPi. To install, open a terminal and run the following command:

# Install Function LLM
$ pip install --upgrade fxn-llm

[!IMPORTANT] Make sure to create an access key by signing onto Function. You'll need it to fetch the predictor at runtime.

[!NOTE] Function LLM requires Python 3.10+

Using the OpenAI Client Locally

To run text generation and embedding models locally using the OpenAI client, patch your OpenAI instance with the locally function:

from openai import OpenAI
from fxn_llm import locally

# 💥 Create your OpenAI client
openai = OpenAI()

# 🔥 Make it local
openai = locally(openai)

# 🚀 Generate embeddings
embeddings = openai.embeddings.create(
    model="@nomic/nomic-embed-text-v1.5-quant",
    input="search_query: Hello world!"
)

[!WARNING] Currently, only openai.embeddings.create is supported. Text generation is coming soon!

Using the Anthropic Client Locally

To run text generation models locally using the Anthopic client, patch your Anthropic instance with the locally function and the following configuration:

from anthropic import Anthropic
from fxn_llm import locally

# 💥 Create your Anthropic client
anthropic = Anthropic()

# 🔥 Make it local
anthropic = locally(openai, provider="anthropic")

# 🚀 Chat
message = anthropic.messages.create(
  model="@meta/llama-3.1-8b-quant",
  messages=[{ "role": "user", "content": "Hello, Llama" }],
  max_tokens=1024,
)

[!CAUTION] Anthropic support is not functional. It is still a work-in-progress.

---

Useful Links

• Discover predictors to use in your apps.
• Join our Discord community.
• Check out our docs.
• Learn more about us on our blog.
• Reach out to us at [e-mail].

Function is a product of NatML Inc.