97e91a83f3
Ruff / Ruff (push) Has been cancelled
Test / Core Tests (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.10) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.11) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.12) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.13) (push) Has been cancelled
Test / Offline Coverage Tests (Python 3.9) (push) Has been cancelled
Test / Full Coverage (Python 3.11) (push) Has been cancelled
Test / Core Provider Tests (OpenAI) (push) Has been cancelled
Test / Core Provider Tests (Anthropic) (push) Has been cancelled
Test / Core Provider Tests (Google) (push) Has been cancelled
Test / Core Provider Tests (Other) (push) Has been cancelled
Test / Anthropic Tests (push) Has been cancelled
Test / Gemini Tests (push) Has been cancelled
Test / Google GenAI Tests (push) Has been cancelled
Test / Vertex AI Tests (push) Has been cancelled
Test / OpenAI Tests (push) Has been cancelled
Test / Writer Tests (push) Has been cancelled
Test / Auto Client Tests (push) Has been cancelled
ty / type-check (push) Has been cancelled
109 lines
3.9 KiB
Markdown
109 lines
3.9 KiB
Markdown
---
|
|
draft: False
|
|
date: 2024-02-12
|
|
title: "Structured outputs with llama-cpp-python, a complete guide w/ instructor"
|
|
description: "Complete guide to using Instructor with llama-cpp-python. Learn how to generate structured, type-safe outputs with llama-cpp-python."
|
|
slug: llama-cpp-python
|
|
tags:
|
|
- patching
|
|
authors:
|
|
- jxnl
|
|
---
|
|
|
|
# Structured outputs with llama-cpp-python, a complete guide w/ instructor
|
|
|
|
This guide demonstrates how to use llama-cpp-python with Instructor to generate structured outputs. You'll learn how to use JSON schema mode and speculative decoding to create type-safe responses from local LLMs.
|
|
|
|
Open-source LLMS are gaining popularity, and llama-cpp-python has made the `llama-cpp` model available to obtain structured outputs using JSON schema via a mixture of [constrained sampling](https://llama-cpp-python.readthedocs.io/en/latest/#json-schema-mode) and [speculative decoding](https://llama-cpp-python.readthedocs.io/en/latest/#speculative-decoding).
|
|
|
|
They also support a [OpenAI compatible client](https://llama-cpp-python.readthedocs.io/en/latest/#openai-compatible-web-server), which can be used to obtain structured output as a in process mechanism to avoid any network dependency.
|
|
|
|
<!-- more -->
|
|
|
|
## Patching
|
|
|
|
Instructor's patch enhances an create call it with the following features:
|
|
|
|
- `response_model` in `create` calls that returns a pydantic model
|
|
- `max_retries` in `create` calls that retries the call if it fails by using a backoff strategy
|
|
|
|
!!! note "Learn More"
|
|
|
|
To learn more, please refer to the [docs](../index.md). To understand the benefits of using Pydantic with Instructor, visit the tips and tricks section of the [why use Pydantic](../why.md) page. If you want to check out examples of using Pydantic with Instructor, visit the [examples](../examples/index.md) page.
|
|
|
|
### See Also
|
|
|
|
- [Getting Started](../getting-started.md) - Quick start guide
|
|
- [Ollama Integration](./ollama.md) - Alternative local model setup
|
|
- [Local Classification](../examples/local_classification.md) - Classification with local models
|
|
- [Open Source Models](../examples/open_source.md) - More open-source model examples
|
|
|
|
# llama-cpp-python
|
|
|
|
Recently llama-cpp-python added support for structured outputs via JSON schema mode. This is a time-saving alternative to extensive prompt engineering and can be used to obtain structured outputs.
|
|
|
|
In this example we'll cover a more advanced use case of JSON_SCHEMA mode to stream out partial models. To learn more [partial streaming](https://github.com/jxnl/instructor/concepts/partial.md) check out partial streaming.
|
|
|
|
## Quick Start with `from_provider`
|
|
|
|
If you run the `llama-cpp-python` server in OpenAI compatible mode, you can use the unified `from_provider` API to patch the client. Simply point the base URL at your local server:
|
|
|
|
```python
|
|
import instructor
|
|
|
|
# Sync client
|
|
client = instructor.from_provider(
|
|
"ollama/openhermes", base_url="http://localhost:8080/v1"
|
|
)
|
|
|
|
# Async client
|
|
async_client = instructor.from_provider(
|
|
"ollama/openhermes", async_client=True, base_url="http://localhost:8080/v1"
|
|
)
|
|
```
|
|
|
|
You can then call `chat.completions.create` just like with any other provider.
|
|
|
|
```python
|
|
import llama_cpp
|
|
import instructor
|
|
from llama_cpp.llama_speculative import LlamaPromptLookupDecoding
|
|
from pydantic import BaseModel
|
|
|
|
|
|
llama = llama_cpp.Llama(
|
|
model_path="../../models/OpenHermes-2.5-Mistral-7B-GGUF/openhermes-2.5-mistral-7b.Q4_K_M.gguf",
|
|
n_gpu_layers=-1,
|
|
chat_format="chatml",
|
|
n_ctx=2048,
|
|
draft_model=LlamaPromptLookupDecoding(num_pred_tokens=2),
|
|
logits_all=True,
|
|
verbose=False,
|
|
)
|
|
|
|
|
|
create = instructor.patch(
|
|
create=llama.create_chat_completion_openai_v1,
|
|
mode=instructor.Mode.JSON_SCHEMA,
|
|
)
|
|
|
|
|
|
class UserDetail(BaseModel):
|
|
name: str
|
|
age: int
|
|
|
|
|
|
user = create(
|
|
messages=[
|
|
{
|
|
"role": "user",
|
|
"content": "Extract `Jason is 30 years old`",
|
|
}
|
|
],
|
|
response_model=UserDetail,
|
|
)
|
|
|
|
print(user)
|
|
#> name='Jason' age=30
|
|
```
|