-
Notifications
You must be signed in to change notification settings - Fork 121
[Search] Drafts how to use OpenAI compatible models with the inference API #935
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changes from 14 commits
d221e33
f515131
6d25e1d
4a07481
a30701f
52c6262
e8bd1f8
521e437
13a7da4
b48d833
43862e5
472c2d1
477dc7c
999a2f3
d1895a2
dd1389e
719daef
14a6291
67b419a
98c786e
2a6566a
64d6769
289f44d
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,185 @@ | ||
--- | ||
applies_to: | ||
stack: ga | ||
serverless: ga | ||
navigation_title: Using OpenAI compatible models | ||
--- | ||
|
||
# Using OpenAI compatible models with the {{infer-cap}} API | ||
|
||
{{es}} enables you to use LLMs through the {{infer}} API, supporting providers such as Amazon Bedrock, Cohere, Google AI, HuggingFace, OpenAI, and more, as a service. | ||
It also allows you to use models deployed in your local environment that have an OpenAI compatible API. | ||
|
||
This page walks you through the process of connecting local models to {{es}} using Ollama. | ||
|
||
[Ollama](https://ollama.com/) enables you to download and run LLM models on your own infrastructure. | ||
For a list of available models compatible with Ollama, refer to this [page](https://ollama.com/library). | ||
|
||
Using Ollama ensures that your interactions remain private, as the models run on your infrastructure. | ||
|
||
## Overview | ||
|
||
In this tutorial, you learn how to: | ||
|
||
* download and run Ollama, | ||
* use ngrok to expose your local web server hosting Ollama over the internet | ||
* connect your local LLM to Playground | ||
|
||
## Download and run Ollama | ||
|
||
1. [Download Ollama](https://ollama.com/download). | ||
2. Install Ollama using the downloaded file. | ||
Enable the command line tool for Ollama during installation. | ||
3. Choose a model from the [list of supported LLMs](https://ollama.com/library). | ||
This tutorial uses `llama 3.2`. | ||
4. Run the following command: | ||
```shell | ||
ollama pull llama3.2 | ||
``` | ||
|
||
### Test the installed model | ||
|
||
After installation, test the model. | ||
|
||
1. Run `ollama run llama3.2` and ask a question, for example, "Are you working?" | ||
If the model is installed successfully, you receive a valid response. | ||
2. When the model is running, an API endpoint is enabled by default on port `11434`. | ||
To test it, make a request to the API using the following command: | ||
```shell | ||
curl http://localhost:11434/api/generate -d '{ | ||
"model": "llama3.2", | ||
"prompt": "What is the capital of France?" | ||
}' | ||
``` | ||
|
||
Refer to the API [documentation](https://github.com/ollama/ollama/blob/main/docs/api.md) to learn more. | ||
The API returns a response similar to this: | ||
```json | ||
{"model":"llama3.2","created_at":"2025-03-26T10:07:05.500614Z","response":"The","done":false} | ||
{"model":"llama3.2","created_at":"2025-03-26T10:07:05.519131Z","response":" capital","done":false} | ||
{"model":"llama3.2","created_at":"2025-03-26T10:07:05.537432Z","response":" of","done":false} | ||
{"model":"llama3.2","created_at":"2025-03-26T10:07:05.556016Z","response":" France","done":false} | ||
{"model":"llama3.2","created_at":"2025-03-26T10:07:05.574815Z","response":" is","done":false} | ||
{"model":"llama3.2","created_at":"2025-03-26T10:07:05.592967Z","response":" Paris","done":false} | ||
{"model":"llama3.2","created_at":"2025-03-26T10:07:05.611558Z","response":".","done":false} | ||
{"model":"llama3.2","created_at":"2025-03-26T10:07:05.630715Z","response":"","done":true,"done_reason":"stop","context":[128006,9125,128007,271,38766,1303,33025,2696,25,6790,220,2366,18,271,128009,128006,882,128007,271,3923,374,279,6864,315,9822,30,128009,128006,78191,128007,271,791,6864,315,9822,374,12366,13],"total_duration":2232589542,"load_duration":1052276792,"prompt_eval_count":32,"prompt_eval_duration":1048833625,"eval_count":8,"eval_duration":130808916} | ||
``` | ||
|
||
## Expose the endpoint using ngrok | ||
|
||
Since the created endpoint only works locally, it cannot be accessed from external services (for example, your Elastic Cloud instance). | ||
szabosteve marked this conversation as resolved.
Show resolved
Hide resolved
|
||
[ngrok](https://ngrok.com/) enables you to expose a local port with a public URL. | ||
|
||
1. Create an ngrok account and follow the [official setup guide](https://dashboard.ngrok.com/get-started/setup). | ||
2. After installing and configuring the ngrok agent, expose the Ollama port by running: | ||
```shell | ||
ngrok http 11434 --host-header="localhost:11434" | ||
``` | ||
The command returns a public link that works as long as ngrok and the Ollama server are running locally: | ||
```shell | ||
Session Status online | ||
Account [email protected] (Plan: Free) | ||
Version 3.18.4 | ||
Region United States (us) | ||
Latency 561ms | ||
Web Interface http://127.0.0.1:4040 | ||
Forwarding https://your-ngrok-endpoint.ngrok-free.app -> http://localhost:11434 | ||
|
||
|
||
Connections ttl opn rt1 rt5 p50 p90 | ||
0 0 0.00 0.00 0.00 0.00 | ||
``` | ||
|
||
3. Copy the ngrok-generated URL from the `Forwarding` line. | ||
4. Test the endpoint again using the new URL: | ||
```shell | ||
curl https://your-ngrok-endpoint.ngrok-free.app/api/generate -d '{ | ||
"model": "llama3.2", | ||
"prompt": "What is the capital of France?" | ||
}' | ||
``` | ||
The response should be similar to the previous one. | ||
|
||
## Connecting the local LLM to Playground | ||
szabosteve marked this conversation as resolved.
Show resolved
Hide resolved
|
||
|
||
Now, create a connector using the public URL from ngrok. | ||
|
||
1. In Kibana, go to **Search > Playground**, and click **Connect to an LLM**. | ||
2. Select **OpenAI** on the fly-out. | ||
3. Provide a name for the connector. | ||
4. Under **Connector settings**, select **Other (OpenAI Compatible Service)** as the OpenAI provider. | ||
5. Paste the ngrok-generated URL into the **URL** field and add the `v1/chat/completions` endpoint. For example: https://your-ngrok-endpoint.ngrok-free.app/v1/chat/completions | ||
6. Specify the default model, for example, `llama3.2`. | ||
7. Provide any random string for the API key (it will not be used for requests). | ||
8. **Save**. | ||
:::{image} /solutions/images/elasticsearch-openai-compatible-connector.png | ||
:alt: Configuring an LLM connector in Playground | ||
:screenshot: | ||
::: | ||
9. Click **Add data sources** and connect your index. | ||
|
||
You can now use Playground with the LLM running locally. | ||
|
||
## Using the local LLM with the {{infer}} API | ||
szabosteve marked this conversation as resolved.
Show resolved
Hide resolved
|
||
|
||
You can use your locally installed LLM with the {{infer}} API. | ||
|
||
Create the {{infer}} endpoint for a `chat_completion` task type with the `openai` service with the following request: | ||
|
||
```console | ||
PUT _inference/chat_completion/llama-completion | ||
{ | ||
"service": "openai", | ||
"service_settings": { | ||
"api_key": "ignored", <1> | ||
"model_id": "llama3.2", <2> | ||
"url": "https://your-ngrok-endpoint.ngrok-free.app/v1/chat/completions" <3> | ||
} | ||
} | ||
``` | ||
|
||
1. The `api_key` parameter is required for the `openai` service and must be set, but the specific value is not important for the local AI service. | ||
2. The model name. | ||
3. The ngrok-generated URL with the chat completion endpoint (`v1/chat/completions`). | ||
|
||
Verify if the {{infer}} endpoint working correctly: | ||
|
||
```console | ||
POST _inference/chat_completion/llama-completion/_stream | ||
{ | ||
"model": "llama3.2", | ||
"messages": [ | ||
{ | ||
"role": "user", | ||
"content": "What is the capital of France?" | ||
} | ||
], | ||
"temperature": 0.7, | ||
"max_completion_tokens": 300 | ||
} | ||
``` | ||
|
||
The request results in a response similar to this: | ||
|
||
```console-result | ||
event: message | ||
data: { | ||
"id" : "chatcmpl-416", | ||
"choices" : [ | ||
{ | ||
"delta" : { | ||
"content" : "The", | ||
"role" : "assistant" | ||
}, | ||
"index" : 0 | ||
} | ||
], | ||
"model" : "llama3.2", | ||
"object" : "chat.completion.chunk" | ||
} | ||
(...) | ||
``` | ||
|
||
## Further reading | ||
|
||
* [Unsing Ollama with the {{infer}} API](https://www.elastic.co/search-labs/blog/ollama-with-inference-api#expose-endpoint-to-the-internet-using-ngrok): A more comprehensive, end-to-end guide to using Ollama with {{es}}. | ||
szabosteve marked this conversation as resolved.
Show resolved
Hide resolved
|
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -43,6 +43,7 @@ toc: | |
- file: search/semantic-search/semantic-search-inference.md | ||
- file: search/semantic-search/semantic-search-elser-ingest-pipelines.md | ||
- file: search/semantic-search/cohere-es.md | ||
- file: search/semantic-search/using-openai-compatible-models.md | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Maybe moving this page up one level to sit directly under "AI-powered search" rather than nested under "Semantic search" would better reflect its broader applicability? Alternatively, we could even create a new top-level category, that highlights LLM integration as a distinct capability that cuts across multiple use cases. |
||
- file: search/rag.md | ||
children: | ||
- file: search/rag/playground.md | ||
|
Uh oh!
There was an error while loading. Please reload this page.