Question Answering - OpenAI
In short
Section titled “In short”- The OpenAI Question and Answer (Q&A) module is a Weaviate module for answer extraction from data through the OpenAI completions endpoint or the Azure OpenAI equivalent.
- The module depends on a text vectorization module that should be running with Weaviate.
- The module adds an
ask {}operator to the GraphQLGet {}queries - The module returns a max. of 1 answer in the GraphQL
_additional {}field. - The answer with the highest
certainty(confidence level) will be returned.
Azure OpenAI or OpenAI?
The module usage instructions may vary based on whether you are using OpenAI directly or Azure OpenAI. Please make sure that you are following the right instructions for your service provider.
For example, the following may vary:
- Parameter names used in the schema, and
- Names of the API key to be used.
Introduction
Section titled “Introduction”The Question and Answer (Q&A) OpenAI module is a Weaviate module for answer extraction from data. It uses an OpenAI completions endpoint to try and extract an answer from the most relevant docs.
This module can be used in GraphQL Get{...} queries, as a search operator. The qna-openai module tries to find an answer in the data objects of the specified class. If an answer is found within the given certainty range, it will be returned in the GraphQL _additional { answer { ... } } field. There will be a maximum of 1 answer returned, if this is above the optionally set certainty. The answer with the highest certainty (confidence level) will be returned.
Inference API key
Section titled “Inference API key”qna-openai requires an API key from OpenAI or Azure OpenAI.
Organization name
Section titled “Organization name”For requests that require the OpenAI organization name, you can provide it at query time by adding it to the HTTP header:
"X-OpenAI-Organization": "YOUR-OPENAI-ORGANIZATION"for OpenAI
Providing the key to Weaviate
Section titled “Providing the key to Weaviate”You can provide your API key in two ways:
- During the configuration of your Docker instance, by adding
OPENAI_APIKEYorAZURE_APIKEYas appropriate underenvironmentto yourDocker Composefile, like this:
environment:
OPENAI_APIKEY: 'your-key-goes-here' # For use with OpenAI. Setting this parameter is optional; you can also provide the key at runtime.
AZURE_APIKEY: 'your-key-goes-here' # For use with Azure OpenAI. Setting this parameter is optional; you can also provide the key at runtime.
...- At run-time (recommended), by providing
"X-OpenAI-Api-Key"or"X-Azure-Api-Key"through the request header. You can provide it using the Weaviate client, like this:
import weaviateclient = weaviate.Client( url = "https://WEAVIATE_INSTANCE_URL", # Replace WEAVIATE_INSTANCE_URL with the URL additional_headers = { "X-OpenAI-Api-Key": "YOUR-OPENAI-API-KEY", # Replace with your API key "X-Azure-Api-Key": "YOUR-AZURE-API-KEY", # Replace with your API key })package mainimport ( "context" "fmt" "github.com/weaviate/weaviate-go-client/v5/weaviate" "github.com/weaviate/weaviate/entities/models")func main() { cfg := weaviate.Config{ Host: "WEAVIATE_INSTANCE_URL/", // Replace with your Weaviate endpoint Scheme: "https", // Replace with your API key Headers: map[string]string{ "X-OpenAI-Api-Key": "YOUR-OPENAI-API-KEY", // Replace with your API key "X-Azure-Api-Key": "YOUR-AZURE-API-KEY", // Replace with your API key } } client, err := weaviate.NewClient(cfg) if err != nil { panic(err) }}Module configuration
Section titled “Module configuration”Docker Compose file (Weaviate Database only)
Section titled “Docker Compose file (Weaviate Database only)”You can enable the OpenAI Q&A module in your Docker Compose file (e.g. docker-compose.yml). Add the qna-openai module (alongside any other module you may need) to the ENABLE_MODULES property, like this:
ENABLE_MODULES: 'text2vec-openai,qna-openai'Here is a full example of a Docker configuration, which uses the qna-openai module in combination with text2vec-openai:
---
services:
weaviate:
command:
- --host
- 0.0.0.0
- --port
- '8080'
- --scheme
- http
image:
cr.weaviate.io/semitechnologies/weaviate:1.38.2
ports:
- 8080:8080
- 50051:50051
restart: on-failure:0
environment:
QUERY_DEFAULTS_LIMIT: 25
AUTHENTICATION_ANONYMOUS_ACCESS_ENABLED: 'true'
PERSISTENCE_DATA_PATH: '/var/lib/weaviate'
ENABLE_MODULES: 'text2vec-openai,qna-openai'
OPENAI_APIKEY: sk-foobar # For use with OpenAI. Setting this parameter is optional; you can also provide the key at runtime.
OPENAI_ORGANIZATION: your-orgname # For use with OpenAI. Setting this parameter is optional; you can also provide the key at runtime.
AZURE_APIKEY: sk-foobar # For use with Azure OpenAI. Setting this parameter is optional; you can also provide the key at runtime.
CLUSTER_HOSTNAME: 'node1'Schema configuration
Section titled “Schema configuration”You can define settings for this module in the schema.
OpenAI vs Azure OpenAI
Section titled “OpenAI vs Azure OpenAI”- OpenAI users can optionally set the
modelparameter. - Azure OpenAI users must set the parameters
resourceNameanddeploymentId.
Model parameters
Section titled “Model parameters”You can also configure additional parameters for the model through the parameters shown below.
Example schema
Section titled “Example schema”For example, the following schema configuration will set Weaviate to use the qna-openai model with the Document class.
The following schema configuration uses the gpt-3.5-turbo-instruct model.
{
"classes": [
{
"class": "Document",
"description": "A class called document",
"vectorizer": "text2vec-openai",
"moduleConfig": {
"qna-openai": {
"model": "gpt-3.5-turbo-instruct", // For OpenAI
"resourceName": "<YOUR-RESOURCE-NAME>", // For Azure OpenAI
"deploymentId": "<YOUR-MODEL-NAME>", // For Azure OpenAI
"maxTokens": 16, // Applicable to both OpenAI and Azure OpenAI
"temperature": 0.0, // Applicable to both OpenAI and Azure OpenAI
"topP": 1, // Applicable to both OpenAI and Azure OpenAI
"frequencyPenalty": 0.0, // Applicable to both OpenAI and Azure OpenAI
"presencePenalty": 0.0 // Applicable to both OpenAI and Azure OpenAI
}
},
"properties": [
{
"dataType": [
"text"
],
"description": "Content that will be vectorized",
"name": "content"
}
]
}
]
}For information on how to use the individual parameters you can check here
How to use
Section titled “How to use”This module adds a search operator to GraphQL Get{...} queries: ask{}. This operator takes the following arguments:
| Field | Data Type | Required | Example value | Description |
|---|---|---|---|---|
question |
string | yes | "What is the name of the Dutch king?" |
The question to be answered. |
properties |
list of strings | no | ["summary"] |
The properties of the queries Class which contains text. If no properties are set, all are considered. |
Notes:
- The GraphQL
Explore { }function does support theasksearcher, but the result is only a beacon to the object containing the answer. It is thus not any different from performing a nearText semantic search with the question. No extraction is happening. - You cannot use the
'ask'operator along with a'neaXXX'operator!
Example query
Section titled “Example query”{
Get {
Article(
ask: {
question: "Who is Stanley Kubrick?",
properties: ["summary"]
},
limit: 1
) {
title
_additional {
answer {
hasAnswer
property
result
startPosition
endPosition
}
}
}
}
}import weaviate
client = weaviate.Client(
url="http://localhost:8080",
additional_headers={
"X-OpenAI-Api-Key": "YOUR-OPENAI-API-KEY"
}
)
ask = {
"question": "Who is Stanley Kubrick?",
"properties": ["summary"]
}
result = (
client.query
.get("Article", ["title", "_additional {answer {hasAnswer property result startPosition endPosition} }"])
.with_ask(ask)
.with_limit(1)
.do()
)
print(result)package main
import (
"context"
"fmt"
"github.com/weaviate/weaviate-go-client/v5/weaviate"
"github.com/weaviate/weaviate-go-client/v5/weaviate/graphql"
)
func main() {
cfg := weaviate.Config{
Host: "localhost:8080",
Scheme: "http",
Headers: map[string]string{"X-OpenAI-Api-Key": "YOUR-OPENAI-API-KEY"},
}
client, err := weaviate.NewClient(cfg)
if err != nil {
panic(err)
}
className := "Article"
fields := []graphql.Field{
{Name: "title"},
{Name: "_additional", Fields: []graphql.Field{
{Name: "answer", Fields: []graphql.Field{
{Name: "hasAnswer"},
{Name: "property"},
{Name: "result"},
{Name: "startPosition"},
{Name: "endPosition"},
}},
}},
}
ask := client.GraphQL().AskArgBuilder().
WithQuestion("Who is Stanley Kubrick?").
WithProperties([]string{"summary"})
ctx := context.Background()
result, err := client.GraphQL().Get().
WithClassName(className).
WithFields(fields...).
WithAsk(ask).
WithLimit(1).
Do(ctx)
if err != nil {
panic(err)
}
fmt.Printf("%v", result)
}echo '{
"query": "{
Get {
Article(
ask: {
question: \"Who is Stanley Kubrick?\",
properties: [\"summary\"]
},
limit: 1
) {
title
_additional {
answer {
hasAnswer
property
result
startPosition
endPosition
}
}
}
}
}
"
}' | curl \
-X POST \
-H 'Content-Type: application/json' \
-H 'Authorization: Bearer learn-weaviate' \
-H "X-OpenAI-Api-Key: $OPENAI_API_KEY" \
-d @- \
https://edu-demo.weaviate.network/v1/graphqlGraphQL response
Section titled “GraphQL response”The answer is contained in a new GraphQL _additional property called answer. It contains the following fields:
hasAnswer(boolean): could an answer be found?result(nullablestring): An answer if one could be found.nullifhasAnswer==falseproperty(nullablestring): The property which contains the answer.nullifhasAnswer==falsestartPosition(int): The character offset where the answer starts.0ifhasAnswer==falseendPosition(int): The character offset where the answer ends0ifhasAnswer==false
Note: startPosition, endPosition and property in the response are not guaranteed to be present. They are calculated by a case-insensitive string matching function against the input text. If the transformer model formats the output differently (e.g. by introducing spaces between tokens which were not present in the original input), the calculation of the position and determining the property fails.
Example response
Section titled “Example response”{
"data": {
"Get": {
"Document": [
{
"_additional": {
"answer": {
"hasAnswer": true,
"result": " Stanley Kubrick is an American filmmaker who is best known for his films, including \"A Clockwork Orange,\" \"Eyes Wide Shut,\" and \"The Shining.\""
}
}
}
]
}
}
}Token limits
Section titled “Token limits”If the number of input tokens exceed the limit of the model, the module will return the OpenAI API's error.
How it works (under the hood)
Section titled “How it works (under the hood)”Under the hood, the model uses a two-step approach. First it performs a semantic search to find the documents (e.g. a Sentence, Paragraph, Article, etc.) most likely to contain the answer. In a second step, Weaviate creates the required prompt as an input to an external call made to the OpenAI Completions endpoint. Weaviate uses the most relevant documents to establish a prompt for which OpenAI extracts the answer. There are three possible outcomes:
- No answer was found because the question can not be answered,
- An answer was found, but did not meet the user-specified minimum certainty, so it was discarded (typically the case when the document is on topic, but does not contain an actual answer to the question), and
- An answer was found that matches the desired certainty. It is returned to the user.
The module performs a semantic search under the hood, so a text2vec-... module is required. It does not need to be of the same type as the qna-... module. For example, you can use a text2vec-contextionary module to perform the semantic search, and a qna-openai module to extract the answer.
Additional information
Section titled “Additional information”Available models
Section titled “Available models”We recommend using:
gpt-3.5-turbo-instruct
The following models are now deprecated:
text-ada-001text-babbage-001text-curie-001text-davinci-002text-davinci-003
Questions and feedback
Section titled “Questions and feedback”Have a question or feedback? Here's how to reach us.